Update app.py

app.py

@@ -1,36 +1,36 @@
 import gradio as gr
-import librosa
-import numpy as np
-import soundfile as sf
+import subprocess
+import os
+import shutil
 
-def vocal_remover(audio):
-    y, sr = librosa.load(audio, sr=None)
+def separate(audio_path):
+    output_dir = "separated"
+    if os.path.exists(output_dir):
+        shutil.rmtree(output_dir)
 
-    # Simple center-channel vocal suppression
-    S = librosa.stft(y)
-    left = S[:, :]
-    right = S[:, :]
+    cmd = [
+        "demucs",
+        "--two-stems=vocals",
+        audio_path
+    ]
 
-    vocals = (left - right) / 2
-    music = (left + right) / 2
+    subprocess.run(cmd, check=True)
 
-    vocals_audio = librosa.istft(vocals)
-    music_audio = librosa.istft(music)
+    base = os.path.splitext(os.path.basename(audio_path))[0]
+    vocals = f"separated/htdemucs/{base}/vocals.wav"
+    music = f"separated/htdemucs/{base}/no_vocals.wav"
 
-    sf.write("vocals.wav", vocals_audio, sr)
-    sf.write("music.wav", music_audio, sr)
+    return vocals, music
 
-    return "vocals.wav", "music.wav"
-
-interface = gr.Interface(
-    fn=vocal_remover,
-    inputs=gr.Audio(type="filepath", label="Upload Audio"),
+demo = gr.Interface(
+    fn=separate,
+    inputs=gr.Audio(type="filepath", label="Upload Audio (max 2–3 min)"),
     outputs=[
         gr.File(label="Vocals"),
         gr.File(label="Music")
     ],
-    title="AI Vocal Remover",
-    description="Upload audio & get vocals and music separated"
+    title="🔥 REAL AI Vocal Remover (Demucs)",
+    description="Studio-quality vocal & music separation using deep learning"
 )
 
-interface.launch()
+demo.launch()