From f903b5cd618dfeea7fca20632a911c673ecfc7b9 Mon Sep 17 00:00:00 2001
From: Hommy <16620803786@163.com>
Date: Sat, 31 Jan 2026 11:58:46 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9C=A8=E5=AF=BC=E5=87=BA=E8=8D=89=E7=A8=BF?=
 =?UTF-8?q?=E5=89=8D=EF=BC=8C=E6=B7=BB=E5=8A=A0=E8=8D=89=E7=A8=BF=E4=B8=8B?=
 =?UTF-8?q?=E8=BD=BD=E6=B5=81=E7=A8=8B=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/utils/video_task_manager.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/utils/video_task_manager.py b/src/utils/video_task_manager.py
index b2da86a..8e8ccd4 100644
--- a/src/utils/video_task_manager.py
+++ b/src/utils/video_task_manager.py
@@ -299,6 +299,21 @@ class VideoGenTaskManager:
             # 更新进度
             task.progress = 30
             
+            # 在导出视频前，先下载草稿
+            logger.info(f"Start downloading draft before export: {task.draft_url}")
+            from src.utils.draft_downloader import download_draft
+            download_success = download_draft(task.draft_url)
+            
+            if not download_success:
+                error_message = f"草稿下载失败: {task.draft_url}"
+                logger.error(error_message)
+                return "", error_message
+            
+            logger.info(f"Draft downloaded successfully: {task.draft_url}")
+            
+            # 更新进度
+            task.progress = 40
+            
             with UIAutomationInitializerInThread():
                 logger.info(f"Begin to export draft: {task.draft_id} -> {outfile}")