import os
import gradio as gr


def function(audio_path, output_dir="example/output"):
    filename = os.path.basename(audio_path).split('.')[0]

    cmd_result = os.popen(f'python spleeter separate -p spleeter:2stems -o {output_dir} {audio_path}').read()
    print(cmd_result)
    return f"example/output/{filename}/accompaniment.wav", f"example/output/{filename}/vocals.wav"

inputs = [
    gr.inputs.Audio(source="upload", type="filepath", label=None, optional=False)
]
outputs = [
    gr.outputs.Audio(type="file", label='accompaniment'),
    gr.outputs.Audio(type="file", label='vocals')
]

examples = ["example/audio_example.mp3"]


iface = gr.Interface(fn=function, inputs=inputs, outputs=outputs, examples=examples)
iface.launch(share=True)

if __name__ == "__main__":
    function("example/audio_example.mp3", "example/output")
    pass