From b6333d0e4963e8ca625f67edc855abc7d6154e65 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <62335616+lucaseduoli@users.noreply.github.com>
Date: Fri, 14 Mar 2025 16:49:06 -0300
Subject: [PATCH] feat: add detailed view for prompt component (#7036)

* Changed prompt component to show the prompt with more details

* Fix classes for edit node

* Fixed test

---------

Co-authored-by: Rodrigo Nader <rodrigosilvanader@gmail.com>
---
 .../components/promptComponent/index.tsx      | 115 ++++++------------
 .../regression/generalBugs-shard-10.spec.ts   |   2 +-
 2 files changed, 36 insertions(+), 81 deletions(-)

diff --git a/src/frontend/src/components/core/parameterRenderComponent/components/promptComponent/index.tsx b/src/frontend/src/components/core/parameterRenderComponent/components/promptComponent/index.tsx
index e8da54b77..04778fcd9 100644
--- a/src/frontend/src/components/core/parameterRenderComponent/components/promptComponent/index.tsx
+++ b/src/frontend/src/components/core/parameterRenderComponent/components/promptComponent/index.tsx
@@ -1,48 +1,18 @@
-import { GRADIENT_CLASS } from "@/constants/constants";
+import SanitizedHTMLWrapper from "@/components/common/sanitizedHTMLWrapper";
+import { regexHighlight } from "@/constants/constants";
 import PromptModal from "@/modals/promptModal";
 import { cn } from "../../../../../utils/utils";
-import IconComponent from "../../../../common/genericIconComponent";
 import { Button } from "../../../../ui/button";
 import { getPlaceholder } from "../../helpers/get-placeholder-disabled";
 import { InputProps, PromptAreaComponentType } from "../../types";
 
 const promptContentClasses = {
-  base: "overflow-hidden text-clip whitespace-nowrap bg-background",
-  editNode: "input-edit-node input-dialog",
-  normal: "primary-input text-muted-foreground",
+  base: "overflow-hidden text-clip whitespace-nowrap bg-background h-fit max-h-28",
+  editNode: "input-edit-node input-dialog py-2",
+  normal: "primary-input text-primary",
   disabled: "disabled-state",
 };
 
-const externalLinkIconClasses = {
-  gradient: ({
-    disabled,
-    editNode,
-  }: {
-    disabled: boolean;
-    editNode: boolean;
-  }) =>
-    disabled
-      ? ""
-      : editNode
-        ? "gradient-fade-input-edit-node "
-        : "gradient-fade-input ",
-  background: ({
-    disabled,
-    editNode,
-  }: {
-    disabled: boolean;
-    editNode: boolean;
-  }) =>
-    disabled
-      ? ""
-      : editNode
-        ? "background-fade-input-edit-node "
-        : "background-fade-input",
-  icon: "icons-parameters-comp absolute right-3 h-4 w-4 shrink-0",
-  editNodeTop: "top-[0.375rem]",
-  normalTop: "top-2.5",
-};
-
 export default function PromptAreaComponent({
   field_name,
   nodeClass,
@@ -54,6 +24,24 @@ export default function PromptAreaComponent({
   id = "",
   readonly = false,
 }: InputProps<string, PromptAreaComponentType>): JSX.Element {
+  const coloredContent = (typeof value === "string" ? value : "")
+    .replace(/</g, "&lt;")
+    .replace(/>/g, "&gt;")
+    .replace(regexHighlight, (match, p1, p2) => {
+      // Decide which group was matched. If p1 is not undefined, do nothing
+      // we don't want to change the text. If p2 is not undefined, then we
+      // have a variable, so we should highlight it.
+      // ! This will not work with multiline or indented json yet
+      if (p1 !== undefined) {
+        return match;
+      } else if (p2 !== undefined) {
+        return `<span class="chat-message-highlight">{${p2}}</span>`;
+      }
+
+      return match;
+    })
+    .replace(/\n/g, "<br />");
+
   const renderPromptText = () => (
     <span
       id={id}
@@ -64,50 +52,20 @@ export default function PromptAreaComponent({
         disabled && !editNode && promptContentClasses.disabled,
       )}
     >
-      {value !== ""
-        ? value
-        : getPlaceholder(disabled, "Type your prompt here...")}
+      {value !== "" ? (
+        <SanitizedHTMLWrapper
+          className="m-0 whitespace-pre-wrap p-0 text-xs"
+          content={coloredContent}
+          suppressWarning={true}
+        />
+      ) : (
+        <span className="text-sm text-muted-foreground">
+          {getPlaceholder(disabled, "Type your prompt here...")}
+        </span>
+      )}
     </span>
   );
 
-  const renderExternalLinkIcon = () => (
-    <>
-      <div
-        className={cn(
-          externalLinkIconClasses.gradient({ disabled, editNode }),
-          editNode
-            ? externalLinkIconClasses.editNodeTop
-            : externalLinkIconClasses.normalTop,
-        )}
-        style={{
-          pointerEvents: "none",
-          background: disabled ? "" : GRADIENT_CLASS,
-        }}
-        aria-hidden="true"
-      />
-      <div
-        className={cn(
-          externalLinkIconClasses.background({ disabled, editNode }),
-          editNode
-            ? externalLinkIconClasses.editNodeTop
-            : externalLinkIconClasses.normalTop,
-          disabled && "bg-border",
-        )}
-        aria-hidden="true"
-      />
-      <IconComponent
-        name={disabled ? "lock" : "Scan"}
-        className={cn(
-          externalLinkIconClasses.icon,
-          editNode
-            ? externalLinkIconClasses.editNodeTop
-            : externalLinkIconClasses.normalTop,
-          disabled ? "text-placeholder-foreground" : "text-foreground",
-        )}
-      />
-    </>
-  );
-
   return (
     <div className={cn("w-full", disabled && "pointer-events-none")}>
       <PromptModal
@@ -124,10 +82,7 @@ export default function PromptAreaComponent({
           className="w-full"
           data-testid="button_open_prompt_modal"
         >
-          <div className="relative w-full">
-            {renderPromptText()}
-            {renderExternalLinkIcon()}
-          </div>
+          <div className="relative w-full">{renderPromptText()}</div>
         </Button>
       </PromptModal>
     </div>
diff --git a/src/frontend/tests/extended/regression/generalBugs-shard-10.spec.ts b/src/frontend/tests/extended/regression/generalBugs-shard-10.spec.ts
index fbe26c453..841ae6f44 100644
--- a/src/frontend/tests/extended/regression/generalBugs-shard-10.spec.ts
+++ b/src/frontend/tests/extended/regression/generalBugs-shard-10.spec.ts
@@ -76,7 +76,7 @@ test(
 
     await page.getByTestId("button_open_prompt_modal").click();
 
-    await page.getByTestId("edit-prompt-sanitized").first().click();
+    await page.getByTestId("edit-prompt-sanitized").last().click();
 
     await page
       .getByTestId("modal-promptarea_prompt_template")