import gradio as gr
from transformers import pipeline
from audio2hero import generate_midi

gradio_app = gr.Interface(
    generate_midi,
    inputs=gr.Audio(label="Input Audio", type="filepath"),
    outputs=gr.File(label="Output MIDI Zip File"),
    title="Audio2Hero AI Autocharter for CloneHero",
    description="""Audio2Hero will generate a medium difficulty Clone Hero chart from any audio file. 
                  Make sure to rename your audio file to 'Artist - Song Name' for the autocharter to correctly
                  generate the song.ini file. The output will be a zip file containing the MIDI file, song.ogg 
                  and song.ini file. The auto charter takes about 3 minutes to generate so please be patient. Hope you enjoy!""",
)

if __name__ == "__main__":
    gradio_app.launch()