From 61f975d32d68456737735faef003b0e9f5fbabe1 Mon Sep 17 00:00:00 2001
From: cristhianzl <cristhian.lousa@gmail.com>
Date: Wed, 4 Sep 2024 17:37:02 -0300
Subject: [PATCH] feat: Refactor chatMessage component for improved readability
 and performance

---
 .../components/chatView/chatMessage/index.tsx | 51 ++++++++-----------
 1 file changed, 20 insertions(+), 31 deletions(-)

diff --git a/src/frontend/src/modals/IOModal/components/chatView/chatMessage/index.tsx b/src/frontend/src/modals/IOModal/components/chatView/chatMessage/index.tsx
index e1739be6196..ad1421661c8 100644
--- a/src/frontend/src/modals/IOModal/components/chatView/chatMessage/index.tsx
+++ b/src/frontend/src/modals/IOModal/components/chatView/chatMessage/index.tsx
@@ -113,18 +113,16 @@ export default function ChatMessage({
     }
   }, [lastMessage]);
 
+  const decodedMessage = decodeURIComponent(chatMessage ?? "");
+  const isEmpty = decodedMessage?.trim() === "";
+
   return (
     <>
-      <div
-        className={classNames(
-          "form-modal-chat-position",
-          chat.isSend ? "" : " ",
-        )}
-      >
+      <div className={cn("form-modal-chat-position", chat.isSend ? "" : " ")}>
         <div
-          className={classNames(
-            "mr-3 mt-1 flex w-24 flex-col items-center gap-1 overflow-hidden px-3 pb-3",
-          )}
+          className={
+            "mr-3 mt-1 flex w-24 flex-col items-center gap-1 overflow-hidden px-3 pb-3"
+          }
         >
           <div className="flex flex-col items-center gap-1">
             <div
@@ -184,7 +182,7 @@ export default function ChatMessage({
                   >
                     {useMemo(
                       () =>
-                        chatMessage === "" && lockChat ? (
+                        isEmpty && lockChat ? (
                           <IconComponent
                             name="MoreHorizontal"
                             className="h-8 w-8 animate-pulse"
@@ -195,7 +193,7 @@ export default function ChatMessage({
                             rehypePlugins={[rehypeMathjax]}
                             className={cn(
                               "markdown prose flex flex-col word-break-break-word dark:prose-invert",
-                              chatMessage === ""
+                              isEmpty
                                 ? "text-chat-trigger-disabled"
                                 : "text-primary",
                             )}
@@ -256,7 +254,7 @@ export default function ChatMessage({
                               },
                             }}
                           >
-                            {chatMessage === "" && !chat.stream_url
+                            {isEmpty && !chat.stream_url
                               ? EMPTY_OUTPUT_SEND_MESSAGE
                               : chatMessage}
                           </Markdown>
@@ -281,18 +279,13 @@ export default function ChatMessage({
                   Display Prompt
                   <IconComponent
                     name="ChevronDown"
-                    className={
-                      "h-3 w-3 transition-all " +
-                      (promptOpen ? "rotate-180" : "")
-                    }
+                    className={`h-3 w-3 transition-all ${promptOpen ? "rotate-180" : ""}`}
                   />
                 </button>
                 <span
                   className={cn(
                     "prose word-break-break-word dark:prose-invert",
-                    chatMessage !== ""
-                      ? "text-primary"
-                      : "text-chat-trigger-disabled",
+                    !isEmpty ? "text-primary" : "text-chat-trigger-disabled",
                   )}
                 >
                   {promptOpen
@@ -324,25 +317,21 @@ export default function ChatMessage({
                         }
                         return <p>{parts}</p>;
                       })
-                    : chatMessage === ""
+                    : isEmpty
                       ? EMPTY_INPUT_SEND_MESSAGE
                       : chatMessage}
                 </span>
               </>
             ) : (
               <div className="flex flex-col">
-                <span
-                  className={`prose word-break-break-word dark:prose-invert ${
-                    chatMessage === ""
-                      ? "text-chat-trigger-disabled"
-                      : "text-primary"
+                <div
+                  className={`whitespace-pre-wrap break-words ${
+                    isEmpty ? "text-chat-trigger-disabled" : "text-primary"
                   }`}
-                  data-testid={
-                    "chat-message-" + chat.sender_name + "-" + chatMessage
-                  }
+                  data-testid={`chat-message-${chat.sender_name}-${chatMessage}`}
                 >
-                  {chatMessage === "" ? EMPTY_INPUT_SEND_MESSAGE : chatMessage}
-                </span>
+                  {isEmpty ? EMPTY_INPUT_SEND_MESSAGE : decodedMessage}
+                </div>
                 {chat.files && (
                   <div className="my-2 flex flex-col gap-5">
                     {chat.files.map((file, index) => {
@@ -362,7 +351,7 @@ export default function ChatMessage({
           </div>
         )}
       </div>
-      <div id={lastMessage ? "last-chat-message" : ""}></div>
+      <div id={lastMessage ? "last-chat-message" : undefined} />
     </>
   );
 }