Update webui.py

webui.py

@@ -1,33 +1,31 @@
 import gradio as gr
 import subprocess
 import os
+import soundfile as sf
+from pathlib import Path
+import separate
 
 def audio_model_inference(files, output_folder, model_path, denoise, margin, chunks, n_fft, dim_t, dim_f):
-    # 构建命令行调用字符串
-    cmd = f"separate.py {' '.join(files)}"
-    if output_folder:
-        cmd += f" -o {output_folder}"
-    if model_path:
-        cmd += f" -m {model_path}"
-    if denoise:
-        cmd += " -d"
-    if margin:
-        cmd += f" -M {margin}"
-    if chunks:
-        cmd += f" -c {chunks}"
-    if n_fft:
-        cmd += f" -F {n_fft}"
-    if dim_t:
-        cmd += f" -t {dim_t}"
-    if dim_f:
-        cmd += f" -f {dim_f}"
-    
-    # 执行命令行调用
-    result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
-    
-    # 检查命令是否成功执行
-    if result.returncode != 0:
-        return f"错误：{result.stderr}"
+    filename = files.split('/')[-1]
+    # 执行调用
+    audio_worker = separate.Predictor(args={
+    										"files": [files],
+    										"output": Path(output_folder),
+    										"model_path": Path(model_path),
+    										"denoise": denoise,
+    										"margin": margin,
+    										"chunks": chunks,
+    										"n_fft": n_fft,
+    										"dim_t": dim_t,
+    										"dim_f": dim_f
+											})
+    vocals, no_vocals, sampling_rate = audio_worker.predict(files)
+    sf.write(os.path.join(output_folder, mp3_filename + "_no_vocals.wav"), no_vocals, sampling_rate)
+    sf.write(os.path.join(output_folder, mp3_filename + "_vocals.wav"), vocals, sampling_rate)
+
+    # 生成分离后的文件名
+    vocals_filename = f"{os.path.splitext(filename)[0]}_vocals.wav"
+    no_vocals_filename = f"{os.path.splitext(filename)[0]}_no_vocals.wav"
     
     # 读取输出文件
     vocals_file = f"{os.path.splitext(os.path.basename(files[0]))[0]}_vocals.wav"