import { useEffect, useMemo, useRef, useState } from "react"; import type { ChangeEvent } from "react"; import { ArrowLeft, ArrowRight, Bookmark, Check, ChevronDown, Download, Grid2X2, ImagePlus, LayoutGrid, List, MoreHorizontal, Plus, Quote, RefreshCw, Search, Sparkles, Trash2, WandSparkles, X } from "lucide-react"; import type { AITask, Asset, ModelConfig, Product } from "../types"; import { MediaLightbox } from "../components/overlays"; import { Pager } from "../components/pager"; import type { Page } from "./route-config"; import { statusPill } from "./stage-config"; import "../ai-tools-page.css"; const TASK_TYPE_LABEL: Record = { model: "模特上身图", platform: "平台套图", image: "图片创作", model_photo: "模特上身图", platform_cover: "平台套图", image_optimize: "图片创作" }; const STATUS_LABEL: Record = { succeeded: "已完成", completed: "已完成", done: "已完成", ok: "已完成", skipped: "已跳过", failed: "失败", error: "失败", running: "生成中", queued: "排队中", polling: "生成中", needs_review: "待确认", // 后端 AITask.Status 全量中文化(原先缺这些会直接透出英文) created: "待处理", reserved: "排队中", submitted: "已提交", postprocessing: "处理中", compensating: "回滚中", cancelled: "已取消" }; function statusText(status: string) { return STATUS_LABEL[status] || status; } // 下载图片:优先 fetch→blob 触发真实下载;跨域失败则回退到新标签打开(用户仍拿到图) async function downloadImage(url: string, filename: string) { try { const res = await fetch(url, { mode: "cors" }); const blob = await res.blob(); const href = URL.createObjectURL(blob); const a = document.createElement("a"); a.href = href; a.download = filename; document.body.appendChild(a); a.click(); a.remove(); URL.revokeObjectURL(href); } catch { window.open(url, "_blank", "noopener"); } } export function AssetFactoryPage({ navigate, aiTasks, assets = [] }: { navigate: (page: Page) => void; aiTasks: AITask[]; assets?: Asset[] }) { // 任务 → 生成结果图:按 asset.origin_task 关联,取首张有预览 URL 的图片文件(脚本/视频任务无图则留占位) const taskImage = useMemo(() => { const map: Record = {}; for (const asset of assets) { const taskId = asset.origin_task; if (!taskId || map[taskId]) continue; const file = asset.files?.find((f) => f.preview_url && (f.content_type?.startsWith("image") ?? true)); if (file?.preview_url) map[taskId] = file.preview_url; } return map; }, [assets]); const cards = [ { page: "modelPhoto" as Page, tag: "[ MODEL · TRY-ON ]", title: "模特上身图", desc: "选择模特，AI 生成商品模特上身效果图", cost: "≈ ¥0.30 / 张" }, { page: "platformCover" as Page, tag: "[ PLATFORM · KIT ]", title: "平台套图", desc: "选择平台模板，AI 生成电商平台套图", cost: "≈ ¥0.50 / 张" }, { page: "imageOptimize" as Page, tag: "[ IMAGE · STUDIO ]", title: "图片创作", desc: "自由创作 AI 图片，适用于详情图 / 海报 / 灵感速写", cost: "≈ ¥0.40 / 组" } ]; const counts = useMemo(() => { const acc = { gen: 0, ok: 0, err: 0 }; for (const task of aiTasks) { const pill = statusPill(task.status); if (pill === "ok") acc.ok += 1; else if (pill === "err") acc.err += 1; else if (pill === "info") acc.gen += 1; } return acc; }, [aiTasks]); // 任务中心筛选:状态 tab / 搜索 / 时间 / 任务类型 / 网格·列表视图 —— 全部对真实 aiTasks 生效 const [filter, setFilter] = useState<"all" | "gen" | "ok" | "err">("all"); const [query, setQuery] = useState(""); const [timeFilter, setTimeFilter] = useState<"all" | "1" | "7" | "30">("all"); const [typeFilter, setTypeFilter] = useState(""); const [view, setView] = useState<"grid" | "list">("grid"); const [openChip, setOpenChip] = useState<"" | "time" | "type">(""); // 任务中心分页:每页 10 条,筛选/搜索变化时回第 1 页 const TASKS_PER_PAGE = 10; const [taskPage, setTaskPage] = useState(1); useEffect(() => { if (!openChip) return; const close = (event: MouseEvent) => { if (!(event.target as HTMLElement).closest(".chip-wrap")) setOpenChip(""); }; document.addEventListener("click", close); return () => document.removeEventListener("click", close); }, [openChip]); const typeOptions = Array.from(new Set(aiTasks.map((t) => t.task_type).filter(Boolean))); const TIME_OPTS: Array<{ value: typeof timeFilter; label: string }> = [ { value: "all", label: "全部时间" }, { value: "1", label: "今天" }, { value: "7", label: "近 7 天" }, { value: "30", label: "近 30 天" } ]; const visible = aiTasks.filter((task) => { const pill = statusPill(task.status); if (filter === "gen" && pill !== "info") return false; if (filter === "ok" && pill !== "ok") return false; if (filter === "err" && pill !== "err") return false; if (typeFilter && task.task_type !== typeFilter) return false; if (timeFilter !== "all" && task.created_at) { const days = (Date.now() - new Date(task.created_at).getTime()) / 86400000; if (days > Number(timeFilter)) return false; } if (query) { const hay = `${TASK_TYPE_LABEL[task.task_type] || task.task_type} ${task.task_type} ${task.id}`.toLowerCase(); if (!hay.includes(query.toLowerCase())) return false; } return true; }); // 筛选条件变化时回到第 1 页 useEffect(() => { setTaskPage(1); }, [filter, query, timeFilter, typeFilter]); const taskTotalPages = Math.max(1, Math.ceil(visible.length / TASKS_PER_PAGE)); const taskCurPage = Math.min(taskPage, taskTotalPages); const paged = visible.slice((taskCurPage - 1) * TASKS_PER_PAGE, taskCurPage * TASKS_PER_PAGE); return (

图片生成

// 一键生成 · 电商视觉素材，提升内容制作效率
{cards.map((card) => (
{card.tag}
{card.title}
{card.desc}
[ {card.cost} ]
))}

任务中心

// {aiTasks.length} 个 · {counts.gen} 生成中 · {counts.ok} 已完成 · {counts.err} 失败
{/* 状态 tabs(转写自 asset-factory.html #tc-tabs) */}
setFilter("all")}>全部 {aiTasks.length}
setFilter("gen")}>生成中 {counts.gen}
setFilter("ok")}>已完成 {counts.ok}
setFilter("err")}>失败 {counts.err}
setQuery(event.target.value)} />
{TIME_OPTS.map((opt) => (
{ setTimeFilter(opt.value); setOpenChip(""); }}> {opt.label}
))}
{ setTypeFilter(""); setOpenChip(""); }}> 全部类型
{typeOptions.length > 0 &&
} {typeOptions.map((t) => (
{ setTypeFilter(t); setOpenChip(""); }}> {TASK_TYPE_LABEL[t] || t}
))}
{(filter !== "all" || query || timeFilter !== "all" || typeFilter) && ( )}
// 显示 {paged.length} / {visible.length} 个任务
{aiTasks.length === 0 ? (
// NO TASKS YET
还没有任务，去上方选一个工序开始生成吧
) : visible.length === 0 ? (
// NO MATCH
没有符合筛选条件的任务
) : view === "grid" ? (
{paged.map((task) => { const pill = statusPill(task.status); const typeLabel = TASK_TYPE_LABEL[task.task_type] || task.task_type; const img = taskImage[task.id]; return (
{img ? {typeLabel} : {task.id.slice(0, 4)}}
{typeLabel}
// {task.task_type}
{(task.created_at || "").slice(0, 10)} {statusText(task.status)}
); })}
) : (
{paged.map((task) => { const pill = statusPill(task.status); const typeLabel = TASK_TYPE_LABEL[task.task_type] || task.task_type; const img = taskImage[task.id]; return ( ); })}
任务 进度 状态 创建于
{img ? {typeLabel} : {task.id.slice(0, 4)}}
{typeLabel}
// {task.task_type}
 {pill === "info" ? (
60%
) : ( {pill === "ok" ? "已完成" : "—"} )} 		{statusText(task.status)} {(task.created_at || "").slice(0, 10)}
)}
); } type WorkMode = "image" | "model" | "cover"; const MODE_META: Record< WorkMode, { title: string; tag: string; desc: string; ratio: string; promptTemplate: (productTitle: string) => string; } > = { image: { title: "图片创作", tag: "[ IMAGE · STUDIO ]", desc: "自由创作 AI 图片，适用于详情图 / 海报 / 灵感速写。", ratio: "1:1", promptTemplate: (title) => `${title}，电商高转化视觉，干净背景，商品主体清晰` }, model: { title: "模特上身图", tag: "[ MODEL · TRY-ON ]", desc: "选择模特和商品，生成电商模特上身图。", ratio: "1:1", promptTemplate: (title) => `${title}，模特上身展示，自然光，真实质感，电商主图` }, cover: { title: "平台套图", tag: "[ PLATFORM · KIT ]", desc: "选择平台模板，一键生成主图 / 封面 / 详情套图。", ratio: "4:5", promptTemplate: (title) => `${title}，电商平台套图，统一视觉，主图 + 详情排版` } }; const RATIO_OPTIONS = ["1:1", "3:4", "4:5", "9:16", "16:9"]; const COUNT_OPTIONS = ["1", "2", "4"]; const MODEL_RATIO_OPTIONS = ["1:1", "3:4", "9:16"]; const MODEL_COUNT_OPTIONS = ["4", "8", "12"]; const COVER_COUNT_OPTIONS = ["4", "8", "12"]; /* 图片创作 · 空态提示词建议 chip(基线 image-optimize EXAMPLES) */ const IMAGE_SUGGESTIONS = [ "一只穿着宇航服的橘猫，漂浮在霓虹色星云中，赛博朋克风", "极简北欧风格的茶杯，白底，自然柔光，产品摄影", "国风水墨海报，主体一只白鹤立于水边，留白构图", "电影感都市夜景，街道湿漉漉反射霓虹，4K 高清" ]; /* 图片创作 · 风格胶囊(基线 image-optimize STYLES) */ const STYLE_OPTIONS = [ { id: "auto", label: "默认" }, { id: "realistic", label: "写实" }, { id: "cinematic", label: "电影感" }, { id: "anime", label: "动漫" }, { id: "oil", label: "油画" }, { id: "cn-ink", label: "国风水墨" } ]; /* 模特上身图 · 真人模特默认占位卡(基线 model-photo Ava/Luna/Mia/Zoe) */ const FALLBACK_MODELS = [ { id: "m1", name: "Ava", tag: "亚洲·25岁·清新" }, { id: "m2", name: "Luna", tag: "亚洲·22岁·学生" }, { id: "m3", name: "Mia", tag: "混血·28岁·OL" }, { id: "m4", name: "Zoe", tag: "亚洲·30岁·健身" } ]; /* 平台套图 · 平台卡(基线 platform-cover · logo 配色用 token 化 className) */ const PLATFORM_OPTIONS = [ { id: "dy", name: "抖音电商", logo: "抖" }, { id: "tb", name: "淘宝", logo: "淘" }, { id: "tm", name: "天猫", logo: "猫" }, { id: "jd", name: "京东", logo: "京" }, { id: "pdd", name: "拼多多", logo: "拼" }, { id: "xhs", name: "小红书", logo: "红" }, { id: "ks", name: "快手", logo: "快" }, { id: "sph", name: "视频号", logo: "视" }, { id: "amz", name: "亚马逊", logo: "a" }, { id: "al", name: "1688", logo: "阿" } ]; export function ImageWorkbenchPage({ mode, products, assets, modelConfigs, onBack, navigate, onGenerate }: { mode: WorkMode; products: Product[]; assets: Asset[]; modelConfigs: ModelConfig[]; onBack: () => void; navigate?: (page: Page) => void; onGenerate: (payload: { prompt: string; mode?: "image" | "model" | "cover"; count?: number }) => Promise<{ assets: Asset[] } | null>; }) { const meta = MODE_META[mode]; const [productId, setProductId] = useState(products[0]?.id || ""); const product = products.find((item) => item.id === productId) || products[0]; const [prompt, setPrompt] = useState(meta.promptTemplate(products[0]?.title || "商品")); const [ratio, setRatio] = useState(meta.ratio); const [style, setStyle] = useState("auto"); const [count, setCount] = useState(mode === "image" ? "4" : "4"); const [pickedIds, setPickedIds] = useState([]); const [generating, setGenerating] = useState(false); const [results, setResults] = useState(null); const [refImage, setRefImage] = useState<{ name: string; url: string } | null>(null); const refInputRef = useRef(null); // 生成结果图片放大预览 const [preview, setPreview] = useState<{ src: string; name: string } | null>(null); // 模特/平台 工作台头部:搜索 + 时间排序 + 模特筛选(对左侧网格真实生效) const [gridQuery, setGridQuery] = useState(""); const [gridSort, setGridSort] = useState<"recent" | "name">("recent"); const [tbOpen, setTbOpen] = useState<"" | "time" | "model">(""); const [searchOpen, setSearchOpen] = useState(false); // 模特网格默认只露 6 个;「全部模特 →」展开后露全部,再点收起(超过 6 个时第 7 个之后本来选不到) const [showAllModels, setShowAllModels] = useState(false); function pickReference(event: ChangeEvent) { const file = event.target.files?.[0]; if (!file) return; setRefImage({ name: file.name, url: URL.createObjectURL(file) }); event.target.value = ""; } const imageModels = modelConfigs.filter((model) => model.capability.includes("image")); /* 模特卡数据来源:assets 中 category==='person' 的真资产(显真图 preview_url + name); 无则回退到基线占位模特卡 Ava/Luna/Mia/Zoe */ const personAssets = useMemo(() => assets.filter((item) => item.category === "person"), [assets]); useEffect(() => { if (product) setPrompt(meta.promptTemplate(product.title)); // mode 或商品切换都重置 prompt 与默认比例 setRatio(meta.ratio); // eslint-disable-next-line react-hooks/exhaustive-deps }, [productId, mode]); function togglePick(id: string) { setPickedIds((prev) => (prev.includes(id) ? prev.filter((item) => item !== id) : [...prev, id])); } const ratioVar = ratio.replace(":", " / "); const candidateCount = Math.max(1, Number(count) || 4); const canGenerate = prompt.trim().length > 0 && !generating; async function runGenerate() { if (!canGenerate) return; setGenerating(true); try { const result = await onGenerate({ prompt: prompt.trim(), mode, count: candidateCount }); // 成功:渲染真图;失败(null):保留原占位,不改 results if (result?.assets) setResults(result.assets); } finally { setGenerating(false); } } const hasResults = !!(results && results.length > 0); /* ── 共享:生成结果网格(§4.18 .gen-card · 三 mode 统一渲染真图)── */ function renderResultGrid() { const cols = (results?.length ?? candidateCount) >= 4 ? 4 : 2; return ( <> setPreview(null)} />
{(hasResults ? results!.map((asset, index) => ({ key: asset.id, index, url: asset.files?.[0]?.preview_url })) : Array.from({ length: candidateCount }).map((_, index) => ({ key: `ph-${index}`, index, url: undefined as string | undefined })) ).map(({ key, index, url }) => (
{url ? ( {`${meta.title} setPreview({ src: url, name: `${meta.title} #${index + 1}` })} /> ) : (
{generating ? "生成中…" : `${ratio} · #${index + 1}`}
)}
{url && ( )} {url && ( )}
))}
); } /* ════════════════════════════════════════════════ mode === "image" → 对话流形态(基线 image-optimize · design.md §4.13) 左会话列表 + 中央 hero/对话流 + 底部 chat 输入栏 ════════════════════════════════════════════════ */ if (mode === "image") { return (
{/* 左 · 会话列表 */} {/* 右 · 对话流 + 底部输入栏 */}
{hasResults ? (
{prompt}
{ratio} · {count} 张 · {STYLE_OPTIONS.find((s) => s.id === style)?.label || "默认"}
{renderResultGrid()}
) : (
// IMAGE · STUDIO

开始你的创作

输入想法、剧本或上传参考，和 Agent 一起把灵感变成电商视觉素材。

{IMAGE_SUGGESTIONS.map((text) => ( ))}
)}
{/* 底部 · chat 输入栏(textarea + 比例/风格/张数 + 发送) */}
{refImage && ( 参考图 {refImage.name.slice(0, 16)} )}