Update app.py

app.py

@@ -26,7 +26,8 @@ def get_mention_instruction(product_mention, product):
         return f"Introduce el producto '{product}' usando una metáfora, conectándolo simbólicamente a la solución que necesita el lector."
     return ""
 
-# Tipos de titulares
+# Ejemplos de titulares por tipo
+
 headline_types = {
     "pregunta": [
         "¿Qué sorprendente secreto revela el éxito detrás de los grandes emprendedores?",
@@ -81,7 +82,7 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
 
     # Crear la configuración del modelo
     generation_config = {
-        "temperature": temperature,
+        "temperature": temperature,  
         "top_p": 0.95,
         "top_k": 64,
         "max_output_tokens": 2048,
@@ -93,18 +94,7 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
         generation_config=generation_config,
         system_instruction="Eres un copywriter de clase mundial, con experiencia en la creación de ganchos, titulares y líneas de asunto que capturan la atención de inmediato. Tu habilidad radica en comprender profundamente las emociones, deseos y desafíos de una audiencia específica."
     )
-    
-    headlines = []
-    for _ in range(number_of_headlines):
-        # Seleccionar un tipo de titular aleatorio
-        headline_type = random.choice(list(headline_types.keys()))
-        # Seleccionar un ejemplo aleatorio del tipo seleccionado
-        headline_example = random.choice(headline_types[headline_type])
-        headlines.append(headline_example)
-
-    # Convertir la lista de titulares en un string
-    headline_string = "\n".join(headlines)
-    
+
     chat_session = model.start_chat(
         history=[
             {
@@ -113,13 +103,7 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
                     f"Tu tarea es crear {number_of_headlines} ganchos o encabezados titulares llamativos diseñados para {target_audience} con el fin de generar interés en {product}. "
                     f"Usa la siguiente mención: {mention_instruction}. "
                     "Los ganchos deben ser de este tipo: "
-                    "El enigma detrás de la leyenda: ¿qué misterios se ocultan en las historias que siempre nos han cautivado?"
-                    "¿Te has preguntado alguna vez qué sorprendente dato podría transformar por completo tu forma de ver el mundo?"
-                    "¿El mejor lugar para guardar los medicamentos? El botiquín del baño, ¿verdad? Incorrecto. En realidad, es el peor. ¡Descubre la verdad en la página 10!"
-                    "El tiempo verbal ideal que hace que tus clientes sientan que ya han comprado de ti."
-                    "Un sistema sencillo para redactar textos persuasivos sin tener que convencer a nadie de que compre."
-                    "La verdad que nunca te contaron en la escuela, ni en casa, sobre cómo ganar dinero con la música."
-                    "¿Cuándo es el momento adecuado para confesarle a una chica que te gusta? Si no lo dices en ese instante, despídete de la posibilidad de conocerla de una manera más íntima."
+                    + " ".join([f"{tipo}: " + ", ".join(ejemplos) for tipo, ejemplos in headline_types.items()])
                 ],
             },
         ]
@@ -159,7 +143,7 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Crear columnas
-col1, col2 = st.columns([1, 2])  # 1: tamaño de la columna izquierda, 2: tamaño de la columna derecha
+col1, col2 = st.columns([1, 2])  
 
 # Columnas de entrada
 with col1:
@@ -167,8 +151,8 @@ with col1:
     product = st.text_input("¿Qué producto tienes en mente?", placeholder="Ejemplo: Curso de Inglés")
     number_of_headlines = st.selectbox("Número de Titulares", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
     temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
-    
-    # Mostrar los titulares generados
+
+# Mostrar los titulares generados
 if st.button("Generar Titulares"):
     if target_audience and product:
         try: