Update app.py

app.py

@@ -1,23 +1,22 @@
-# app.py
 import gradio as gr
-from separation_utils import process_audio
+import librosa
+import soundfile as sf
+import numpy as np
+from utils import run_separation
 
-interface = gr.Interface(
-    fn=process_audio,
-    inputs=gr.Audio(type="filepath", label="Upload Audio File"),
+def separate_and_play(file):
+    separated = run_separation(file)
+    return separated['background'], separated['foreground']
+
+demo = gr.Interface(
+    fn=separate_and_play,
+    inputs=gr.Audio(type="filepath", label="Upload Mixed Audio"),
     outputs=[
-        gr.Audio(type="filepath", label="Foreground Audio 2DFT"),
-        gr.Textbox(label="PSNR (dB) 2DFT", interactive=False),
-        gr.Textbox(label="KL Divergence 2DFT", interactive=False),
-        gr.Audio(type="filepath", label="Foreground Audio Repet"),
-        gr.Textbox(label="PSNR (dB) Repet", interactive=False),
-        gr.Textbox(label="KL Divergence Repet", interactive=False),
-        gr.Audio(type="filepath", label="Foreground Audio Repet Sim"),
-        gr.Textbox(label="PSNR (dB) Repet Sim", interactive=False),
-        gr.Textbox(label="KL Divergence Repet Sim", interactive=False)
+        gr.Audio(label="Background"),
+        gr.Audio(label="Foreground"),
     ],
-    title="Audio Foreground Separator",
-    description="Upload an audio file to separate its foreground using 2DFT, Repet, and Repet_Sim models."
+    title="Speaker Source Separation",
+    description="Upload a mixed audio file to hear separated background and foreground tracks using traditional audio separation methods."
 )
 
-interface.launch()
+demo.launch()