Update app.py

app.py

@@ -40,13 +40,15 @@ def get_mention_instruction(product_mention, product):
     return ""
 
 # Función para generar postdata (refuerzo del CTA)
-def generate_postdata(target_audience, product, call_to_action, temperature):
+def generate_postdata(target_audience, product, call_to_action, temperature, theme_or_focus):
     # Instrucción para la generación de postdata
     postdata_instruction = (
         f"Eres un experto copywriter especializado en escribir mensajes persuasivos. "
         f"Tu tarea es generar una postdata que refuerce el llamado a la acción '{call_to_action}' para el público objetivo '{target_audience}', "
         f"relacionado con el producto '{product}'. La postdata debe comenzar siempre con 'P.D.' y debe reforzar la urgencia o deseo de tomar acción."
-        "La postdata debe ser breve, clara y persuasiva, y no debe ser una repetición del CTA. Ejemplos de postdata incluyen:\n"
+        f"La postdata debe ser breve, clara y persuasiva, y no debe ser una repetición del CTA. "
+        f"El tema o enfoque para la postdata es '{theme_or_focus}'."
+        "Ejemplos de postdata incluyen:\n"
         "P.D. No te pierdas la oportunidad de transformar tu futuro, ¡te esperamos en el evento!\n"
         "P.D. Recuerda que las plazas son limitadas, ¡inscríbete ahora antes de que se agoten!"
     )
@@ -119,7 +121,7 @@ def generate_ctas(number_of_ctas, target_audience, product, call_to_action, temp
         "temperature": temperature,  
         "top_p": 0.85,       
         "top_k": 128,        
-        "max_output_tokens": 8196,
+        "max_output_tokens": 2048,
         "response_mime_type": "text/plain",
     }
 
@@ -183,6 +185,7 @@ with col1:
     target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Estudiantes Universitarios")
     product = st.text_input("¿Qué producto tienes en mente?", placeholder="Ejemplo: Curso de Inglés")
     call_to_action = st.text_input("¿Qué acción deseas que tomen?", placeholder="Ejemplo: Inscribirse al curso")
+    theme_or_focus = st.text_input("Tema o enfoque para la postdata", placeholder="Ejemplo: urgencia, beneficio, descuento")
     number_of_ctas = st.selectbox("Número de llamados a la acción", options=[1, 2, 3, 4, 5], index=2)
     temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=0.8, step=0.1)
 
@@ -196,7 +199,7 @@ if submit:
             intro_text = generate_intro_text()  # Generar el texto introductorio
             generated_ctas = generate_ctas(number_of_ctas, target_audience, product, call_to_action, temperature)
             # Generar la postdata
-            postdata = generate_postdata(target_audience, product, call_to_action, temperature)
+            postdata = generate_postdata(target_audience, product, call_to_action, temperature, theme_or_focus)
             
             col2.markdown(f"""
                 <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">