From 1e7c7bb2e938f5481ab15a1ff7c2b5428fd76933 Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Sat, 21 Feb 2026 04:06:43 +0800
Subject: [PATCH] fix(canvas): show generation model label above output images

---
 components/canvas/CanvasChatPanel.tsx | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/components/canvas/CanvasChatPanel.tsx b/components/canvas/CanvasChatPanel.tsx
index 47dde8cbf..db43d5419 100644
--- a/components/canvas/CanvasChatPanel.tsx
+++ b/components/canvas/CanvasChatPanel.tsx
@@ -5105,6 +5105,11 @@ export function CanvasChatPanel({ chatId, initialMessages = [], onClose, onOpenF
               ((!hasAssistantMedia && nextAssistantHasMedia) || isLatestAssistantDuringGeneration);
             const showVoteActions = message.role === "assistant" && !hideVoteActions;
             const showModelMeta = hideVoteActions && Boolean(assistantModelName);
+            const mediaModelName =
+              assistantModelName ||
+              (message.role === "assistant" && selectedModelId !== "auto"
+                ? currentModel?.displayName || ""
+                : "");
             const shouldTypeAssistantMessage =
               message.role === "assistant" &&
               isLoading &&
@@ -5351,6 +5356,19 @@ export function CanvasChatPanel({ chatId, initialMessages = [], onClose, onOpenF
                                     ) : null}
                                     {imageUrls.length > 0 && (
                                       <div className={cn("space-y-2", textWithoutImages ? "mt-3" : "mt-0")}>
+                                        {mediaModelName ? (
+                                          <div className="flex items-center gap-1.5 text-[#a4acb8]">
+                                            <svg
+                                              viewBox="0 0 24 24"
+                                              fill="currentColor"
+                                              className="h-3.5 w-3.5"
+                                              aria-hidden="true"
+                                            >
+                                              <path d="M10.8 2.4a2.4 2.4 0 0 1 2.4 0l6.2 3.58a2.4 2.4 0 0 1 1.2 2.08v7.16a2.4 2.4 0 0 1-1.2 2.08l-6.2 3.58a2.4 2.4 0 0 1-2.4 0l-6.2-3.58a2.4 2.4 0 0 1-1.2-2.08V8.06a2.4 2.4 0 0 1 1.2-2.08l6.2-3.58Zm1.2 5.1a4.5 4.5 0 1 0 0 9 4.5 4.5 0 0 0 0-9Zm0 2.25a2.25 2.25 0 1 1 0 4.5 2.25 2.25 0 0 1 0-4.5Z" />
+                                            </svg>
+                                            <span className="text-[12px] font-medium">{mediaModelName}</span>
+                                          </div>
+                                        ) : null}
                                         <div className="flex flex-wrap gap-2">
                                           {imageUrls.map((url, idx) => (
                                             <div