Update app.py

app.py

@@ -10,24 +10,31 @@ 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, language, mood, target_audience, product_type):
+def get_gemini_response(input_prompt, genre, length, language, mood, target_audience, cta_type, benefit):
     if not input_prompt:
         return "Por favor, escribe un mensaje para generar contenido."
     
     model = genai.GenerativeModel('gemini-2.0-flash')
     full_prompt = f"""
-    You are a professional storyteller. Write a {mood} {genre} in {language} using a natural, engaging, and conversational style. The story should flow seamlessly and feel personal, not forced.  
-    Use approximately {length} words (but prioritize quality over exact word count).  
-    
-    **Structure:**  
-    1. **Introduce a relatable problem**: Describe a scenario that captures the attention of {target_audience}. Use emotions and common experiences in their lives.  
-    2. **Develop the story**: Expand on the problem with details, showing the difficulties the protagonist faces. Use a natural and immersive narrative.  
-    3. **Consequences of inaction**: What happens if the problem is not solved? Build urgency without sounding forced.  
-    4. **Organic solution**: Introduce {product_type} as a logical option in the story, without making it sound like an advertisement. The story should naturally lead to the solution.  
-    5. **Emotional or motivational ending**: End with a reflection or an implicit call to action.  
+    You are a professional storyteller. Write a {mood} {genre} in {language} using a natural, engaging, and conversational style. 
+    The story should be about "{input_prompt}" and use exactly {length} words.
+
+    Target Audience: {target_audience}
+    CTA Type: {cta_type}
+    Main Benefit: {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 specific call to action based on the type:
+       - For product purchase: Mention how the product helps achieve {benefit}
+       - For PDF guide: Highlight how the guide provides steps to {benefit}
+       - For webinar: Emphasize how attending will teach them to {benefit}
+
+    Make the CTA feel natural and focus on the value for the reader.
     
-    The story should be about "{input_prompt}". Ensure the topic is naturally integrated into the narrative.
-     
     ### **Ejemplo de historia bien escrita:**  
     En una lejana galaxia, el maestro Xylo dictaba clases en la Tierra. 
 
@@ -48,7 +55,6 @@ def get_gemini_response(input_prompt, genre, length, language, mood, target_audi
     Inscríbete hoy aquí.
     
     **Importante:**  
-    - **No menciones directamente {product_type} en el texto**. En su lugar, intégralo de forma sutil y natural en la narrativa.  
     - **Haz que la historia sea humana y emocional**, evitando frases promocionales evidentes.  
     """
     response = model.generate_content([full_prompt])
@@ -74,7 +80,16 @@ with col1:
     
     # Campos de texto libre
     target_audience = st.text_input("Público Objetivo:", placeholder="Ejemplo: Profesionales de 25-35 años interesados en tecnología...")
-    product_type = st.text_input("Producto o Servicio:", placeholder="Ejemplo: Curso de marketing digital...")
+    
+    # Replace product_type with CTA type
+    cta_type = st.selectbox("Tipo de llamado a la acción:", [
+        "Comprar producto",
+        "Descargar guía PDF",
+        "Asistir a webinar"
+    ])
+    
+    # Add benefit input
+    benefit = st.text_input("Beneficio principal:", placeholder="Ejemplo: Aumentar ventas en 30 días, Aprender técnicas avanzadas...")
     
     # Opciones avanzadas en acordeón
     with st.expander("Opciones avanzadas"):
@@ -90,8 +105,8 @@ with col1:
 with col2:
     if generate_button:
         response = get_gemini_response(
-            input_prompt, genre, length, language, 
-            mood, target_audience, product_type
+            input_prompt, genre, length, language,
+            mood, target_audience, cta_type, benefit
         )
         st.subheader("Contenido generado:")
         st.write(response)