Update app.py

app.py

@@ -26,6 +26,54 @@ 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
+headline_types = {
+    "pregunta": [
+        "¿Qué sorprendente secreto revela el éxito detrás de los grandes emprendedores?",
+        "¿Estás haciendo estos errores comunes en tu rutina diaria?",
+    ],
+    "historia": [
+        "La increíble transformación de alguien que decidió salir de su zona de confort.",
+        "El viaje inesperado que llevó a un joven a descubrir su pasión por la música.",
+    ],
+    "afirmación": [
+        "La clave para una vida más feliz está en tus decisiones diarias.",
+        "Cada fracaso es una oportunidad disfrazada para aprender y crecer.",
+    ],
+    "llamado_a_la_accion": [
+        "Descubre cómo mejorar tu productividad en solo una semana.",
+        "Únete a nuestro curso y transforma tu carrera profesional hoy mismo.",
+    ],
+    "consejo": [
+        "5 consejos para mejorar tu concentración mientras trabajas desde casa.",
+        "Aprende a gestionar tu tiempo de manera efectiva con estas simples estrategias.",
+    ],
+    "frase_inspiradora": [
+        "El único modo de hacer un gran trabajo es amar lo que haces.",
+        "La mejor forma de predecir el futuro es crearlo.",
+    ],
+    "dato_impresionante": [
+        "¿Sabías que el 70% de las personas no están satisfechas con su trabajo actual?",
+        "Los estudios muestran que dedicar al menos 30 minutos al día a la lectura puede aumentar tu productividad.",
+    ],
+    "mitologia": [
+        "La verdad detrás del mito de que solo los jóvenes pueden ser emprendedores exitosos.",
+        "Desmitificando la creencia de que necesitas un título universitario para tener éxito.",
+    ],
+    "numerico": [
+        "10 formas sorprendentes de aumentar tu productividad en el trabajo.",
+        "7 hábitos que transformarán tu vida en solo un mes.",
+    ],
+    "impactante": [
+        "¡Descubre el oscuro secreto que las empresas no quieren que sepas!",
+        "La verdad escalofriante sobre los ingredientes de tus alimentos favoritos.",
+    ],
+    "amarillista": [
+        "¡No podrás creer lo que esta persona hizo para hacerse rica de la noche a la mañana!",
+        "¡Alerta! Estos hábitos cotidianos podrían estar arruinando tu salud sin que te des cuenta.",
+    ],
+}
+
 # Función para generar titulares
 def generate_headlines(number_of_headlines, target_audience, product, temperature):
     product_mention = get_random_product_mention()
@@ -33,7 +81,7 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
 
     # Crear la configuración del modelo
     generation_config = {
-        "temperature": temperature,  # Usar el valor del slider aquí
+        "temperature": temperature,
         "top_p": 0.95,
         "top_k": 64,
         "max_output_tokens": 2048,
@@ -45,7 +93,18 @@ 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=[
             {
@@ -111,17 +170,17 @@ with col1:
     
     # Mostrar los titulares generados
 if st.button("Generar Titulares"):
-        if target_audience and product:
-            try:
-                # Obtener la respuesta del modelo
-                generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature)
-                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>
-                        <p>{generated_headlines}</p>
-                    </div>
-                """, unsafe_allow_html=True)
-            except ValueError as e:
-                col2.error(f"Error: {str(e)}")
-        else:
-            col2.error("Por favor, proporciona el público objetivo y el producto.")
+    if target_audience and product:
+        try:
+            # Obtener la respuesta del modelo
+            generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature)
+            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>
+                    <p>{generated_headlines}</p>
+                </div>
+            """, unsafe_allow_html=True)
+        except ValueError as e:
+            col2.error(f"Error: {str(e)}")
+    else:
+        col2.error("Por favor, proporciona el público objetivo y el producto.")