Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
 import tensorflow as tf
 from tensorflow.python.ops.numpy_ops import np_config
+import gradio as gr 
+import tempfile
 import numpy as np
 import atexit
 
@@ -16,49 +18,49 @@ if not os.path.exists("/home/user/.spleeter_installed"):
 
 
 # Now import normal modules 
-import gradio as gr 
 from spleeter.separator import Separator 
 from spleeter.audio.adapter import AudioAdapter 
-import tempfile
 
-# Initialize Spleeter
+
+
+
+
+
+
+
+# Initialize Spleeter separator
 separator = Separator("spleeter:2stems", multiprocess=False)
 audio_loader = AudioAdapter.default()
 
 def separate_audio(audio_file):
+    """
+    Separate the vocals and instrumental from the input audio file.
+    Returns filepaths for the separated stems and a status message.
+    """
     # Load audio as NumPy array
     waveform, sample_rate = audio_loader.load(audio_file, sample_rate=44100)
-    waveform = np.array(waveform)  # ensure NumPy array
-
-    # Separate stems
+    
+    # Separate stems using Spleeter
     pred = separator.separate(waveform)
 
-    # Save results to temporary folder
-    tmp = tempfile.mkdtemp()
-    vocals_path = os.path.join(tmp, "vocals.wav")
-    instr_path = os.path.join(tmp, "instrumental.wav")
+    # Save results in a temporary directory
+    tmp_dir = tempfile.mkdtemp()
+    vocals_path = os.path.join(tmp_dir, "vocals.wav")
+    instr_path = os.path.join(tmp_dir, "instrumental.wav")
 
     audio_loader.save(vocals_path, pred["vocals"], sample_rate)
     audio_loader.save(instr_path, pred["accompaniment"], sample_rate)
 
     return vocals_path, instr_path, "Done!"
 
-# Create Gradio interface
+# Gradio interface
 iface = gr.Interface(
     fn=separate_audio,
-    inputs=gr.Audio(type="filepath"),
-    outputs=[gr.Audio(), gr.Audio(), gr.Textbox()],
-    title="Spleeter Vocal/Instrumental Separator"
+    inputs=gr.Audio(type="filepath", label="Input Audio"),
+    outputs=[gr.Audio(label="Vocals"), gr.Audio(label="Instrumental"), gr.Textbox(label="Status")],
+    title="Spleeter Vocal/Instrumental Separator",
+    description="Upload an audio file to separate it into vocals and instrumental tracks."
 )
 
-# Ensure clean shutdown
-def cleanup():
-    try:
-        iface.close()
-    except Exception:
-        pass
-
-atexit.register(cleanup)
-
 if __name__ == "__main__":
     iface.launch()