Update app.py

app.py

@@ -185,23 +185,22 @@ headline_formulas = {
     }
 }
 
-# Función para generar titulares
 def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula):
-
     # Crear la configuración del modelo
     generation_config = {
-        "temperature": temperature,  
-        "top_p": 0.65,        # Considerar un poco menos de palabras probables
-        "top_k": 360,        # Aumentar las palabras candidatas para más variedad
-        "max_output_tokens": 8196,  # Mantenerlo igual, pero puedes aumentar si deseas más detalle
+        "temperature": temperature,
+        "top_p": 0.65,
+        "top_k": 360,
+        "max_output_tokens": 8196,
     }
 
     model = genai.GenerativeModel(
         model_name="gemini-2.0-flash",
         generation_config=generation_config,
-)
-# Incluir las instrucciones del sistema en el prompt principal
-system_prompt = """You are a world-class copywriter, with expertise in crafting hooks, headlines, and subject lines that immediately capture the reader's attention, prompting them to open the email or continue reading.
+    )
+
+    # Incluir las instrucciones del sistema en el prompt principal
+    system_prompt = """You are a world-class copywriter, with expertise in crafting hooks, headlines, and subject lines that immediately capture the reader's attention, prompting them to open the email or continue reading.
 
 OBJECTIVE:
 - Generate persuasive and specific headlines in Spanish
@@ -239,7 +238,6 @@ IMPORTANT:
 - Adapt language to the audience
 - Focus on transformative benefits"""
 
-# Crear un mensaje para el modelo
     headlines_instruction = (
         f"{system_prompt}\n\n"
         f"Tu tarea es crear {number_of_headlines} titulares irresistibles diseñados para {target_audience}. "
@@ -253,7 +251,6 @@ IMPORTANT:
         f"Tu objetivo es crear titulares tan intrigantes que sea imposible no seguir leyendo."
     )
 
-    # Crear la sesión de chat
     chat_session = model.start_chat(
         history=[
             {
@@ -264,7 +261,7 @@ IMPORTANT:
     )
 
     response = chat_session.send_message("Genera los titulares")
-    return response.text  # Este return debe estar dentro de la función
+    return response.text
 
 # Configurar la interfaz de usuario con Streamlit
 st.set_page_config(page_title="Enchanted Hooks", layout="wide")