import gradio as gr


def asr_inference(audio_file):
    if not audio_file:
        return "Please upload a wav file"
    model = None
    text_output = "Developing " + audio_file
    return text_output


with gr.Blocks(title="FireRedASR") as demo:
    gr.HTML(
        "<h1 style='text-align: center'>FireRedASR</h1>"
    )
    gr.Markdown("Upload an audio file (wav) to get speech-to-text results.")

    with gr.Row():
        with gr.Column():
            audio_file = gr.Audio(label="Upload Audio", sources=["microphone", "upload"], type="filepath")
            #audio_file = gr.Audio(label="Upload wav file", sources=["upload"], type="filepath")
            asr_button = gr.Button("Start Recognition", variant="primary")

        with gr.Column():
            text_output = gr.Textbox(label="Model Result", interactive=False, lines=6, max_lines=12)

    asr_button.click(
        fn=asr_inference,
        inputs=[audio_file],
        outputs=[text_output]
    )


demo.launch()