Update app.py

app.py

@@ -1,6 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 import os
+from gtts import gTTS  # Importamos gTTS
+
+# Función para convertir texto a audio
+def convertir_texto_a_audio(texto, nombre_archivo):
+    tts = gTTS(text=texto, lang='es')
+    tts.save(nombre_archivo)
 
 # Creamos un modelo de resumen más ligero
 resumidor = pipeline("summarization", model="t5-small")
@@ -18,13 +24,13 @@ def transformar_texto(archivo):
             texto = f.read()
     except Exception as e:
         raise gr.Error(f"Error al leer el archivo: {e}")
-    
+
     # Limitación para la versión gratuita
     max_palabras = 1000  # Puedes ajustar este número
     num_palabras = len(texto.split())
     if num_palabras > max_palabras:
         raise gr.Error(f"El texto excede el límite de {max_palabras} palabras para la versión gratuita.")
-    
+
     try:
         # Resumimos el texto
         resumen = resumidor(
@@ -36,7 +42,7 @@ def transformar_texto(archivo):
         )[0]['summary_text']
     except Exception as e:
         raise gr.Error(f"Error al resumir el texto: {e}")
-    
+
     try:
         # Generamos una conversación basada en el resumen
         prompt = f"Crear una conversación entre dos personas sobre: {resumen}"
@@ -48,22 +54,21 @@ def transformar_texto(archivo):
         )[0]['generated_text']
     except Exception as e:
         raise gr.Error(f"Error al generar la conversación: {e}")
-    
-    # Guardamos la conversación en un archivo en /tmp
-    ruta_archivo = os.path.join("/tmp", "conversacion.txt")
+
+    # Convertimos la conversación a audio
+    ruta_audio = os.path.join("/tmp", "conversacion.mp3")
     try:
-        with open(ruta_archivo, "w", encoding='utf-8') as f:
-            f.write(conversacion)
+        convertir_texto_a_audio(conversacion, ruta_audio)
     except Exception as e:
-        raise gr.Error(f"Error al guardar la conversación: {e}")
-    
-    return ruta_archivo
+        raise gr.Error(f"Error al convertir texto a audio: {e}")
+
+    return ruta_audio
 
 # Creamos la interfaz de usuario actualizada
 interfaz = gr.Interface(
     fn=transformar_texto,
     inputs=gr.File(label="Sube tu documento de texto (.txt)"),
-    outputs=gr.File(label="Descarga la conversación generada"),
+    outputs=gr.Audio(type="file", label="Escucha la conversación generada"),
     title="📝➡️🎙️ Transformador de Texto a Conversación",
     description="""
     Sube un documento de texto y conviértelo en una conversación tipo podcast entre dos personas.