From 2820ce0c4db8b4e5ca68b90f0b0079a2d9274a2e 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: Wed, 3 Sep 2025 19:10:25 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E4=B8=8B=E8=BD=BD=EF=BC=9A?=
 =?UTF-8?q?=E5=88=86=E9=95=9C=E4=B8=8B=E8=BD=BD=E3=80=81=E6=9C=80=E7=BB=88?=
 =?UTF-8?q?=E8=A7=86=E9=A2=91=E4=B8=8B=E8=BD=BD=E3=80=81=E5=88=97=E8=A1=A8?=
 =?UTF-8?q?=E4=B8=AD=E8=A7=86=E9=A2=91=E4=B8=8B=E8=BD=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 components/pages/create-to-video2.tsx         | 21 ++++++++++++++++---
 components/pages/work-flow/media-viewer.tsx   | 16 +++++++-------
 .../pages/work-flow/use-workflow-data.tsx     |  2 +-
 utils/tools.ts                                | 17 +++++++++++++++
 4 files changed, 43 insertions(+), 13 deletions(-)

diff --git a/components/pages/create-to-video2.tsx b/components/pages/create-to-video2.tsx
index 0a1fe47..7f4da38 100644
--- a/components/pages/create-to-video2.tsx
+++ b/components/pages/create-to-video2.tsx
@@ -1,7 +1,7 @@
 "use client";
 
 import { useState, useEffect, useRef, useCallback } from 'react';
-import { Loader2, Check, Pencil, Trash } from 'lucide-react';
+import { Loader2, Check, Pencil, Trash, Download } from 'lucide-react';
 import { useRouter } from 'next/navigation';
 import './style/create-to-video2.css';
 
@@ -10,6 +10,9 @@ import { ChatInputBox } from '@/components/ChatInputBox/ChatInputBox';
 import cover_image1 from '@/public/assets/cover_image3.jpg';
 import { motion } from 'framer-motion';
 import { Tooltip, Button } from 'antd';
+import { GlassIconButton } from '@/components/ui/glass-icon-button';
+import { downloadVideo } from '@/utils/tools';
+
 
 
 // ideaText已迁移到ChatInputBox组件中
@@ -217,6 +220,18 @@ export default function CreateToVideo2() {
             />
           )}
 
+          {/* 下载按钮 右上角 */}
+          {(project.final_video_url || project.final_simple_video_url) && (
+            <div className="absolute top-1 right-1">
+              <Tooltip placement="top" title="Download">
+                <Button size="small" type="text" className="w-[2.5rem] h-[2.5rem] rounded-full items-center justify-center p-0 hidden group-hover:flex transition-all duration-300 hover:bg-white/15" onClick={() => {
+                  downloadVideo(project.final_video_url || project.final_simple_video_url);
+                }}><Download className="w-4 h-4 text-white" /></Button>
+              </Tooltip>
+            </div>
+          )}
+          
+
           {/* 状态标签 - 左上角 */}
           <div className="absolute top-3 left-3">
             {StatusBadge((project.status === 'COMPLETED' || project.final_simple_video_url) ? 'completed' : project.status === 'FAILED' ? 'failed' : 'pending')}
@@ -232,12 +247,12 @@ export default function CreateToVideo2() {
               </h2>
               {/* TODO 编辑标题 */}
               {/* <Tooltip title="Edit Title">
-                <Button size="small" type="text" className="p-0 hidden group-hover:block transition-all duration-300"><Pencil className="w-4 h-4 text-white" /></Button>
+                <Button size="small" type="text" className="w-[2.5rem] h-[2.5rem] rounded-full items-center justify-center p-0 hidden group-hover:flex transition-all duration-300 hover:bg-white/15" /></Button>
               </Tooltip> */}
             </div>
             {/* TODO 删除 */}
             {/* <Tooltip title="Delete">
-              <Button size="small" type="text" className="p-0 hidden group-hover:block transition-all duration-300"><Trash className="w-4 h-4 text-white" /></Button>
+              <Button size="small" type="text" className="w-[2.5rem] h-[2.5rem] rounded-full items-center justify-center p-0 hidden group-hover:flex transition-all duration-300 hover:bg-white/15" /></Button>
             </Tooltip> */}
           </div>
         </div>
diff --git a/components/pages/work-flow/media-viewer.tsx b/components/pages/work-flow/media-viewer.tsx
index d005b41..c009940 100644
--- a/components/pages/work-flow/media-viewer.tsx
+++ b/components/pages/work-flow/media-viewer.tsx
@@ -10,7 +10,7 @@ import { mockScriptData } from '@/components/script-renderer/mock';
 import { Skeleton } from '@/components/ui/skeleton';
 import { TaskObject } from '@/api/DTO/movieEdit';
 import { Button, Tooltip } from 'antd';
-import { Video } from 'lucide-react';
+import { downloadVideo } from '@/utils/tools';
 
 interface MediaViewerProps {
   taskObject: TaskObject;
@@ -273,14 +273,6 @@ export const MediaViewer = React.memo(function MediaViewer({
     };
   }, []);
 
-  // 下载视频
-  const downloadVideo = (url: string) => {
-    const a = document.createElement('a');
-    a.href = url;
-    a.download = url.split('/').pop() || '';
-    a.click();
-  };
-
   // 渲染音量控制组件
   const renderVolumeControls = () => (
     <div className="flex items-center gap-2">
@@ -371,6 +363,12 @@ export const MediaViewer = React.memo(function MediaViewer({
                 onClick={() => handleEditClick('3', 'final')}
               />
             </Tooltip>
+            {/* 下载按钮 */}
+            <Tooltip placement="top" title="Download video">
+              <GlassIconButton icon={Download} size='sm' onClick={() => {
+                downloadVideo(taskObject.final.url);
+              }} />
+            </Tooltip>
             {showGotoCutButton && (
               <Tooltip placement="top" title='Go to AI-powered editing platform'>
                 <GlassIconButton icon={Scissors} size='sm' onClick={onGotoCut} />
diff --git a/components/pages/work-flow/use-workflow-data.tsx b/components/pages/work-flow/use-workflow-data.tsx
index 98310c9..0305917 100644
--- a/components/pages/work-flow/use-workflow-data.tsx
+++ b/components/pages/work-flow/use-workflow-data.tsx
@@ -397,7 +397,7 @@ export function useWorkflowData() {
 
       const { current: taskCurrent } = tempTaskObject;
 
-      taskCurrent.title = title || 'generating...';
+      taskCurrent.title = name || 'generating...';
       taskCurrent.tags = tags || [];
       taskCurrent.status = status as Status; 
 
diff --git a/utils/tools.ts b/utils/tools.ts
index 444a947..7a79fd3 100644
--- a/utils/tools.ts
+++ b/utils/tools.ts
@@ -77,3 +77,20 @@ export function createScreenAdapter(): void {
   // 将样式添加到head
   document.head.appendChild(style);
 }
+
+export const downloadVideo = async (url: string) => {
+  try {
+    const response = await fetch(url);
+    const blob = await response.blob();
+    const blobUrl = window.URL.createObjectURL(blob);
+    const a = document.createElement('a');
+    a.href = blobUrl;
+    a.download = url.split('/').pop() || 'video.mp4';
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+    window.URL.revokeObjectURL(blobUrl);
+  } catch (error) {
+    console.error('下载视频失败:', error);
+  }
+};
\ No newline at end of file