app.py

import os
import replicate
import gradio as gr

# Asegúrate de configurar tu REPLICATE_API_TOKEN
os.environ["REPLICATE_API_TOKEN"] = "<tu-token-aqui>"

# Función para transcribir el audio
def transcribe_audio(audio_file):
    # Ejecuta el modelo de replicate
    output = replicate.run(
        "vaibhavs10/incredibly-fast-whisper:3ab86df6c8f54c11309d4d1f930ac292bad43ace52d10c80d87eb258b3c9f79c",
        input={
            "task": "transcribe",
            "audio": audio_file,  # El archivo de audio cargado en Gradio
            "language": "None",  # Detecta automáticamente el idioma
            "timestamp": "chunk",  # Incluye marcas de tiempo
            "batch_size": 64,
            "diarise_audio": False
        }
    )
    return output['text']  # Devuelve solo el texto transcrito

# Crea la interfaz de Gradio
with gr.Blocks() as demo:
    gr.Markdown("# Transcripción de Audio usando Whisper")
    audio_input = gr.Audio(source="upload", type="filepath", label="Sube tu archivo de audio")
    output_text = gr.Textbox(label="Transcripción")
    
    transcribe_button = gr.Button("Transcribir")
    transcribe_button.click(fn=transcribe_audio, inputs=audio_input, outputs=output_text)

# Ejecuta la aplicación
demo.launch()