import gradio as gr
from asr import transcribe_auto  # Import the updated function

# Speech-to-Text Interface
mms_transcribe = gr.Interface(
    fn=transcribe_auto,
    inputs=gr.Audio(),
    outputs="text",
    title="Multilingual Speech-to-Text",
    description="Automatically detects and transcribes English or Swahili speech.",
    allow_flagging="never",
)

# Main Gradio App
with gr.Blocks() as demo:
    gr.Markdown("<p align='center' style='font-size: 20px;'>MMS Speech-to-Text</p>")
    gr.HTML("<center>Upload or record audio. The model will automatically detect the language (English or Swahili) and transcribe.</center>")

    mms_transcribe.render()

if __name__ == "__main__":
    demo.queue()
    demo.launch()