Update app.py

app.py

@@ -1,89 +1,23 @@
 import gradio as gr
+import autoeditor
 import os
-import subprocess
-import sys
 
-def edit_video(input_video, margin=0.2, min_clip=2, min_cut=2):
-    """
-    使用 Auto-Editor 編輯影片，移除靜音部分並生成輸出。
+def edit_video(input_video):
+    if not input_video:
+        return "請上傳影片檔案。"
     
-    參數：
-    - input_video: 上傳的影片檔案路徑
-    - margin: 靜音檢測的邊距（秒）
-    - min_clip: 最小保留片段長度（秒）
-    - min_cut: 最小移除片段長度（秒）
+    output_path = "edited_video.mp4"
+    # 修正為 --output_file（下劃線）或 -o
+    autoeditor.main([input_video, "--edit", "audio", "--margin", "0.2sec", "--output_file", output_path])
     
-    返回：
-    - (output_video_path, status_message): 處理後的影片路徑和狀態訊息
-    """
-    try:
-        # 確保輸出目錄存在
-        output_dir = "outputs"
-        os.makedirs(output_dir, exist_ok=True)
-        
-        # 定義輸出檔案路徑
-        output_path = os.path.join(output_dir, "edited_video.mp4")
-        
-        # 檢查 FFmpeg 是否可用
-        result = subprocess.run(["ffmpeg", "-version"], capture_output=True, text=True)
-        if result.returncode != 0:
-            return None, "錯誤：FFmpeg 未正確安裝，請檢查環境配置。"
-        
-        # 使用 Auto-Editor 的 CLI 命令
-        auto_editor_args = [
-            "auto-editor",
-            input_video,
-            "--output_file", output_path,
-            "--margin", f"{margin}s",
-            "--min_clip_length", f"{min_clip}s",
-            "--min_cut_length", f"{min_cut}s",
-            "--no_open"
-        ]
-        
-        # 執行 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):
-            return None, "錯誤：Auto-Editor 未能生成輸出檔案。"
-        
-        return output_path, "成功：影片已處理完成！"
-    
-    except Exception as e:
-        return None, f"錯誤：{str(e)}"
+    return output_path  # 返回可下載的檔案
 
-# 定義 Gradio 介面
-def create_gradio_interface():
-    with gr.Blocks() as interface:
-        gr.Markdown("# Auto-Editor on Hugging Face Spaces")
-        gr.Markdown("上傳影片，自動移除靜音部分並生成編輯後的影片。")
-        
-        with gr.Row():
-            with gr.Column():
-                input_video = gr.Video(label="上傳影片")
-                margin = gr.Slider(minimum=0, maximum=1, value=0.2, step=0.01, label="靜音邊距 (秒)")
-                min_clip = gr.Slider(minimum=0, maximum=10, value=2, step=0.1, label="最小保留片段 (秒)")
-                min_cut = gr.Slider(minimum=0, maximum=10, value=2, step=0.1, label="最小移除片段 (秒)")
-                submit_button = gr.Button("開始編輯")
-            
-            with gr.Column():
-                output_video = gr.Video(label="編輯後的影片")
-                output_text = gr.Textbox(label="狀態訊息")
-        
-        # 綁定按鈕事件
-        submit_button.click(
-            fn=edit_video,
-            inputs=[input_video, margin, min_clip, min_cut],
-            outputs=[output_video, output_text]
-        )
-    
-    return interface
+iface = gr.Interface(
+    fn=edit_video,
+    inputs=gr.Video(label="上傳影片或音訊檔案"),
+    outputs=gr.File(label="下載編輯後的檔案"),
+    title="Auto-Editor Demo on CPU",
+    description="上傳影片，自動移除靜音部分。處理時間視檔案大小而定（CPU 環境下可能需數分鐘）。"
+)
 
-# 啟動 Gradio 應用
-if __name__ == "__main__":
-    interface = create_gradio_interface()
-    interface.launch()
+iface.launch()