Update app.py

app.py

@@ -1,39 +1,40 @@
+import os
+
+# Install Spleeter manually if missing
+if not os.path.exists("/home/user/.spleeter_installed"):
+    import install_spleeter
+    open("/home/user/.spleeter_installed", "w").close()
+
+# Now import all normal modules
 import gradio as gr
 from spleeter.separator import Separator
 from spleeter.audio.adapter import AudioAdapter
 import tempfile
-import os
 
 MODEL_PATH = "pretrained_models/2stems"
-
 separator = Separator(MODEL_PATH)
 audio_loader = AudioAdapter.default()
 
-def separate_audio(audio_file):
-    if audio_file is None:
-        return None, None, "Upload an audio file."
 
+def separate_audio(audio_file):
     waveform, sample_rate = audio_loader.load(audio_file)
-    prediction = separator.separate(waveform)
+    pred = separator.separate(waveform)
+
+    tmp = tempfile.mkdtemp()
+    vocals = f"{tmp}/vocals.wav"
+    instr = f"{tmp}/instrumental.wav"
 
-    tmpdir = tempfile.mkdtemp()
-    vocals_path = os.path.join(tmpdir, "vocals.wav")
-    instr_path = os.path.join(tmpdir, "instrumental.wav")
+    audio_loader.save(vocals, pred["vocals"], sample_rate)
+    audio_loader.save(instr, pred["accompaniment"], sample_rate)
 
-    audio_loader.save(vocals_path, prediction["vocals"], sample_rate)
-    audio_loader.save(instr_path, prediction["accompaniment"], sample_rate)
+    return vocals, instr, "Done!"
 
-    return vocals_path, instr_path, "Done!"
 
 iface = gr.Interface(
-    fn=separate_audio,
-    inputs=gr.Audio(type="filepath", label="Upload Audio"),
-    outputs=[
-        gr.Audio(label="Vocals"),
-        gr.Audio(label="Instrumental"),
-        gr.Textbox(label="Status")
-    ],
-    title="🎵 Spleeter Vocal/Instrumental Separator"
+    separate_audio,
+    inputs=gr.Audio(type="filepath"),
+    outputs=[gr.Audio(), gr.Audio(), gr.Textbox()],
+    title="Spleeter Vocal/Instrumental Separator"
 )
 
 iface.launch()