Update app.py

app.py

@@ -84,54 +84,32 @@ def separate_audio(input_audio):
     temp_wav = None
     
     try:
-        logger.info(f"[{process_id}] 🚀 收到新請求: {input_audio}")
+        # 使用固定臨時目錄而非隨機生成
+        tmp_dir = "/tmp/gradio_outputs"
+        os.makedirs(tmp_dir, exist_ok=True)
         
-        # 1️⃣ 檔案驗證與轉換
-        if not os.path.exists(input_audio):
-            raise gr.Error("❌ 檔案不存在，請重新上傳")
+        # 建立唯一辨識子目錄
+        process_id = datetime.now().strftime("%Y%m%d%H%M%S%f")
+        output_dir = os.path.join(tmp_dir, process_id)
+        os.makedirs(output_dir)
         
-        if os.path.getsize(input_audio) > 50 * 1024 * 1024:  # 50MB 限制
-            raise gr.Error("❌ 檔案大小超過 50MB 限制")
-            
-        logger.info(f"[{process_id}] 🔁 轉換標準音檔格式...")
-        temp_wav = convert_to_wav(input_audio)
-        validate_audio(temp_wav)
-        
-        # 2️⃣ 建立輸出目錄
-        out_dir = tempfile.mkdtemp()
-        outfilename = os.path.join(out_dir, "output.wav")
-        logger.info(f"[{process_id}] 📁 建立臨時輸出目錄: {out_dir}")
+        # 輸出路徑改為固定結構
+        outfilename = os.path.join(output_dir, "output.wav")
         
-        # 3️⃣ 執行語音分離
-        logger.info(f"[{process_id}] 🧠 開始執行語音分離...")
+        # 執行分離
         sep_files = dpt_sep_process(temp_wav, model=model, outfilename=outfilename)
         
-        # 4️⃣ 驗證輸出結果
+        # 確保輸出檔案存在
         for f in sep_files:
             if not os.path.exists(f):
-                raise gr.Error(f"❌ 缺少輸出檔案: {f}")
-            validate_audio(f)
+                raise gr.Error(f"❌ 缺失輸出檔案: {f}")
         
-        logger.info(f"[{process_id}] ✅ 處理成功完成")
         return sep_files
         
-    except RuntimeError as e:
-        if "CUDA out of memory" in str(e):
-            logger.error(f"[{process_id}] 💥 CUDA 記憶體不足")
-            raise gr.Error("⚠️ 記憶體不足，請上傳較短的音檔") from e
-        else:
-            raise
     except Exception as e:
-        logger.error(f"[{process_id}] ❌ 處理錯誤: {str(e)}\n{traceback.format_exc()}")
-        raise gr.Error(f"⚠️ 處理失敗: {str(e)}") from e
-    finally:
-        # 清理臨時檔案
-        if temp_wav and os.path.exists(temp_wav):
-            try:
-                os.unlink(temp_wav)
-                logger.info(f"[{process_id}] 🧹 臨時檔案已清理")
-            except Exception as clean_err:
-                logger.warning(f"[{process_id}] ⚠️ 清理失敗: {str(clean_err)}")
+        # 清理失敗時保留臨時資料夾以利除錯
+        logger.error(f"⚠️ 處理失敗: {str(e)}")
+        raise gr.Error("❌ 處理失敗，請檢查日誌") from e
 
 # 🎯 description 內容（轉為 HTML）
 description_html = """
@@ -175,9 +153,10 @@ AUDIO_INPUT = gr.Audio(
     max_length=180  # 最大 3 分鐘
 )
 
+# 修改 Gradio 輸出設定
 AUDIO_OUTPUTS = [
-    gr.Audio(label="🗣️ 語音軌道 1", type="filepath"),
-    gr.Audio(label="🗣️ 語音軌道 2", type="filepath")
+    gr.Audio(label="🗣️ 語音軌道 1", type="filepath", format="wav"),
+    gr.Audio(label="🗣️ 語音軌道 2", type="filepath", format="wav")
 ]
 
 # 🚀 啟動應用程式