Update app.py

app.py

@@ -1,4 +1,4 @@
-# Código actualizado con selección de tipos de encabezados
+# Código actualizado para incluir selección de tipos de encabezados en el acordeón
 
 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
@@ -84,24 +84,23 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,  
-        "top_p": 0.85,
-        "top_k": 128,
-        "max_output_tokens": 2048,
+        "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
         "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.",
+        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).'"
     )
 
-    # Crear un mensaje para el modelo que incluye tipos específicos
-    selected_types = random.sample(selected_types, min(number_of_headlines, len(selected_types)))
+    # Crear un mensaje para el modelo que incluye tipos específicos, pero no los menciona en la salida
     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 los siguientes tipos de titulares: " + ", ".join(selected_types) + "."
+        f"Asegúrate de utilizar los siguientes tipos de titulares: {', '.join(selected_types)}."
     )
 
     chat_session = model.start_chat(
@@ -113,8 +112,8 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
         ]
     )
 
-    response = chat_session.send_message("Genera los titulares")
-    return response.text
+    response = chat_session.send_message("Genera los titulares")  # Enviar mensaje para obtener la respuesta
+    return response.text  # Regresar la respuesta directamente
 
 # Configurar la interfaz de usuario con Streamlit
 st.set_page_config(page_title="Enchanted Hooks", layout="wide")
@@ -147,7 +146,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:
@@ -155,13 +154,13 @@ 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 y selección de tipos
+    # Crear acordión para la creatividad
     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:",
+        selected_types = st.multiselect(
+            "Elige los tipos de encabezados que deseas incluir",
             options=list(headline_types.keys()),
-            default=list(headline_types.keys())
+            default=["pregunta", "historia"]
         )
 
     # Botón de enviar
@@ -169,12 +168,10 @@ with col1:
 
 # Mostrar los titulares generados
 if submit:
-    if target_audience and product:
+    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_headline_types
-            )
+            generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature, selected_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>
@@ -184,4 +181,4 @@ if submit:
         except ValueError as e:
             col2.error(f"Error: {str(e)}")
     else:
-        col2.error("Por favor, proporciona el público objetivo y el producto.")
+        col2.error("Por favor, proporciona el público objetivo, el producto y selecciona al menos un tipo de encabezado.")