From 545f7d4d44db9e6a1d1564edb201d8c30797152b Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Sat, 21 Feb 2026 03:20:30 +0800
Subject: [PATCH] fix(ui): remove duplicate loading line and show agent avatar
 in auto mode

---
 components/image-generation-loading.tsx |  4 ++--
 components/multimodal-input.tsx         | 28 ++++++++++++++++---------
 2 files changed, 20 insertions(+), 12 deletions(-)

diff --git a/components/image-generation-loading.tsx b/components/image-generation-loading.tsx
index 2b2081096..f8468a23e 100644
--- a/components/image-generation-loading.tsx
+++ b/components/image-generation-loading.tsx
@@ -62,7 +62,7 @@ export const ImageGenerationLoading = memo(
           </div>
 
           <div className="text-[15px] leading-7 text-[#3a4352]">
-            现在开始生成设计图，使用 {resolvedModelName} 模型，预计需要 {estimateRangeLabel}。
+            现在开始生成设计图。
           </div>
 
           <div className="flex items-center gap-2 text-[#a8b1bf]">
@@ -73,7 +73,7 @@ export const ImageGenerationLoading = memo(
           <div
             className={cn(
               "relative overflow-hidden rounded-[22px] border border-[#e1e4ea] bg-[#eaedf2]",
-              count === 1 ? "h-[420px] w-full max-w-[560px]" : "grid gap-3 p-3",
+              count === 1 ? "h-[420px] w-full max-w-[460px]" : "grid gap-3 p-3",
               count > 1 && "grid-cols-2"
             )}
           >
diff --git a/components/multimodal-input.tsx b/components/multimodal-input.tsx
index 3313b3e1b..cf2078ad9 100644
--- a/components/multimodal-input.tsx
+++ b/components/multimodal-input.tsx
@@ -1478,31 +1478,39 @@ function PureModelSelectorCompact({
   }
 
   if (readOnly) {
+    const readOnlyAvatarUrl =
+      selectedModelId === "auto"
+        ? dynamicModelsData?.agentModel?.avatarUrl || selectedModel?.avatarUrl || null
+        : selectedModel?.avatarUrl || null;
+    const readOnlyName =
+      selectedModelId === "auto"
+        ? dynamicModelsData?.agentModel?.displayName || "Agent模式"
+        : selectedModel?.name || "模型";
+
     if (avatarOnly) {
       return (
-        <Button
-          className="h-8 w-8 rounded-full border border-border p-0 hover:bg-transparent"
-          variant="ghost"
-          disabled
-          aria-label={selectedModel?.name || "Agent模式"}
+        <div
+          className="flex h-8 w-8 items-center justify-center rounded-full border border-border bg-background"
+          aria-label={readOnlyName}
+          title={readOnlyName}
         >
-          {selectedModel?.avatarUrl ? (
+          {readOnlyAvatarUrl ? (
             <img
-              src={selectedModel.avatarUrl}
-              alt={selectedModel.name}
+              src={readOnlyAvatarUrl}
+              alt={readOnlyName}
               className="h-7 w-7 rounded-full object-cover"
             />
           ) : (
             <Bot className="h-4 w-4" />
           )}
-        </Button>
+        </div>
       );
     }
 
     return (
       <Button className="h-8 px-2" variant="ghost" disabled>
         <CpuIcon size={16} />
-        <span className="hidden font-medium text-xs sm:block">{selectedModel?.name}</span>
+        <span className="hidden font-medium text-xs sm:block">{readOnlyName}</span>
       </Button>
     );
   }