Update app.py

app.py

@@ -1,4 +1,4 @@
-# Código original actualizado
+# Código actualizado con selección de tipos de encabezados
 
 from dotenv import load_dotenv
 import streamlit as st
@@ -15,7 +15,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Función para obtener una mención del producto de manera probabilística
 def get_random_product_mention():
     mentions = ["Directa", "Indirecta", "Metafórica"]
-    probabilities = [0.34, 0.33, 0.33]  
+    probabilities = [0.34, 0.33, 0.33]
     return random.choices(mentions, probabilities)[0]
 
 # Crear la instrucción de mención basada en la opción seleccionada
@@ -29,7 +29,6 @@ def get_mention_instruction(product_mention, product):
     return ""
 
 # Ejemplos de titulares por tipo
-
 headline_types = {
     "pregunta": [
         "¿Qué sorprendente secreto revela el éxito detrás de los grandes emprendedores?",
@@ -78,44 +77,44 @@ headline_types = {
 }
 
 # Función para generar titulares
-def generate_headlines(number_of_headlines, target_audience, product, temperature):
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_types):
     product_mention = get_random_product_mention()
     mention_instruction = get_mention_instruction(product_mention, product)
 
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,  
-        "top_p": 0.85,        # Considerar un poco menos de palabras probables
-        "top_k": 128,        # Aumentar las palabras candidatas para más variedad
-        "max_output_tokens": 2048,  # Mantenerlo igual, pero puedes aumentar si deseas más detalle
+        "top_p": 0.85,
+        "top_k": 128,
+        "max_output_tokens": 2048,
         "response_mime_type": "text/plain",
-}
+    }
 
     model = genai.GenerativeModel(
         model_name="gemini-1.5-flash",
         generation_config=generation_config,
-        system_instruction="You are a world-class copywriter, with expertise in crafting hooks, headlines, and subject lines that immediately capture the reader's attention, prompting them to open the email or continue reading. Your skill lies in deeply understanding the emotions, desires, and challenges of a specific audience. You are also an expert in creating benefits that connect symptoms with problems, allowing you to design personalized strategies that resonate and motivate action. You know how to use proven structures to attract your target audience, generating interest and creating a powerful connection. Your task is to generate unusual, creative, and fascinating subject lines or headlines that spark curiosity and encourage the reader to engage further. Respond in Spanish and use a numbered list format. Important: Only answer with subject lines, never include explanations or categories, like this: 'La leyenda del padre soltero: Dice que nunca hay tiempo suficiente. El yoga te enseña a usar mejor el tiempo que tienes, incluso cuando te parece imposible (este bullet es curioso).'"
+        system_instruction="You are a world-class copywriter, with expertise in crafting hooks, headlines, and subject lines that immediately capture the reader's attention, prompting them to open the email or continue reading. Your skill lies in deeply understanding the emotions, desires, and challenges of a specific audience. You are also an expert in creating benefits that connect symptoms with problems, allowing you to design personalized strategies that resonate and motivate action. You know how to use proven structures to attract your target audience, generating interest and creating a powerful connection. Your task is to generate unusual, creative, and fascinating subject lines or headlines that spark curiosity and encourage the reader to engage further. Respond in Spanish and use a numbered list format. Important: Only answer with subject lines, never include explanations or categories.",
     )
 
-    # Crear un mensaje para el modelo que incluye tipos específicos, pero no los menciona en la salida
-    selected_types = random.sample(list(headline_types.keys()), min(number_of_headlines, len(headline_types)))
+    # Crear un mensaje para el modelo que incluye tipos específicos
+    selected_types = random.sample(selected_types, min(number_of_headlines, len(selected_types)))
     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}. "
-        "Asegúrate de utilizar diferentes tipos de titulares sin mencionar sus tipos en la salida."
+        "Asegúrate de utilizar los siguientes tipos de titulares: " + ", ".join(selected_types) + "."
     )
 
     chat_session = model.start_chat(
         history=[
-        {
-            "role": "user",
-            "parts": [headlines_instruction],
-        },
+            {
+                "role": "user",
+                "parts": [headlines_instruction],
+            },
         ]
     )
 
-    response = chat_session.send_message("Genera los titulares")  # Enviar mensaje para obtener la respuesta
-    return response.text  # Regresar la respuesta directamente
+    response = chat_session.send_message("Genera los titulares")
+    return response.text
 
 # Configurar la interfaz de usuario con Streamlit
 st.set_page_config(page_title="Enchanted Hooks", layout="wide")
@@ -148,7 +147,7 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Crear columnas
-col1, col2 = st.columns([1, 2])  
+col1, col2 = st.columns([1, 2])
 
 # Columnas de entrada
 with col1:
@@ -156,9 +155,14 @@ 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)
 
-    # Crear acordión para la creatividad
+    # Crear acordión para la creatividad y selección de tipos
     with st.expander("Personaliza tus titulares"):
         temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+        selected_headline_types = st.multiselect(
+            "Selecciona los tipos de encabezados que deseas generar:",
+            options=list(headline_types.keys()),
+            default=list(headline_types.keys())
+        )
 
     # Botón de enviar
     submit = st.button("Generar Titulares")
@@ -168,7 +172,9 @@ if submit:
     if target_audience and product:
         try:
             # Obtener la respuesta del modelo
-            generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature)
+            generated_headlines = generate_headlines(
+                number_of_headlines, target_audience, product, temperature, selected_headline_types
+            )
             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>