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" && (
+
+
+
+ )}
{/* 🆕 操作按钮(悬停时显示) */}
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);