import gradio as gr
from api import UrduWhisper

model = UrduWhisper()

def run_transcription(audio):
    return model.transcribe(audio)

with gr.Blocks() as demo:
    gr.Markdown("# 🎙 Urdu Tiny Whisper — Custom ASR")
    audio_input = gr.Audio(type="filepath")
    output = gr.Textbox(label="Transcribed Urdu Text")
    btn = gr.Button("Transcribe")
    btn.click(run_transcription, inputs=audio_input, outputs=output)

demo.launch(server_name="0.0.0.0", server_port=7860)