Update app.py

app.py

@@ -10,30 +10,52 @@ dialogador = pipeline("text-generation", model="microsoft/DialoGPT-small")
 
 def transformar_texto(archivo):
     if archivo is None:
-        return "Por favor, sube un archivo de texto."
+        raise gr.Error("Por favor, sube un archivo de texto.")
+
     try:
-        # Lee el contenido del archivo
-        texto = archivo.decode('utf-8')
+        # Lee el contenido del archivo subido
+        with open(archivo.name, 'r', encoding='utf-8') as f:
+            texto = f.read()
     except Exception as e:
-        return f"Error al leer el archivo: {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:
-        return f"El texto excede el límite de {max_palabras} palabras para la versión gratuita."
+        raise gr.Error(f"El texto excede el límite de {max_palabras} palabras para la versión gratuita.")
     
-    # Resumimos el texto
-    resumen = resumidor(texto, max_length=50, min_length=25, do_sample=False)[0]['summary_text']
+    try:
+        # Resumimos el texto
+        resumen = resumidor(
+            texto,
+            max_length=50,
+            min_length=25,
+            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}")
     
-    # Generamos una conversación basada en el resumen
-    prompt = f"Crear una conversación entre dos personas sobre: {resumen}"
-    conversacion = dialogador(prompt, max_length=200, num_return_sequences=1)[0]['generated_text']
+    try:
+        # Generamos una conversación basada en el resumen
+        prompt = f"Crear una conversación entre dos personas sobre: {resumen}"
+        conversacion = dialogador(
+            prompt,
+            max_length=200,
+            num_return_sequences=1,
+            clean_up_tokenization_spaces=True
+        )[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")
-    with open(ruta_archivo, "w", encoding='utf-8') as f:
-        f.write(conversacion)
+    try:
+        with open(ruta_archivo, "w", encoding='utf-8') as f:
+            f.write(conversacion)
+    except Exception as e:
+        raise gr.Error(f"Error al guardar la conversación: {e}")
     
     return ruta_archivo
 
@@ -63,3 +85,4 @@ interfaz = gr.Interface(
 
 interfaz.launch()
 
+