import gradio as gr
from audio_separator import Separator

def separate_audio(audio_file):
    separator = Separator()
    stems = separator.separate(audio_file, stem_names=["instrumental", "vocals", "backing_vocals"])
    return stems["instrumental"], stems["vocals"], stems["backing_vocals"]

iface = gr.Interface(
    fn=separate_audio,
    inputs=gr.Audio(type="filepath", label="Upload Audio"),
    outputs=[
        gr.Audio(label="Instrumental"),
        gr.Audio(label="Vocals"),
        gr.Audio(label="Backing Vocals")
    ],
    title="Audio Separator",
    description="Separate audio into instrumental, vocals, and backing vocals using audio-separator."
)

iface.launch()