Update app.py

app.py

@@ -1,25 +1,41 @@
-import replicate
-import gradio
 import os
+import replicate
+import gradio as gr
+from dotenv import load_dotenv
 
-# Cargar el token desde las variables de entorno
+# Cargar variables de entorno desde .env
+load_dotenv()
+
+# Asegúrate de configurar tu REPLICATE_API_TOKEN
 replicate_token = os.getenv("REPLICATE_API_TOKEN")
 
-# Verifica que el token se haya cargado
 if not replicate_token:
     raise ValueError("No se ha encontrado el token de API de Replicate.")
 
-# Ejecuta el modelo
-output = replicate.run(
-    "vaibhavs10/incredibly-fast-whisper:3ab86df6c8f54c11309d4d1f930ac292bad43ace52d10c80d87eb258b3c9f79c",
-    input={
-        "task": "transcribe",
-        "audio": "https://replicate.delivery/pbxt/Js2Fgx9MSOCzdTnzHQLJXj7abLp3JLIG3iqdsYXV24tHIdk8/OSR_uk_000_0050_8k.wav",
-        "language": "None",
-        "timestamp": "chunk",
-        "batch_size": 64,
-        "diarise_audio": False
-    }
-)
+# Función para transcribir el audio
+def transcribe_audio(audio_file):
+    with open(audio_file, "rb") as audio:
+        output = replicate.run(
+            "vaibhavs10/incredibly-fast-whisper:3ab86df6c8f54c11309d4d1f930ac292bad43ace52d10c80d87eb258b3c9f79c",
+            input={
+                "task": "transcribe",
+                "audio": audio,  # 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(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)
 
-print(output)
+# Ejecuta la aplicación
+demo.launch()