import whisper
import gradio as gr

# Cargar el modelo de Whisper
model = whisper.load_model("base")

# Función principal que procesa el archivo de audio
def transcribe_audio(audio_file):
    # Realiza la transcripción en español
    result = model.transcribe(audio_file, language="es")
    transcription = result["text"]
    return transcription

# Crear la interfaz de Gradio
iface = gr.Interface(
    fn=transcribe_audio,  # La función que realiza la transcripción
    inputs=gr.Audio(type="filepath"),  # Entrada: archivo de audio
    outputs=gr.Textbox(label="Transcripción en Español"),  # Salida: cuadro de texto con la transcripción
    title="Transcriptor de Audio",
    description="Sube un archivo de audio en español para obtener su transcripción."
)

# Iniciar la aplicación
if __name__ == "__main__":
    iface.launch()