From cd18159b26db407b85bda569dcdfab922ef9419c Mon Sep 17 00:00:00 2001
From: an-lee <an.lee.work@gmail.com>
Date: Tue, 17 Sep 2024 18:11:28 +0800
Subject: [PATCH] improve style

---
 .../components/messages/assistant-message.tsx |  2 +-
 .../components/messages/user-message.tsx      | 23 ++++++-------------
 2 files changed, 8 insertions(+), 17 deletions(-)

diff --git a/enjoy/src/renderer/components/messages/assistant-message.tsx b/enjoy/src/renderer/components/messages/assistant-message.tsx
index b8c415ea..c66c7d1a 100644
--- a/enjoy/src/renderer/components/messages/assistant-message.tsx
+++ b/enjoy/src/renderer/components/messages/assistant-message.tsx
@@ -183,7 +183,7 @@ export const AssistantMessageComponent = (props: {
 
         {configuration.type === "gpt" && (
           <MarkdownWrapper
-            className="message-content select-text prose dark:prose-invert"
+            className="message-content select-text prose dark:prose-invert max-w-full"
             data-source-type="Message"
             data-source-id={message.id}
           >
diff --git a/enjoy/src/renderer/components/messages/user-message.tsx b/enjoy/src/renderer/components/messages/user-message.tsx
index 05795532..9c79a674 100644
--- a/enjoy/src/renderer/components/messages/user-message.tsx
+++ b/enjoy/src/renderer/components/messages/user-message.tsx
@@ -19,7 +19,11 @@ import {
   DropdownMenuSeparator,
   toast,
 } from "@renderer/components/ui";
-import { SpeechPlayer, ConversationShortcuts } from "@renderer/components";
+import {
+  SpeechPlayer,
+  ConversationShortcuts,
+  MarkdownWrapper,
+} from "@renderer/components";
 import { useContext, useState } from "react";
 import { AppSettingsProviderContext } from "@renderer/context";
 import {
@@ -93,22 +97,9 @@ export const UserMessageComponent = (props: {
         </Avatar>
       </div>
       <div className="flex flex-col gap-2 px-4 py-2 bg-sky-500/30 border-sky-500 rounded-lg shadow-sm w-full mb-2">
-        <Markdown
-          className="select-text prose dark:prose-invert"
-          components={{
-            a({ node, children, ...props }) {
-              try {
-                new URL(props.href ?? "");
-                props.target = "_blank";
-                props.rel = "noopener noreferrer";
-              } catch (e) {}
-
-              return <a {...props}>{children}</a>;
-            },
-          }}
-        >
+        <MarkdownWrapper className="message-content select-text prose dark:prose-invert max-w-full">
           {message.content}
-        </Markdown>
+        </MarkdownWrapper>
 
         {Boolean(speech) && <SpeechPlayer speech={speech} />}