Update app.py

app.py

@@ -2,6 +2,7 @@ from dotenv import load_dotenv
 import streamlit as st
 import os
 import google.generativeai as genai
+from story_formulas import story_formulas
 
 # Cargar variables de entorno
 load_dotenv()
@@ -10,67 +11,30 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Función para obtener la respuesta del modelo Gemini
-def get_gemini_response(input_prompt, genre, length, mood, target_audience, cta_type):
+def get_gemini_response(input_prompt, formula_type, length, mood, target_audience, cta_type):
     if not input_prompt:
         return "Por favor, escribe un mensaje para generar contenido."
     
+    formula = story_formulas[formula_type]
+    
     model = genai.GenerativeModel('gemini-2.0-flash')
     full_prompt = f"""
-    You are a creative storyteller. Write a {mood} {genre} in Spanish about "{input_prompt}" using everyday words and natural conversation. 
+    You are a creative storyteller. Write a {mood} story in Spanish about "{input_prompt}" using everyday words and natural conversation. 
     Use exactly {length} words.
 
     Target Audience: {target_audience}
     Call to Action: {cta_type}
+    Story Formula: {formula_type}
 
-    First, identify the main benefit that would resonate most with {target_audience} based on the story topic and CTA type.
-    Then, write a story that naturally leads to this benefit.
-
-    Structure:
-    1. Introduce a relatable problem that captures the audience's attention
-    2. Develop the story with natural narrative and emotional connection
-    3. Show consequences of inaction
-    4. Present the solution organically
-        5. Close with a natural invitation to {cta_type} that shows how it will transform their situation
-       - For product purchase: Show how buying transforms their situation
-       - For PDF guide: Highlight how the guide provides the knowledge they need
-       - For webinar: Emphasize how the live training will help them succeed
+    {formula["description"]}
 
     Keep in mind:
     - Write for {target_audience}
-    - Use simple, clear language that everyone can understand
-    - Avoid technical terms or complex words
+    - Use simple, clear language
     - Tell the story in short paragraphs
     - End with a natural {cta_type} invitation
-    
-    Example format:
-    Make the CTA feel natural and focus on the transformation they'll experience.
-    
-    ### **Ejemplo de historia bien escrita:**  
-    En una lejana galaxia, el maestro Xylo dictaba clases en la Tierra. 
-
-    Sus métodos eran monótonos, y sus alumnos bostezaban sin parar. 
-        
-    “¿Por qué no prestan atención?”, se preguntaba.  
-        
-    Frustrado, investigó y descubrió que el problema no eran los niños, sino sus clases aburridas. 
-    
-    Por eso decidió cambiar.  
-    
-    Tomó un curso online para hacer sus lecciones dinámicas y sorprendentes. 
-    
-    Pronto, sus alumnos rieron, participaron y aprendieron con entusiasmo.  
-    
-    Si Xylo pudo transformar sus clases, tú también. 
-    
-    Inscríbete hoy aquí.
-    
-    **Importante:**  
-    - **Haz que la historia sea humana y emocional**, evitando frases promocionales evidentes.  
-    - Format the output in clear, separate paragraphs with line breaks
-    - Only write {genre}, anything else.
     """
     response = model.generate_content([full_prompt])
-    
     return response.parts[0].text if response and response.parts else "Error al generar contenido."
 
 # Configurar la aplicación Streamlit
@@ -90,16 +54,17 @@ col1, col2 = st.columns([1, 1])
 with col1:
     # Entrada de texto principal
     input_prompt = st.text_area("Escribe de qué quieres que trate la historia:", placeholder="Escribe aquí tu idea...")
+    target_audience = st.text_input("Público Objetivo:", placeholder="Ejemplo: Profesionales de 25-35 años...")
+    cta_type = st.text_input("Llamado a la acción:", placeholder="Ejemplo: Curso de marketing digital...")
     
-    # Campos de texto libre
-    target_audience = st.text_input("Público Objetivo:", placeholder="Ejemplo: Profesionales de 25-35 años interesados en tecnología...")
-    
-    # CTA como campo de texto libre
-    cta_type = st.text_input("Llamado a la acción:", placeholder="Ejemplo: Curso de marketing digital, Guía para emprendedores, Webinar gratuito...")
-    
-    # Opciones avanzadas en acordeón
     with st.expander("Opciones avanzadas"):
-        genre = st.selectbox("Tipo de texto:", ["Historia", "Poema", "Cita", "Ensayo"])
+        formula_type = st.selectbox(
+            "Fórmula narrativa:",
+            options=list(story_formulas.keys()),
+            format_func=lambda x: x
+        )
+        st.info(story_formulas[formula_type]["description"])
+        
         length = st.slider("Longitud del texto (palabras):", 
                           min_value=100, 
                           max_value=150,