Update app.py

app.py

@@ -1,23 +1,61 @@
 import gradio as gr
-import autoeditor
+import subprocess
 import os
+import tempfile
+import shutil
 
 def edit_video(input_video):
     if not input_video:
-        return "請上傳影片檔案。"
+        return "請上傳影片檔案。", None
     
-    output_path = "edited_video.mp4"
-    # 修正為 --output_file（下劃線）或 -o
-    autoeditor.main([input_video, "--edit", "audio", "--margin", "0.2sec", "--output_file", output_path])
-    
-    return output_path  # 返回可下載的檔案
+    try:
+        # 創建臨時輸出檔案
+        with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as tmp_file:
+            output_path = tmp_file.name
+        
+        # 使用 subprocess 調用 auto-editor 命令
+        cmd = [
+            'auto-editor', 
+            input_video,
+            '--margin', '0.2sec',
+            '--output', output_path
+        ]
+        
+        # 執行命令
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=300)
+        
+        if result.returncode == 0:
+            return "編輯完成！", output_path
+        else:
+            return f"錯誤：{result.stderr}", None
+            
+    except subprocess.TimeoutExpired:
+        return "處理超時（5分鐘限制）", None
+    except Exception as e:
+        return f"處理錯誤：{str(e)}", None
 
-iface = gr.Interface(
-    fn=edit_video,
-    inputs=gr.Video(label="上傳影片或音訊檔案"),
-    outputs=gr.File(label="下載編輯後的檔案"),
-    title="Auto-Editor Demo on CPU",
-    description="上傳影片，自動移除靜音部分。處理時間視檔案大小而定（CPU 環境下可能需數分鐘）。"
-)
+# 創建 Gradio 介面
+with gr.Blocks(title="Auto-Editor Demo") as iface:
+    gr.Markdown("# Auto-Editor 影片編輯工具")
+    gr.Markdown("上傳影片，自動移除靜音部分。處理時間視檔案大小而定。")
+    
+    with gr.Row():
+        with gr.Column():
+            input_video = gr.File(
+                label="上傳影片檔案", 
+                file_types=['video']
+            )
+            edit_btn = gr.Button("開始編輯", variant="primary")
+        
+        with gr.Column():
+            status_text = gr.Textbox(label="狀態")
+            output_file = gr.File(label="下載編輯後的檔案")
+    
+    edit_btn.click(
+        fn=edit_video,
+        inputs=[input_video],
+        outputs=[status_text, output_file]
+    )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()