import gradio as gr
import torch
import torchaudio
import os
import tempfile
import logging
import traceback
from datetime import datetime
# 設定日誌系統
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)
# 檢查 Hugging Face 環境變數
if not os.getenv("SpeechSeparation"):
logger.warning("⚠️ 環境變數 SpeechSeparation 未設定！請在 Hugging Face Space 的 Secrets 中設定 HF_TOKEN")
# 載入模型模組
try:
logger.info("🔧 開始載入語音分離模型...")
from DPTNet_eval.DPTNet_quant_sep import load_dpt_model, dpt_sep_process
logger.info("✅ 模型模組載入成功")
except ImportError as e:
logger.error(f"❌ 模組載入失敗: {str(e)}")
raise RuntimeError("本地模組路徑配置錯誤") from e
# 全域模型初始化
try:
logger.info("🔄 初始化模型中...")
model = load_dpt_model()
logger.info(f"🧠 模型載入完成，運行設備: {'GPU' if torch.cuda.is_available() else 'CPU'}")
except Exception as e:
logger.error(f"💣 模型初始化失敗: {str(e)}")
raise RuntimeError("模型載入異常終止") from e
def validate_audio(path):
"""驗證音檔格式與內容有效性"""
try:
info = torchaudio.info(path)
logger.info(f"🔊 音檔資訊: 采樣率={info.sample_rate}Hz, 通道數={info.num_channels}")
if info.num_channels not in [1, 2]:
raise gr.Error("❌ 不支援的音檔通道數（僅支援單聲道或立體聲）")
if info.sample_rate < 8000 or info.sample_rate > 48000:
raise gr.Error("❌ 不支援的采樣率（需介於 8kHz~48kHz）")
return info.sample_rate
except Exception as e:
logger.error(f"⚠️ 音檔驗證失敗: {str(e)}")
raise gr.Error("❌ 無效的音訊檔案格式")
def convert_to_wav(input_path):
"""統一轉換為 16kHz WAV 格式"""
try:
# 使用 torchaudio 保持一致性
waveform, sample_rate = torchaudio.load(input_path)
# 單聲道轉換
if waveform.shape[0] > 1:
waveform = torch.mean(waveform, dim=0, keepdim=True)
# 重采樣至 16kHz
if sample_rate != 16000:
resampler = torchaudio.transforms.Resample(orig_freq=sample_rate, new_freq=16000)
waveform = resampler(waveform)
# 建立臨時 WAV 檔案
with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as tmpfile:
torchaudio.save(tmpfile.name, waveform, 16000, bits_per_sample=16)
logger.info(f"📝 已生成標準 WAV 檔案: {tmpfile.name}")
return tmpfile.name
except Exception as e:
logger.error(f"⚠️ 音檔轉換失敗: {str(e)}")
raise gr.Error("❌ 音訊格式轉換失敗")
def separate_audio(input_audio):
"""主處理函式"""
process_id = datetime.now().strftime("%Y%m%d%H%M%S%f")
temp_wav = None
try:
logger.info(f"[{process_id}] 🚀 收到新請求: {input_audio}")
# 1️⃣ 檔案驗證與轉換
if not os.path.exists(input_audio):
raise gr.Error("❌ 檔案不存在，請重新上傳")
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}")
# 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)
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)}")
# 🎯 description 內容（轉為 HTML）
description_html = """
"""
EXAMPLES = [
["examples/sample1.wav"],
["examples/sample2.mp3"]
]
AUDIO_INPUT = gr.Audio(
label="🔊 上傳混合音檔",
type="filepath",
sources=["upload", "microphone"],
show_label=True,
max_length=180 # 最大 3 分鐘
)
AUDIO_OUTPUTS = [
gr.Audio(label="🗣️ 語音軌道 1", type="filepath"),
gr.Audio(label="🗣️ 語音軌道 2", type="filepath")
]
# 🚀 啟動應用程式
interface = gr.Interface(
fn=separate_audio,
inputs=AUDIO_INPUT,
outputs=AUDIO_OUTPUTS,
title="🎙️ 語音分離，上傳一段混音音檔（支援.mp3, .wav），自動分離出兩個人的聲音；Deep Learning 101",
description=description_html,
examples=EXAMPLES,
allow_flagging="never",
cache_examples=False,
theme="default"
)
LAUNCH_CONFIG = {
"server_name": "0.0.0.0",
"server_port": 7860,
"share": False,
"debug": True,
"auth": None,
"inbrowser": True,
"quiet": False
}
if __name__ == "__main__":
logger.info("🚀 啟動 Gradio 服務...")
interface.launch(**LAUNCH_CONFIG)