Update app.py

app.py

@@ -4,85 +4,121 @@ import os
 import google.generativeai as genai
 
 load_dotenv()
-
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Función para obtener respuesta del modelo Gemini (P.A.S.A)
-def get_pasa_response(target_audience, product, action, mood, length):
+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.
+FORMAT RULES:
+- Each headline must start with number and period
+- One headline per line
+- No explanations or categories
+- Add a line break between each headline
+- Avoid unnecessary : symbols
+- Each headline must be a complete and intriguing sentence
+IMPORTANT ANGLE INSTRUCTIONS:
+- The selected angle MUST be applied to EVERY headline
+- The angle modifies HOW the formula is expressed, not its structure
+- Think of the angle as a "tone overlay" on the formula
+- The formula provides the structure, the angle provides the style
+- Both must work together seamlessly
+FORMAT EXAMPLE:
+1. Titular 1.
+2. Titular 2.
+3. Titular 3.
+4. Titular 4.
+5. Titular 5.
+IMPORTANT:
+- Each headline must be unique and memorable
+- Avoid clichés and generalities
+- Maintain an intriguing but credible tone
+- Adapt speaking language from the audience
+- Focus on transformative benefits
+- Follow the selected angle style while maintaining formula structure"""
+
+story_formulas = {
+    "PASA": {
+        "description": """
+        La fórmula P.A.S.A. se estructura de manera natural y fluida:
+        1. **Problema**:
+           - Identificación del dolor específico
+           - Situación actual problemática
+           - Frustración principal
+        2. **Agitación**:
+           - Consecuencias emocionales
+           - Impacto en la vida diaria
+           - Costos de no resolver
+        3. **Solución**:
+           - Presentación del producto/servicio
+           - Beneficios clave
+           - Transformación posible
+        4. **Acción**:
+           - Siguiente paso claro
+           - Motivación para actuar
+           - Beneficio inmediato
+        """
+    },
+    "ADP": {
+        "description": """
+        La fórmula A.D.P. fluye naturalmente:
+        1. **Antes**:
+           - Estado actual
+           - Frustraciones diarias
+           - Situación problemática
+        2. **Después**:
+           - Escenario ideal
+           - Beneficios alcanzados
+           - Nueva realidad
+        3. **Puente**:
+           - Conexión solución-resultado
+           - Transformación posible
+           - Camino al cambio
+        """
+    },
+    "GHA": {
+        "description": """
+        La fórmula G.H.A. se desarrolla naturalmente:
+        1. **Gancho**:
+           - Apertura impactante
+           - Elemento sorpresa
+           - Conexión inmediata
+        2. **Historia**:
+           - Narrativa envolvente
+           - Desarrollo personal
+           - Aprendizaje clave
+        3. **Acción**:
+           - Conclusión natural
+           - Invitación clara
+           - Motivación final
+        """
+    }
+}
+
+def generate_story(formula_type, **kwargs):
+    """Función unificada para generar historias"""
     model = genai.GenerativeModel("gemini-1.5-flash")
-    full_prompt = f"""
-    You are a skilled copywriter and storyteller. Create a persuasive story in Spanish following the P.A.S.A. formula (Problema, Agitación, Solución, Acción). 
-    Do not explicitly label or explain each section (Problema, Agitación, Solución, Acción). Instead, weave these elements naturally into a cohesive, flowing narrative that emotionally connects with the {target_audience}.
-    The story should have {length} words, with a {mood} tone. Here are the details:
-    1. Problema: Describe a specific problem that keeps the {target_audience} awake at night.
-    2. Agitación: Amplify the emotional and practical consequences of this problem.
-    3. Solución: Highlight how the product '{product}' solves this problem and improves their life.
-    4. Acción: Adapt the provided call-to-action '{action}' and craft it in a way that motivates the reader to take immediate steps, using compelling language that encourages them to act.
-    Write a cohesive, engaging, and emotionally resonant narrative tailored to connect with the {target_audience}.
-    """
-    response = model.generate_content([full_prompt])
-    if response and response.parts:
-        text = response.parts[0].text
-        return text.strip()
-    else:
-        raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")
 
-# Función para obtener respuesta del modelo Gemini (A.D.P.)
-def get_adp_response(target_audience, product, action, mood, length):
-    model = genai.GenerativeModel("gemini-1.5-flash")
-    full_prompt = f"""
-    You are a skilled copywriter and storyteller. Create a creative and persuasive story in Spanish following the A.D.P. formula (Antes, Después, Puente). 
-    Do not explicitly label or explain each section (Antes, Después, Puente). Instead, weave these elements naturally into a cohesive, flowing narrative that emotionally connects with the {target_audience}.
-    The story should always be written in the second person (using 'tú') and should have unexpected twists to engage the {target_audience}. 
-    The story should have {length} words, with a {mood} tone. Here are the details:
-
-    1. Antes: Describe the current problem and frustrations that you {target_audience} are facing. Create an immersive situation where the reader can recognize themselves.
-    2. Después: Paint a desired scenario where the problem of {target_audience} is solved and the benefits are clearly evident. Use a surprising twist to show how life can change.
-    3. Puente: Introduce your solution '{product}' as the logical bridge between your frustration and the desired result. Make sure to emphasize the transformation you’ll experience if you take action '{action}'. Conclude with a compelling call to action '{action}' inviting the reader to take immediate steps to improve their situation.
-    Ensure that the story has a creative flow with unexpected turns and keeps the reader hooked, motivating them to act at the end.
-    Write a cohesive, engaging, and emotionally resonant narrative tailored to connect deeply with you, the {target_audience}.
-    """
-    response = model.generate_content([full_prompt])
-    if response and response.parts:
-        text = response.parts[0].text
-        return text.strip()
-    else:
-        raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")
+    if formula_type not in story_formulas:
+        raise ValueError("Fórmula no válida")
 
-# Función para obtener respuesta del modelo Gemini (G.H.A.)
-def get_gha_response(target_audience, product, action, mood, length, story_topic):
-    model = genai.GenerativeModel("gemini-1.5-flash")
-    full_prompt = f"""
-    You are a skilled copywriter and storyteller. Create a persuasive story in Spanish following the G.H.A. formula (Gancho, Historia, Acción). 
-    Do not explicitly label or explain each section (Gancho, Historia, Acción). Instead, weave these elements naturally into a cohesive, flowing narrative that emotionally connects with the {target_audience}.
-    The story should have {length} words, with a {mood} tone. Here are the details:
-    
-    1. **Gancho**: Begin with a hook that immediately grabs the attention of the {target_audience}. It could be a provocative question, an unexpected fact, or a statement that challenges their beliefs or assumptions. The goal is to create curiosity and make them want to read more.
-
-    2. **Historia**: Tell a personal or fictional story based on the following story topic: "{story_topic}". Incorporate the following elements seamlessly into the narrative:
-       - What was the initial situation or challenge the protagonist faced?
-       - What conflict or problem arose and how did it affect them emotionally?
-       - How did they feel and what did they learn from that experience?
-       - What action did they take to resolve the situation or what did they learn in the process?
-       - What was the outcome or lesson learned that could apply to the {target_audience}'s life?
-       Make sure the story is engaging and retains the attention of the reader while providing a meaningful takeaway that resonates with their own struggles or aspirations.
-
-    3. **Acción**: Conclude with a natural call to action '{action}' that invites the reader to take the next step, based on the lesson or transformation in the story. Tie the benefits of the {product} into the CTA, emphasizing how taking action will help them achieve similar results. 
-       For example:
-       - "Lo que aprendí es que la vulnerabilidad te conecta con los demás, y tú también puedes lograrlo si compartes tus experiencias reales. Si deseas experimentar la calma y claridad mental que trae la práctica del yoga, únete a nuestro webinar y descubre cómo el yoga puede ayudarte a gestionar el estrés y mejorar tu bienestar."
-       - "Tras superar mi fracaso, entendí que la mente y el cuerpo están más conectados de lo que pensaba, y tú también puedes descubrir ese equilibrio. Si quieres sentirte más tranquilo, centrado y en control, participa en nuestro webinar de yoga y transforma tu vida."
-       - "Lo que descubrí es que la práctica constante te lleva a un cambio profundo, y tú también puedes lograrlo si decides hacer algo por ti mismo. Únete a nuestra sesión de yoga y aprende cómo liberarte de tensiones, aumentar tu flexibilidad y encontrar la paz interior."
-       Empower the reader by showing them how the benefits of the {product} can lead to a transformation. The call to action should be clear, inviting them to take immediate steps based on the benefits of the {product}.
-
-    Ensure that the entire narrative flows smoothly, making the call to action feel like a natural next step after the story, rather than a forced conclusion.
-    """
-
-    response = model.generate_content([full_prompt])
+    formula_description = story_formulas[formula_type]["description"]
+
+    prompt = f"""{system_prompt}
+
+Como experto copywriter, crea una historia persuasiva en español de {kwargs['length']} palabras con un tono {kwargs['mood']}, siguiendo la estructura:
+
+{formula_description}
+
+Público objetivo: {kwargs['target_audience']}
+Producto/Servicio: {kwargs['product']}
+Llamada a acción: {kwargs['action']}
+"""
+
+    if formula_type == "GHA" and 'story_topic' in kwargs:
+        prompt += f"\nTema de la historia: {kwargs['story_topic']}"
+
+    response = model.generate_content([prompt])
     if response and response.parts:
-        text = response.parts[0].text
-        return text.strip()
-    else:
-        raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")
+        return response.parts[0].text.strip()
+    raise ValueError("No se pudo generar la historia")
 
 # Inicializar la aplicación Streamlit
 st.set_page_config(page_title="Generador de Historias", page_icon=":pencil:", layout="wide")