Update app.py

app.py

@@ -2,30 +2,33 @@ import gradio as gr
 import subprocess
 import os
 import uuid
+import shutil
 
-def clean_audio(audio):
+def clean_audio(audio_path):
     uid = str(uuid.uuid4())
-    input_path = f"/tmp/{uid}.wav"
-    output_dir = f"/tmp/out_{uid}"
+    out_dir = f"/tmp/out_{uid}"
 
-    os.makedirs(output_dir, exist_ok=True)
-
-    # Save input
-    with open(input_path, "wb") as f:
-        f.write(audio)
+    os.makedirs(out_dir, exist_ok=True)
 
     # Demucs command
-    cmd = f"demucs -n htdemucs {input_path} -o {output_dir}"
-    subprocess.call(cmd, shell=True)
+    cmd = [
+        "demucs",
+        "-n", "htdemucs",
+        audio_path,
+        "-o", out_dir
+    ]
+    subprocess.run(cmd, check=True)
 
-    vocals = f"{output_dir}/htdemucs/{uid}/vocals.wav"
+    vocals_path = os.path.join(
+        out_dir, "htdemucs", os.path.splitext(os.path.basename(audio_path))[0], "vocals.wav"
+    )
 
-    return vocals
+    return vocals_path
 
 gr.Interface(
     fn=clean_audio,
-    inputs=gr.Audio(type="binary", label="Upload Audio"),
-    outputs=gr.Audio(label="Clean Vocals"),
+    inputs=gr.Audio(type="filepath", label="Upload Audio"),
+    outputs=gr.Audio(type="filepath", label="Clean Vocals"),
     title="AI Voice Cleaner & Vocal Remover",
-    description="Noise remove + music separation using Demucs"
+    description="Noise removal & vocal separation using Demucs (HF Spaces compatible)"
 ).launch()