Update app.py

app.py

@@ -1,155 +1,222 @@
 import gradio as gr
 import os
 import tempfile
-from pathlib import Path
 import subprocess
 import shutil
+from pathlib import Path
+import sys
 
 def setup_environment():
-    """設置環境和必要的系統包"""
+    """設置環境"""
     try:
-        # 安裝FFmpeg (如果需要)
-        subprocess.run(["apt", "update"], check=True, capture_output=True)
-        subprocess.run(["apt", "install", "-y", "ffmpeg"], check=True, capture_output=True)
-    except:
-        pass
+        # 確保系統包已安裝
+        subprocess.run(["which", "ffmpeg"], check=True, capture_output=True)
+        print("✅ FFmpeg已安裝")
+    except subprocess.CalledProcessError:
+        print("⚠️ FFmpeg未找到，但Docker鏡像應該已包含")
 
-def download_and_clone_repo():
-    """克隆GitHub倉庫"""
+def clone_repo():
+    """克隆項目倉庫"""
     repo_url = "https://github.com/SamurAIGPT/AI-Youtube-Shorts-Generator.git"
     repo_dir = "./AI-Youtube-Shorts-Generator"
     
-    if not os.path.exists(repo_dir):
-        subprocess.run(["git", "clone", repo_url], check=True)
+    if os.path.exists(repo_dir):
+        shutil.rmtree(repo_dir)
     
-    return repo_dir
+    try:
+        subprocess.run(["git", "clone", repo_url, repo_dir], 
+                      check=True, capture_output=True, text=True)
+        return repo_dir, "✅ 項目克隆成功"
+    except subprocess.CalledProcessError as e:
+        return None, f"❌ 克隆失敗: {e.stderr}"
+
+def install_project_deps(repo_dir):
+    """安裝項目依賴"""
+    try:
+        # 進入項目目錄安裝依賴
+        original_dir = os.getcwd()
+        os.chdir(repo_dir)
+        
+        # 檢查是否有requirements.txt
+        if os.path.exists("requirements.txt"):
+            result = subprocess.run([
+                sys.executable, "-m", "pip", "install", "-r", "requirements.txt"
+            ], capture_output=True, text=True, timeout=180)
+            
+            if result.returncode != 0:
+                os.chdir(original_dir)
+                return False, f"依賴安裝失敗: {result.stderr}"
+        
+        os.chdir(original_dir)
+        return True, "✅ 項目依賴安裝成功"
+        
+    except Exception as e:
+        if 'original_dir' in locals():
+            os.chdir(original_dir)
+        return False, f"安裝過程出錯: {str(e)}"
 
-def process_youtube_video(youtube_url, openai_api_key, progress=gr.Progress()):
-    """處理YouTube視頻生成Shorts"""
+def process_video(youtube_url, openai_api_key, progress=gr.Progress()):
+    """處理YouTube視頻"""
+    
+    if not youtube_url.strip():
+        return None, "❌ 請輸入YouTube URL"
     
-    if not youtube_url or not openai_api_key:
-        return None, "請提供YouTube URL和OpenAI API Key"
+    if not openai_api_key.strip():
+        return None, "❌ 請輸入OpenAI API Key"
+    
+    # 驗證URL格式
+    if not any(domain in youtube_url for domain in ['youtube.com', 'youtu.be']):
+        return None, "❌ 請輸入有效的YouTube URL"
     
     progress(0.1, desc="初始化環境...")
     setup_environment()
     
     progress(0.2, desc="克隆項目代碼...")
-    repo_dir = download_and_clone_repo()
+    repo_dir, clone_msg = clone_repo()
+    if not repo_dir:
+        return None, clone_msg
+    
+    progress(0.3, desc="安裝項目依賴...")
+    deps_ok, deps_msg = install_project_deps(repo_dir)
+    if not deps_ok:
+        return None, deps_msg
     
     # 設置環境變量
     env = os.environ.copy()
-    env["OPENAI_API"] = openai_api_key
+    env["OPENAI_API_KEY"] = openai_api_key
     
     try:
-        progress(0.3, desc="開始處理視頻...")
+        progress(0.5, desc="處理視頻中...")
         
-        # 切換到項目目錄
         original_dir = os.getcwd()
         os.chdir(repo_dir)
         
-        # 安裝項目依賴
-        subprocess.run(["pip", "install", "-r", "requirements.txt"], 
-                      check=True, env=env)
-        
-        progress(0.5, desc="下載和分析視頻...")
-        
-        # 創建臨時輸入文件來模擬用戶輸入
+        # 創建輸入文件
         with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
-            f.write(youtube_url + '\n')
+            f.write(f"{youtube_url}\n")
             input_file = f.name
         
-        # 運行主程序
+        # 執行主程序
         with open(input_file, 'r') as f:
             result = subprocess.run(
-                ["python", "main.py"], 
+                [sys.executable, "main.py"],
                 stdin=f,
-                capture_output=True, 
-                text=True, 
+                capture_output=True,
+                text=True,
                 env=env,
-                timeout=300  # 5分鐘超時
+                timeout=600  # 10分鐘超時
             )
         
-        progress(0.8, desc="生成Shorts視頻...")
-        
-        # 清理臨時文件
+        # 清理輸入文件
         os.unlink(input_file)
         
-        # 查找生成的視頻文件
-        output_files = []
-        for ext in ['*.mp4', '*.mov', '*.avi']:
-            output_files.extend(Path('.').glob(f"**/{ext}"))
+        progress(0.8, desc="查找生成的視頻...")
+        
+        # 查找輸出視頻
+        video_files = []
+        for pattern in ['*.mp4', '*.avi', '*.mov']:
+            video_files.extend(Path('.').rglob(pattern))
         
         os.chdir(original_dir)
         
         progress(1.0, desc="處理完成!")
         
-        if output_files:
-            # 返回第一個找到的視頻文件
-            video_path = str(output_files[0])
-            return video_path, f"成功生成 {len(output_files)} 個Shorts視頻!"
+        if video_files:
+            # 復制視頻到當前目錄以便訪問
+            output_video = str(video_files[0])
+            local_video = f"./output_{os.path.basename(output_video)}"
+            shutil.copy2(os.path.join(repo_dir, output_video), local_video)
+            
+            return local_video, f"✅ 成功生成視頻! 找到 {len(video_files)} 個文件。"
         else:
-            return None, f"處理完成但未找到輸出視頻。\n控制台輸出：\n{result.stdout}\n錯誤：\n{result.stderr}"
+            error_msg = f"處理完成但未找到視頻文件。\n\n標準輸出:\n{result.stdout}\n\n錯誤輸出:\n{result.stderr}"
+            return None, error_msg
             
     except subprocess.TimeoutExpired:
-        return None, "處理超時。請嘗試較短的視頻或檢查網絡連接。"
-    except Exception as e:
         os.chdir(original_dir)
-        return None, f"處理過程中出現錯誤：{str(e)}"
+        return None, "❌ 處理超時 (10分鐘)。請嘗試較短的視頻。"
+    except Exception as e:
+        if 'original_dir' in locals():
+            os.chdir(original_dir)
+        return None, f"❌ 處理失敗: {str(e)}"
 
 # 創建Gradio界面
-with gr.Blocks(title="AI YouTube Shorts Generator") as demo:
+with gr.Blocks(title="AI YouTube Shorts Generator", theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
     # 🎬 AI YouTube Shorts Generator
     
-    將長視頻自動轉換為吸引人的YouTube Shorts！
+    自動將長視頻轉換為吸引人的YouTube Shorts！
     
-    ## 功能特點：
-    - 🎯 自動提取視頻亮點
-    - 🗣️ 智能說話者檢測  
-    - ✂️ 垂直格式裁剪
-    - 🤖 基於GPT-4的內容分析
+    ## ✨ 功能特點
+    - 🎯 **智能內容分析**: 使用GPT-4識別視頻亮點
+    - 🗣️ **說話者檢測**: 自動檢測不同說話者
+    - ✂️ **垂直裁剪**: 轉換為Shorts格式 (9:16)
+    - 🎵 **語音轉錄**: 基於Whisper的高質量轉錄
     """)
     
     with gr.Row():
         with gr.Column(scale=1):
             youtube_url = gr.Textbox(
-                label="YouTube視頻URL", 
-                placeholder="https://www.youtube.com/watch?v=...",
-                info="請輸入要處理的YouTube視頻鏈接"
+                label="📺 YouTube視頻URL", 
+                placeholder="https://www.youtube.com/watch?v=dQw4w9WgXcQ",
+                info="輸入要處理的YouTube視頻鏈接"
             )
             
             openai_key = gr.Textbox(
-                label="OpenAI API Key", 
+                label="🔑 OpenAI API Key", 
                 type="password",
-                placeholder="sk-...",
-                info="需要GPT-4訪問權限的API Key"
+                placeholder="sk-proj-...",
+                info="需要GPT-4/GPT-4o訪問權限"
             )
             
-            process_btn = gr.Button("🚀 生成Shorts", variant="primary")
+            with gr.Row():
+                process_btn = gr.Button("🚀 開始生成", variant="primary", size="lg")
+                clear_btn = gr.Button("🗑️ 清除", variant="secondary")
             
         with gr.Column(scale=1):
-            output_video = gr.Video(label="生成的Shorts視頻")
-            output_message = gr.Textbox(label="處理狀態", lines=5)
+            output_video = gr.Video(label="🎥 生成的Shorts")
+            output_status = gr.Textbox(
+                label="📊 處理狀態", 
+                lines=8, 
+                max_lines=10,
+                show_copy_button=True
+            )
     
     gr.Markdown("""
-    ## 使用說明：
-    1. 輸入YouTube視頻URL
-    2. 提供OpenAI API Key (需要GPT-4權限)
-    3. 點擊生成按鈕等待處理完成
-    4. 下載生成的Shorts視頻
-    
-    ## 注意事項：
-    - 處理時間取決於視頻長度，通常需要2-5分鐘
-    - 確保YouTube視頻可以公開訪問
-    - API Key將用於內容分析，不會被保存
+    ## 📋 使用說明
+    
+    1. **輸入URL**: 粘貼YouTube視頻鏈接
+    2. **API Key**: 提供OpenAI API密鑰 (需GPT-4權限)
+    3. **開始處理**: 點擊生成按鈕，等待5-10分鐘
+    4. **下載視頻**: 處理完成後下載生成的Shorts
+    
+    ## ⚠️ 注意事項
+    
+    - **處理時間**: 取決於視頻長度，通常5-10分鐘
+    - **視頻要求**: 確保YouTube視頻公開可訪問
+    - **API用量**: 會消耗OpenAI API額度用於內容分析
+    - **格式支持**: 輸出為MP4格式的豎向視頻 (9:16)
+    
+    ---
+    💡 **提示**: 較短的源視頻 (10-30分鐘) 處理效果更佳
     """)
     
+    # 事件綁定
     process_btn.click(
-        fn=process_youtube_video,
+        fn=process_video,
         inputs=[youtube_url, openai_key],
-        outputs=[output_video, output_message],
+        outputs=[output_video, output_status],
         show_progress=True
     )
+    
+    clear_btn.click(
+        lambda: ("", "", None, ""),
+        outputs=[youtube_url, openai_key, output_video, output_status]
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True
+    )