Update app.py

app.py

@@ -79,37 +79,53 @@ def convert_to_wav(input_path):
         raise gr.Error("❌ 音訊格式轉換失敗")
 
 def separate_audio(input_audio):
-    """主處理函式"""
     process_id = datetime.now().strftime("%Y%m%d%H%M%S%f")
     temp_wav = None
     
     try:
-        # 使用固定臨時目錄而非隨機生成
-        tmp_dir = "/tmp/gradio_outputs"
-        os.makedirs(tmp_dir, exist_ok=True)
+        logger.info(f"[{process_id}] 🚀 收到新請求: {input_audio}")
         
-        # 建立唯一辨識子目錄
-        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)
+        # 1️⃣ 檢查檔案大小
+        if os.path.getsize(input_audio) > 50 * 1024 * 1024:
+            raise gr.Error("❌ 檔案超過 50MB 限制")
+            
+        # 2️⃣ 轉換為標準格式
+        logger.info(f"[{process_id}] 🔁 轉換標準音檔...")
+        temp_wav = convert_to_wav(input_audio)
+        validate_audio(temp_wav)
         
-        # 輸出路徑改為固定結構
+        # 3️⃣ 建立固定輸出目錄
+        output_dir = os.path.join("/tmp/gradio_outputs", process_id)
+        os.makedirs(output_dir)
         outfilename = os.path.join(output_dir, "output.wav")
         
-        # 執行分離
+        # 4️⃣ 執行語音分離
+        logger.info(f"[{process_id}] 🧠 開始分離...")
         sep_files = dpt_sep_process(temp_wav, model=model, outfilename=outfilename)
         
-        # 確保輸出檔案存在
+        # 5️⃣ 驗證輸出
         for f in sep_files:
             if not os.path.exists(f):
                 raise gr.Error(f"❌ 缺失輸出檔案: {f}")
+            validate_audio(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}] 💥 GPU 記憶體不足")
+            raise gr.Error("⚠️ 請縮短音檔長度") from e
+        else:
+            raise
     except Exception as e:
-        # 清理失敗時保留臨時資料夾以利除錯
-        logger.error(f"⚠️ 處理失敗: {str(e)}")
-        raise gr.Error("❌ 處理失敗，請檢查日誌") from 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):
+            os.unlink(temp_wav)
+            logger.info(f"[{process_id}] 🧹 臨時檔案已清理")
 
 # 🎯 description 內容（轉為 HTML）
 description_html = """