Update app.py

app.py

@@ -218,6 +218,73 @@ story_formulas = {
     }
 }
 
+def generate_headline_for_story(model, story, formula_type, product, target_audience, mood):
+    """Función para generar un titular basado en la historia y la fórmula seleccionada"""
+
+    # Configurar el prompt según la fórmula seleccionada
+    formula_prompts = {
+        "GPS": f"""
+        Genera un titular usando la fórmula GPS (Goal-Period-Solution):
+        1. Meta: ¿Qué desea lograr {target_audience}?
+        2. Periodo: ¿En qué marco temporal o situación cotidiana?
+        3. Superación: Usa un conector con toque de humor (sin, incluso si, aunque, etc.)
+
+        La historia a titular es:
+        {story}
+        """,
+        "AIDA": f"""
+        Genera un titular usando la fórmula AIDA:
+        1. Atención: Usa un gancho poderoso
+        2. Interés: Desarrolla el gancho inicial
+        3. Deseo: Amplifica la emoción
+        4. Acción: Cierre natural
+
+        La historia a titular es:
+        {story}
+        """,
+        "4U": f"""
+        Genera un titular usando la fórmula 4U:
+        1. Útil: Beneficio práctico
+        2. Urgente: Motivador de acción
+        3. Único: Diferenciador memorable
+        4. Ultra-específico: Detalles precisos
+
+        La historia a titular es:
+        {story}
+        """,
+        "Númerica Suprema": f"""
+        Genera un titular usando la Fórmula Numérica Suprema:
+        1. Artículo plural (Los/Las)
+        2. Número específico
+        3. Adjetivo emocional
+        4. Palabra clave en plural
+        5. Razón
+        6. Promesa
+
+        La historia a titular es:
+        {story}
+        """
+    }
+
+    # Usar el prompt predeterminado si la fórmula no está en el diccionario
+    headline_prompt = formula_prompts.get(formula_type, f"""
+    Genera un titular persuasivo y memorable en español para la siguiente historia.
+    El titular debe:
+    - Conectar con {target_audience}
+    - Mantener un tono {mood}
+    - Ser conciso y memorable
+    - No mencionar directamente "{product}"
+    - Generar curiosidad y deseo de leer más
+
+    Historia:
+    {story}
+    """)
+
+    response = model.generate_content([headline_prompt])
+    if response and response.parts:
+        return response.parts[0].text.strip()
+    return None
+    
 def generate_story(formula_type, target_audience, product, action, mood, length, story_topic=None):
     """Función unificada para generar historias"""
     model = genai.GenerativeModel("gemini-2.0-flash")
@@ -443,8 +510,14 @@ IMPORTANT FINAL REMINDER:
     story_instruction = f"{natural_instruction}\n\n{story_instruction}"
     
     response = model.generate_content([story_instruction])
-    if response and response.parts:
-        return response.parts[0].text.strip()
+        if response and response.parts:
+            story = response.parts[0].text.strip()
+            # Generar el titular usando la fórmula seleccionada
+            headline = generate_headline_for_story(model, story, formula_type, product, target_audience, mood)
+            # Combinar titular y historia
+            if headline:
+                return f"# {headline}\n\n{story}"
+            return story
     raise ValueError("No se pudo generar la historia")
     
 # Inicializar la aplicación Streamlit