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);