Update app.py

app.py

@@ -77,7 +77,7 @@ headline_types = {
 }
 
 # Función para generar titulares
-def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_types):
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_types, selected_formula):
     product_mention = get_random_product_mention()
     mention_instruction = get_mention_instruction(product_mention, product)
 
@@ -100,7 +100,8 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
     headlines_instruction = (
         f"Tu tarea es crear {number_of_headlines} ganchos o encabezados titulares llamativos diseñados para {target_audience} "
         f"con el fin de generar interés en {product}. Usa la siguiente mención: {mention_instruction}. "
-        f"Asegúrate de utilizar los siguientes tipos de titulares: {', '.join(selected_types)}."
+        f"Asegúrate de utilizar los siguientes tipos de titulares: {', '.join(selected_types)}. "
+        f"Y usa la fórmula de titular seleccionada: {selected_formula}"
     )
 
     chat_session = model.start_chat(
@@ -199,7 +200,7 @@ if submit:
     if target_audience and product and selected_types:
         try:
             # Obtener la respuesta del modelo
-            generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature, selected_types)
+            generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature, selected_types, selected_formula)
             col2.markdown(f"""
                 <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">
                     <h4>Observa la magia en acción:</h4>