From f22af3df3a31b29bbba9e3c4760b4b34c9f1a49d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=8C=97=E6=9E=B3?=
 <7854742+wang_rumeng@user.noreply.gitee.com>
Date: Thu, 9 Oct 2025 17:28:56 +0800
Subject: [PATCH] watermark_videos

---
 .../pages/work-flow/use-workflow-data.tsx     | 36 ++++++++++---------
 1 file changed, 19 insertions(+), 17 deletions(-)

diff --git a/components/pages/work-flow/use-workflow-data.tsx b/components/pages/work-flow/use-workflow-data.tsx
index 6921cf6..cbfc550 100644
--- a/components/pages/work-flow/use-workflow-data.tsx
+++ b/components/pages/work-flow/use-workflow-data.tsx
@@ -439,35 +439,37 @@ export function useWorkflowData({ onEditPlanGenerated, editingStatus, onExportFa
           }
         }
 
-        // 粗剪
-        if (task.task_name === 'generate_final_simple_video') {
-          if (task.task_result && task.task_result.video) {
-            taskCurrent.currentStage = 'final_video';
-            taskCurrent.final.url = task.task_result.video;
-            taskCurrent.final.note = 'simple';
-            taskCurrent.status = 'COMPLETED';
-          }
-        }
-
-        // generate_export_video
-        if (task.task_name === 'generate_export_video' || task.task_name === 'combiner_videos') {
+        // 合成视频
+        if (task.task_name === 'combiner_videos') {
           if (task.task_status === 'COMPLETED') {
             taskCurrent.currentStage = 'final_video';
             taskCurrent.final.url = task.task_result.video;
-            taskCurrent.final.note = task.task_name === 'generate_export_video' ? 'export' : 'combiner';
+            taskCurrent.final.note = 'combiner';
             taskCurrent.status = 'COMPLETED';
-            // 停止轮询
-            setNeedStreamData(false);
           }
           if (task.task_status === 'FAILED' || task.task_status === 'ERROR') {
-            console.log('----------视频导出失败');
             taskCurrent.status = 'FAILED';
             
             // 触发导出失败回调
             if (onExportFailed) {
               onExportFailed();
             }
-            
+            // 停止轮询
+            setNeedStreamData(false);
+          }
+        }
+
+        // 水印视频
+        if (task.task_name === 'watermark_videos') {
+          if (task.task_status === 'COMPLETED') {
+            taskCurrent.currentStage = 'final_video';
+            taskCurrent.final.url = task.task_result.video;
+            taskCurrent.final.note = 'watermark';
+            taskCurrent.status = 'COMPLETED';
+            // 停止轮询
+            setNeedStreamData(false);
+          }
+          if (task.task_status === 'FAILED' || task.task_status === 'ERROR') {
             // 停止轮询
             setNeedStreamData(false);
           }