1c7bd9e2d0
- 在图片预览组件左上角添加变清晰和抠图按钮 - 创建图片处理加载组件（类似Gemini生图的动态卡片效果） - 实现点击按钮后的处理逻辑：创建临时消息、调用API、更新消息 - 添加消息保存API用于保存处理后的图片消息 - 当消息文本为'正在变清晰...'或'正在抠图...'时显示动态加载组件 - 处理完成后替换加载卡片为处理后的图片并保存到数据库
56 lines
1.9 KiB
TypeScript
56 lines
1.9 KiB
TypeScript
"use client";
|
|
|
|
import { type ComponentProps, memo } from "react";
|
|
import { Streamdown } from "streamdown";
|
|
import { cn } from "@/lib/utils";
|
|
import { ImagePreview } from "@/components/image-preview";
|
|
import { VideoPreview } from "@/components/video-preview";
|
|
|
|
type ResponseProps = ComponentProps<typeof Streamdown> & {
|
|
onImageEdit?: (imageUrl: string) => void;
|
|
onDrawingComplete?: (annotatedImageUrl: string) => void;
|
|
onImageEnhance?: (imageUrl: string) => void;
|
|
onImageRmbg?: (imageUrl: string) => void;
|
|
};
|
|
|
|
export const Response = memo(
|
|
({ className, onImageEdit, onDrawingComplete, onImageEnhance, onImageRmbg, ...props }: ResponseProps) => (
|
|
<Streamdown
|
|
className={cn(
|
|
"size-full [&>*:first-child]:mt-0 [&>*:last-child]:mb-0 [&_code]:whitespace-pre-wrap [&_code]:break-words [&_pre]:max-w-full [&_pre]:overflow-x-auto",
|
|
className
|
|
)}
|
|
components={{
|
|
img: ({ src, alt }: { src?: string; alt?: string }) => {
|
|
if (!src) {
|
|
return null;
|
|
}
|
|
return <ImagePreview
|
|
src={src}
|
|
alt={alt}
|
|
onImageEdit={onImageEdit}
|
|
onDrawingComplete={onDrawingComplete}
|
|
onImageEnhance={onImageEnhance}
|
|
onImageRmbg={onImageRmbg}
|
|
/>;
|
|
},
|
|
video: ({ src, alt }: { src?: string; alt?: string }) => {
|
|
if (!src) {
|
|
return null;
|
|
}
|
|
return <VideoPreview src={src} alt={alt} />;
|
|
},
|
|
}}
|
|
{...props}
|
|
/>
|
|
),
|
|
(prevProps: ResponseProps, nextProps: ResponseProps) =>
|
|
prevProps.children === nextProps.children &&
|
|
prevProps.onImageEdit === nextProps.onImageEdit &&
|
|
prevProps.onDrawingComplete === nextProps.onDrawingComplete &&
|
|
prevProps.onImageEnhance === nextProps.onImageEnhance &&
|
|
prevProps.onImageRmbg === nextProps.onImageRmbg
|
|
);
|
|
|
|
Response.displayName = "Response";
|