Update app.py

app.py

@@ -7,24 +7,100 @@ load_dotenv()
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Función para obtener respuesta del modelo Gemini
-def get_gemini_response(target_audience, product, action, mood, length):
+# Función para obtener respuesta del modelo Gemini (P.A.S.A)
+def get_pasa_response(target_audience, product, action, mood, length):
     model = genai.GenerativeModel("gemini-1.5-flash")
-    
-    # Crear el prompt para el modelo
     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}.
-"""
+    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 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 have {length} words, with a {mood} tone. Here are the details:
+    1. **Antes**: Describe the current problem or frustration that the {target_audience} is facing. Reflect on their daily struggles, fears, and obstacles. For example, you can show how they may feel like their efforts are going unnoticed, creating a sense of frustration or helplessness. Think about these questions: 
+    - What frustrations, fears, or problems is your audience facing? 
+    - What is preventing them from achieving their goals? 
+    - How does it feel to live with this problem? 
+    Example: "Pasas horas escribiendo contenido, pero al final sientes que nadie lo lee o conecta contigo. Es como gritar en un desierto, esperando que alguien escuche."
+
+    2. **Después**: Now, paint a picture of the ideal scenario where the problem is resolved. Show how life would be improved, focusing on the benefits your audience would experience. Integrate questions like:
+    - How would life be for your audience if the problem was solved? 
+    - What specific benefits would they gain? 
+    - What positive emotions would they feel? 
+    Example: "Ahora imagina que cada vez que publicas una historia, tu audiencia reacciona, comenta y comparte. Tus mensajes se llenan de prospectos interesados, y tus ventas crecen sin sentir que estás forzando nada."
+
+    3. **Puente**: Connect the "Antes" and "Después" by presenting the product '{product}' as the solution. Explain how this product bridges the gap between the frustration and the ideal outcome, and how it can improve their situation. The solution should feel logical and accessible. Think about these questions: 
+    - How does your solution specifically resolve the problem? 
+    - What makes your solution unique or effective? 
+    - What simple steps should the reader take to achieve this transformation? 
+    At the end, adapt the provided call-to-action '{action}' to encourage the reader to take immediate action, emphasizing that this is the logical next step to achieve the benefits and desired transformation.
+    Example: "La clave está en aprender a contar historias que conecten emocionalmente con tu audiencia. Con la fórmula A.D.P., puedes guiar a tus lectores desde su frustración actual hasta el resultado que desean. Hoy te enseñaré exactamente cómo hacerlo en mi clase en vivo a las 05:00 pm."
 
+    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 (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**: Start with an attention-grabbing hook that immediately captures the {target_audience}'s interest. It can be a question, a surprising fact, or a statement that challenges their beliefs. Some key elements to consider are: 
+    - What surprising or intriguing thing can you ask or say that makes your audience think?
+    - What belief can you challenge or what mistake can you highlight to grab their attention?
+    - Example: "¿Qué pasaría si tus fracasos fueran la clave para tus mayores éxitos?"
+    - Example: "Lo que nadie te dice: los errores venden más que los aciertos."
+    - Example: "¿Recuerdas tu peor fracaso? Es justo lo que necesitas para conectar con tu audiencia."
+
+    2. **Historia**: Tell an engaging and emotional story that resonates with the {target_audience}. This can be a personal experience, a fictional story, a movie, a fable, or any narrative that interests the reader and connects with their situation. It should have a meaningful lesson or takeaway. Consider the following questions to craft the story:
+    - ¿Cuál era tu situación inicial?
+    - ¿Qué problema o conflicto surgió?
+    - ¿Cómo te sentiste ante ese problema y qué aprendiste?
+    - ¿Qué hiciste para resolverlo o qué aprendiste?
+    - ¿Cuál fue el resultado positivo o la lección que aprendiste?
+    You can use a personal experience, a movie scene, a fictional character, or even create an imaginative story that resonates with the audience. Example: 
+    "Era mi primera venta importante y estaba emocionado. El día había llegado, y con él, mi gran oportunidad. Había preparado todo con tanto esmero que casi podía sentir el éxito en el aire. Pero cuando llegó el momento… mi cliente no apareció. Me quedé allí, solo, mirando la pantalla, sintiendo una presión en el pecho. ¿Qué había hecho mal? La duda me invadió, y por un momento, sentí que todo había sido en vano. Me costó entenderlo, pero finalmente lo vi: mi mensaje no resonaba con sus necesidades, no estaba hablando su idioma. Decidí cambiar el enfoque. Ya no quise seguir ocultando mis fracasos. Empecé a ser más honesto, a compartir mis inseguridades y mis aprendizajes, a mostrar que detrás de los resultados, había un ser humano que también había caído y se había levantado. Desde ese día, mis ventas mejoraron, pero lo más importante es que comencé a conectar de verdad con las personas. Ahora sé que la autenticidad y la vulnerabilidad son las claves para generar confianza. Y eso lo cambio todo."
+
+    3. **Acción**: Conclude with a strong and motivating call-to-action, linking the story's lesson with a step that the reader can take immediately. Consider these questions to craft the action:
+    - What can your audience do right now to achieve similar results?
+    - How can they benefit by following the steps you learned in your story?
+    - What makes the action you're suggesting the logical next step? 
+    - Example: "Lo que aprendí es que ser honesto en tus historias conecta más con la gente, y tú también puedes lograrlo si compartes tus experiencias reales. Si quieres saber cómo hacerlo, únete a mi clase en vivo hoy a las 05:00 pm."
+    You may adapt this phrase in a natural way to ensure the message flows smoothly and doesn't sound repetitive.
+
+    4. **Producto**: Use the product '{product}' as part of the solution in the story. Explain how it will help solve the problem your target audience is facing.
+    5. **Llamado a la Acción**: Create a clear and compelling call to action that motivates the {target_audience} to take immediate action based on the lesson from the story. Use the provided action '{action}' but make it sound natural in the context of the story.
+    6. **Público Objetivo**: Write the story specifically for the {target_audience}. Consider their needs, pain points, and aspirations while crafting the story.
+    7. **De qué quieres que trate la historia**: Please explain what you would like the story to focus on. It could be a personal experience, a movie, a fictional character, or anything that resonates with your audience's situation. The key is that it should grab their attention and connect emotionally with them.
+    
+    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()
@@ -32,33 +108,13 @@ Write a cohesive, engaging, and emotionally resonant narrative tailored to conne
         raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")
 
 # Inicializar la aplicación Streamlit
-st.set_page_config(page_title="Generador P.A.S.A.", page_icon=":pencil:", layout="wide")
+st.set_page_config(page_title="Generador de Historias", page_icon=":pencil:", layout="wide")
 
-st.markdown("<h1 style='text-align: center;'>Generador de Historias P.A.S.A.</h1>", unsafe_allow_html=True)
+st.markdown("<h1 style='text-align: center;'>Generador de Historias</h1>", unsafe_allow_html=True)
 st.markdown("<h3 style='text-align: center;'>Crea historias persuasivas con inteligencia artificial, diseñadas para conectar emocionalmente con tu audiencia.</h3>", unsafe_allow_html=True)
 
-# CSS personalizado para el botón
-st.markdown("""
-    <style>
-    div.stButton > button {
-        background-color: #FFCC00; 
-        color: black; 
-        width: 90%;
-        height: 60px;
-        font-weight: bold;
-        font-size: 22px; 
-        text-transform: uppercase; 
-        border: 1px solid #000000;
-        border-radius: 8px;
-        display: block;
-        margin: 0 auto;
-    }
-    div.stButton > button:hover {
-        background-color: #FFD700; 
-        color: black;
-    }
-    </style>
-    """, unsafe_allow_html=True)
+# Crear el selector para elegir la fórmula
+formula = st.radio("Selecciona la fórmula para generar tu historia:", ["P.A.S.A.", "A.D.P.", "G.H.A."])
 
 # Crear las columnas para el diseño (40% para la izquierda, 60% para la derecha)
 col1, col2 = st.columns([2, 3])
@@ -74,6 +130,9 @@ with col1:
         mood = st.selectbox("Tono de la historia:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
         length = st.slider("Longitud de la historia (palabras):", min_value=50, max_value=200, value=150, step=10)
 
+    if formula == "G.H.A.":
+        story_topic = st.text_area("De qué quieres que trate la historia", placeholder="Explica si hay algo específico sobre lo que te gustaría contar (puede ser una vivencia personal, película, cuento, personaje ficticio, etc.).")
+
     # Botón para generar contenido
     submit = st.button("Generar mi historia")
 
@@ -82,9 +141,20 @@ with col2:
     if submit:
         if target_audience and product and action:
             try:
-                response = get_gemini_response(target_audience, product, action, mood, length)
-                st.subheader("Historia generada:")
-                st.write(response)
+                if formula == "P.A.S.A.":
+                    response = get_pasa_response(target_audience, product, action, mood, length)
+                elif formula == "A.D.P.":
+                    response = get_adp_response(target_audience, product, action, mood, length)
+                elif formula == "G.H.A.":
+                    if story_topic:
+                        response = get_gha_response(target_audience, product, action, mood, length, story_topic)
+                    else:
+                        st.error("Por favor, completa todos los campos requeridos para la fórmula G.H.A.")
+                        response = ""
+                
+                if response:
+                    st.subheader("Historia generada:")
+                    st.write(response)
             except ValueError as e:
                 st.error(f"Error: {str(e)}")
         else: