Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
-import auto_editor
 import os
-import shutil
 import subprocess
+import sys
 
 def edit_video(input_video, margin=0.2, min_clip=2, min_cut=2):
     """
@@ -15,7 +14,7 @@ def edit_video(input_video, margin=0.2, min_clip=2, min_cut=2):
     - min_cut: 最小移除片段長度（秒）
     
     返回：
-    - 處理後的影片路徑
+    - (output_video_path, status_message): 處理後的影片路徑和狀態訊息
     """
     try:
         # 確保輸出目錄存在
@@ -25,28 +24,37 @@ def edit_video(input_video, margin=0.2, min_clip=2, min_cut=2):
         # 定義輸出檔案路徑
         output_path = os.path.join(output_dir, "edited_video.mp4")
         
-        # 確保 Auto-Editor 和 FFmpeg 可執行
-        subprocess.run(["ffmpeg", "-version"], check=True)
+        # 檢查 FFmpeg 是否可用
+        result = subprocess.run(["ffmpeg", "-version"], capture_output=True, text=True)
+        if result.returncode != 0:
+            return None, "錯誤：FFmpeg 未正確安裝，請檢查環境配置。"
         
-        # 執行 Auto-Editor，傳遞參數
+        # 使用 Auto-Editor 的 CLI 命令
         auto_editor_args = [
+            "auto-editor",
             input_video,
             "--output_file", output_path,
-            "--margin", str(margin),
-            "--min_clip_length", str(min_clip),
-            "--min_cut_length", str(min_cut),
-            "--no_open"  # 避免打開檔案
+            "--margin", f"{margin}s",
+            "--min_clip_length", f"{min_clip}s",
+            "--min_cut_length", f"{min_cut}s",
+            "--no_open"
         ]
-        auto_editor.main(auto_editor_args)
+        
+        # 執行 Auto-Editor
+        result = subprocess.run(auto_editor_args, capture_output=True, text=True)
+        
+        # 檢查命令是否成功
+        if result.returncode != 0:
+            return None, f"錯誤：Auto-Editor 執行失敗 - {result.stderr}"
         
         # 檢查輸出檔案是否存在
         if not os.path.exists(output_path):
-            raise Exception("Auto-Editor 未能生成輸出檔案")
+            return None, "錯誤：Auto-Editor 未能生成輸出檔案。"
         
-        return output_path
+        return output_path, "成功：影片已處理完成！"
     
     except Exception as e:
-        return f"錯誤：{str(e)}"
+        return None, f"錯誤：{str(e)}"
 
 # 定義 Gradio 介面
 def create_gradio_interface():