{status === 'running' && (
diff --git a/frontend/src/stores/generation.ts b/frontend/src/stores/generation.ts
index d7346d1..6686a2f 100755
--- a/frontend/src/stores/generation.ts
+++ b/frontend/src/stores/generation.ts
@@ -1,5 +1,5 @@
import { create } from 'zustand'
-import type { Asset, GenerateRequest } from '../api/types'
+import type { Asset, GenerateRequest, PipelineStage } from '../api/types'
import { submitGenerate, getTask, getAssets } from '../api/generate'
type Status = 'idle' | 'submitting' | 'running' | 'completed' | 'failed'
@@ -7,9 +7,12 @@ type Status = 'idle' | 'submitting' | 'running' | 'completed' | 'failed'
interface GenerationState {
taskId: string | null
projectId: string | null
+ stage: PipelineStage | null
progress: number
status: Status
statusText: string
+ retryCount: number
+ rejectReason: string | null
assets: Asset[]
error: string | null
submit: (req: GenerateRequest) => Promise
@@ -25,18 +28,21 @@ function stopPolling() {
}
}
-export const useGenerationStore = create((set, get) => ({
+export const useGenerationStore = create((set) => ({
taskId: null,
projectId: null,
+ stage: null,
progress: 0,
status: 'idle',
statusText: '',
+ retryCount: 0,
+ rejectReason: null,
assets: [],
error: null,
submit: async (req) => {
stopPolling()
- set({ status: 'submitting', error: null, statusText: '提交中...' })
+ set({ status: 'submitting', error: null, statusText: '提交中...', stage: null })
try {
const { taskId } = await submitGenerate(req)
@@ -44,24 +50,26 @@ export const useGenerationStore = create((set, get) => ({
taskId,
projectId: req.projectId,
status: 'running',
- progress: 10,
+ progress: 5,
+ stage: 'prompt_builder',
statusText: '任务已提交，等待生成...',
})
- // 开始轮询进度
pollTimer = setInterval(async () => {
try {
const task = await getTask(taskId)
- set({
- progress: task.progress ?? get().progress,
+ set((s) => ({
+ stage: task.stage ?? s.stage,
+ progress: task.progress ?? s.progress,
+ retryCount: task.retryCount ?? s.retryCount,
statusText:
task.status === 'running'
- ? '生成中...'
+ ? stageLabel(task.stage)
: task.status === 'pending'
? '排队中...'
: task.status,
- })
+ }))
if (task.status === 'completed') {
stopPolling()
@@ -71,6 +79,7 @@ export const useGenerationStore = create((set, get) => ({
set({
status: 'completed',
progress: 100,
+ stage: 'format_adapter',
statusText: '生成完成',
assets,
})
@@ -85,7 +94,7 @@ export const useGenerationStore = create((set, get) => ({
} catch {
// 网络错误不中断轮询
}
- }, 2000)
+ }, 1500)
} catch (err) {
stopPolling()
set({ status: 'failed', error: (err as Error).message, statusText: '提交失败' })
@@ -97,11 +106,24 @@ export const useGenerationStore = create((set, get) => ({
set({
taskId: null,
projectId: null,
+ stage: null,
progress: 0,
status: 'idle',
statusText: '',
+ retryCount: 0,
+ rejectReason: null,
assets: [],
error: null,
})
},
}))
+
+function stageLabel(stage?: string): string {
+ switch (stage) {
+ case 'prompt_builder': return '优化提示词...'
+ case 'asset_generator': return '生成素材中...'
+ case 'quality_supervisor': return '质检中...'
+ case 'format_adapter': return '格式转换中...'
+ default: return '生成中...'
+ }
+}
diff --git a/frontend/vite.config.ts b/frontend/vite.config.ts
index 7f74e10..2f4f4e8 100755
--- a/frontend/vite.config.ts
+++ b/frontend/vite.config.ts
@@ -14,6 +14,10 @@ export default defineConfig({
target: 'http://localhost:8080',
changeOrigin: true,
},
+ '/generation': {
+ target: 'http://localhost:8080',
+ changeOrigin: true,
+ },
},
},
})
From e51e1b6742f575931fc6b2e42d8701f580bc9260 Mon Sep 17 00:00:00 2001
From: Gmaker689 <1711322114@qq.com>
Date: Mon, 25 May 2026 14:29:02 +0800
Subject: [PATCH 3/3] =?UTF-8?q?feat:=20.env=20example=E7=A4=BA=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
backend/.env.example | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/backend/.env.example b/backend/.env.example
index f9a7524..28a254c 100755
--- a/backend/.env.example
+++ b/backend/.env.example
@@ -20,9 +20,10 @@ GEN2D_LLM_TEMPERATURE=0.7
GEN2D_LLM_MAX_TOKENS=2048
# 文生图模型（OpenAI 兼容 Images API）
-GEN2D_IMAGE_BASE_URL=https://api.suchuang.vip/v1
-GEN2D_IMAGE_API_KEY=
-GEN2D_IMAGE_MODEL=gpt-image-1
+# 模型可选: dall-e-2, dall-e-3, gpt-image-1, gpt-image-2
+GEN2D_IMAGE_BASE_URL=https://api.weelinking.com/v1
+GEN2D_IMAGE_API_KEY=sk-your-api-key
+GEN2D_IMAGE_MODEL=gpt-image-2
GEN2D_IMAGE_WIDTH=1024
GEN2D_IMAGE_HEIGHT=1024
GEN2D_IMAGE_QUALITY=low