Update app.py

app.py

@@ -10,7 +10,7 @@ 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, cta_type, benefit):
+def get_gemini_response(input_prompt, genre, length, language, mood, target_audience, cta_type):
     if not input_prompt:
         return "Por favor, escribe un mensaje para generar contenido."
     
@@ -21,7 +21,9 @@ def get_gemini_response(input_prompt, genre, length, language, mood, target_audi
 
     Target Audience: {target_audience}
     CTA Type: {cta_type}
-    Main Benefit: {benefit}
+
+    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
@@ -33,7 +35,7 @@ def get_gemini_response(input_prompt, genre, length, language, mood, target_audi
        - 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.
+    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. 
@@ -74,6 +76,7 @@ st.markdown("""
 col1, col2 = st.columns([1, 1])
 
 # Columna izquierda para inputs
+# Remove the benefit input field
 with col1:
     # Entrada de texto principal
     input_prompt = st.text_area("Escribe de qué quieres que trate la historia:", placeholder="Escribe aquí tu idea...")
@@ -81,7 +84,7 @@ 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...")
     
-    # Replace product_type with CTA type
+    # CTA type only
     cta_type = st.selectbox("Tipo de llamado a la acción:", [
         "Comprar producto",
         "Descargar guía PDF",
@@ -106,7 +109,7 @@ with col2:
     if generate_button:
         response = get_gemini_response(
             input_prompt, genre, length, language,
-            mood, target_audience, cta_type, benefit
+            mood, target_audience, cta_type
         )
         st.subheader("Contenido generado:")
         st.write(response)