Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 import os
 import google.generativeai as genai
 import random
+import re  # Importar el módulo re para expresiones regulares
 
 # Cargar las variables de entorno
 load_dotenv()
@@ -10,6 +11,15 @@ load_dotenv()
 # Configurar la API de Google
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
+# Función para limpiar el texto
+def clean_text(text):
+    # Eliminar espacios en blanco al inicio y al final
+    text = text.strip()
+    # Reemplazar múltiples espacios en blanco por uno solo
+    text = re.sub(r'\s+', ' ', text)
+    # Aquí puedes agregar más reglas de limpieza si es necesario
+    return text
+
 # Función para obtener una mención del producto de manera probabilística
 def get_random_product_mention():
     mentions = ["Directa", "Indirecta", "Metafórica"]
@@ -114,8 +124,8 @@ def generate_ctas(number_of_ctas, target_audience, product, call_to_action, temp
     try:
         response = model.generate_content([ctas_instruction])
         
-        # Extraer el texto de la respuesta
-        generated_ctas = response.candidates[0].content.parts[0].text.strip()  # Modificado aquí
+        # Extraer el texto de la respuesta y limpiar
+        generated_ctas = clean_text(response.candidates[0].content.parts[0].text)  # Aplicar limpieza
         
         # Retornar el resultado
         return generated_ctas