import os
import gradio as gr
import tempfile
from uuid import uuid4
from audio_sep_splitter import batch_process_for_file

def split_audio(audio_filepath):
    output_file = f"{uuid4()}.mp3"
    stem = "voice"
    aggressiveness_factor = 2
    splitter = "phoenix"
    audio_file = audio_filepath.name
    vocal_file = batch_process_for_file(audio_file, output_file, stem, aggressiveness_factor, splitter)
    return output_file

interface = gr.Interface(
    split_audio,
    gr.Audio("Upload audio"),
    gr.Audio("Download audio")
)

if __name__=="__main__":
    interface.queue().launch(auth=(os.environ.get("USERNAME"), os.environ.get("PASSWORD")))