Update app.py

app.py

@@ -21,12 +21,19 @@ def get_model(temperature):
     }
     return genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
 
-def generate_benefits(number_of_benefits, target_audience, product, temperature, selected_formula):
+# Update the generate_benefits function to include selected_angle
+def generate_benefits(number_of_benefits, target_audience, product, temperature, selected_formula, selected_angle):
     if not target_audience or not product:
         return "Por favor, completa todos los campos requeridos."
     
     model = get_model(temperature)
-    benefits_instruction = create_instruction(number_of_benefits, target_audience, product, selected_formula)
+    benefits_instruction = create_instruction(
+        number_of_benefits, 
+        target_audience, 
+        product, 
+        selected_formula,
+        selected_angle
+    )
     
     response = model.generate_content([benefits_instruction], generation_config={"temperature": temperature})
     return response.parts[0].text if response and response.parts else "Error generating content."
@@ -89,7 +96,7 @@ with col1:
     submit = st.button("Generar Beneficios")
 
 # Mostrar los beneficios generados
-# Update the submit section
+# Update the submit section to include selected_angle
 if submit:
     if target_audience and product and selected_formula:
         with st.spinner('Generando beneficios...'):
@@ -98,7 +105,8 @@ if submit:
                 target_audience, 
                 product, 
                 temperature, 
-                selected_formula
+                selected_formula,
+                selected_angle
             )
             if not isinstance(generated_benefits, str):
                 col2.error("Error al generar beneficios")