From 349a362f27571f25b84051f83a0935de5010272e Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Mon, 29 Dec 2025 08:09:42 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=9C=A8=E7=9B=B8=E5=86=8C=E4=B8=AD?=
 =?UTF-8?q?=E4=B8=BA=E5=8F=98=E6=B8=85=E6=99=B0=E5=90=8E=E7=9A=84=E5=9B=BE?=
 =?UTF-8?q?=E7=89=87=E6=B7=BB=E5=8A=A0=E5=8F=B3=E4=B8=8A=E8=A7=92=E5=9B=BE?=
 =?UTF-8?q?=E6=A0=87=E6=A0=87=E8=AF=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/(chat)/gallery/page.tsx | 7 +++++++
 app/api/gallery/route.ts    | 2 ++
 2 files changed, 9 insertions(+)

diff --git a/app/(chat)/gallery/page.tsx b/app/(chat)/gallery/page.tsx
index d70bd7d..c618e4b 100644
--- a/app/(chat)/gallery/page.tsx
+++ b/app/(chat)/gallery/page.tsx
@@ -18,6 +18,7 @@ type GalleryImage = {
   imageSize?: string;  // 图片尺寸（如 "2K", "4K"）
   createdAt: string;
   chatId: string;
+  modelId?: string;  // 模型ID，用于标识图片类型（如 "enhance-1.0" 表示变清晰后的图片）
 };
 
 type Pagination = {
@@ -287,6 +288,12 @@ export default function GalleryPage() {
                       placeholder="blur"
                       blurDataURL="data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMjAwIiBoZWlnaHQ9IjI3MCIgeG1sbnM9Imh0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnIj48cmVjdCB3aWR0aD0iMjAwIiBoZWlnaHQ9IjI3MCIgZmlsbD0iI2VlZSIvPjwvc3ZnPg=="
                     />
+                    {/* 变清晰标识图标 */}
+                    {img.modelId === "enhance-1.0" && (
+                      <div className="absolute top-2 left-2 bg-blue-500/90 text-white rounded-full p-1.5 shadow-lg" title="变清晰后的图片">
+                        <Sparkles className="h-4 w-4" />
+                      </div>
+                    )}
                   </div>
                   {/* 🆕 操作按钮（悬停时显示） */}
                   <div className="absolute top-2 right-2 opacity-0 group-hover:opacity-100 transition-opacity flex gap-2">
diff --git a/app/api/gallery/route.ts b/app/api/gallery/route.ts
index d5251c9..88ccadb 100644
--- a/app/api/gallery/route.ts
+++ b/app/api/gallery/route.ts
@@ -39,6 +39,7 @@ export async function GET(request: NextRequest) {
         imageSize: generatedImage.imageSize,
         createdAt: generatedImage.createdAt,
         chatId: generatedImage.chatId,
+        modelId: generatedImage.modelId,  // 添加 modelId 字段
       })
       .from(generatedImage)
       .where(eq(generatedImage.userId, userId))
@@ -56,6 +57,7 @@ export async function GET(request: NextRequest) {
       imageSize: img.imageSize,
       createdAt: img.createdAt,
       chatId: img.chatId,
+      modelId: img.modelId,  // 添加 modelId 字段
     }));
 
     const totalPages = Math.ceil(total / pageSize);