import gradio as gr
from transformers import pipeline

# Load fine-tuned Urdu Whisper model
pipe = pipeline(
    "automatic-speech-recognition",
    model="hadiqa123/whisper-small-ur",
    device=-1  # Use 0 if GPU is available
)

# Transcription function
def transcribe(audio):
    if audio is None:
        return "⚠️ No audio input detected."
    result = pipe(audio)
    return result["text"]

# Gradio interface
gr.Interface(
    fn=transcribe,
    inputs=gr.Audio(type="filepath", label="🎙️ Record or Upload Urdu Audio"),
    outputs=gr.Textbox(label="📝 Transcribed Urdu Text"),
    title="🗣️ Urdu Speech to Text Converter",
).launch()