Update app.py

app.py

@@ -8,6 +8,9 @@ from pydub import AudioSegment
 tts_juan = TTS(model_name="tts_models/es/mai/tacotron2-DDC", progress_bar=False, gpu=False)
 tts_maria = TTS(model_name="tts_models/es/css10/vits", progress_bar=False, gpu=False)
 
+# Modelo de resumen
+resumidor = pipeline("summarization", model="google/pegasus-xsum")
+
 # Función para convertir texto a audio con voces diferentes
 def convertir_texto_a_audio(texto, nombre_archivo, voz):
     if voz == "Juan":
@@ -58,22 +61,29 @@ def transformar_texto(archivo):
     except Exception as e:
         raise gr.Error(f"Error al leer el archivo: {e}")
 
-    max_palabras = 500  # Reducimos el límite de palabras
+    max_palabras = 2000  # Nuevo límite de palabras
     num_palabras = len(texto.split())
     if num_palabras > max_palabras:
         raise gr.Error(f"El texto excede el límite de {max_palabras} palabras.")
 
-    # Limitar el texto a un número máximo de caracteres
-    max_caracteres = 1500
-    if len(texto) > max_caracteres:
-        texto = texto[:max_caracteres]
+    # Resumimos el texto para que el prompt no sea demasiado largo
+    try:
+        resumen = resumidor(
+            texto,
+            max_length=200,  # Ajusta según tus necesidades
+            min_length=100,
+            do_sample=False,
+            clean_up_tokenization_spaces=True
+        )[0]['summary_text']
+    except Exception as e:
+        raise gr.Error(f"Error al resumir el texto: {e}")
 
     try:
-        # Generamos la conversación utilizando el texto completo
+        # Generamos la conversación utilizando el resumen
         prompt = f"""
 Crear una conversación educativa entre Juan y María sobre el siguiente tema:
 
-{texto}
+{resumen}
 
 La conversación debe tener el formato:
 
@@ -85,7 +95,7 @@ Por favor, asegúrate de que la conversación cubra los puntos principales del t
 """
         output = dialogador(
             prompt,
-            max_length=512,
+            max_length=1024,  # Aumentamos el límite de tokens generados
             num_return_sequences=1,
             clean_up_tokenization_spaces=True
         )
@@ -116,7 +126,3 @@ interfaz = gr.Interface(
 )
 
 interfaz.launch()
-
-
-
-