Update app.py

app.py

@@ -1,4 +1,4 @@
-# Código actualizado con los ángulos añadidos
+# Código actualizado con las modificaciones solicitadas para las fórmulas
 
 from dotenv import load_dotenv
 import streamlit as st
@@ -80,109 +80,124 @@ headline_formulas = {
     }
 }
 
-# Añadir los ángulos al acordeón
-angles = {
-    "autoridad": {
-        "description": "La idea que las personas siguen lo que dicen los expertos reconocidos, los doctores y sus batas blancas, los certificados a la vista, los títulos académicos y nobiliarios, etc.",
-        "examples": [
-            "El Secreto del Suplemento Usado Por Los Navy SEALs Estadounidenses Para Combatir El Dolor, Heridas, Grasa Corporal, y Problemas Cardiovasculares",
-            "La doctora Daniela Bermudez experta en Tanatología explica cómo la resiliencia nos ayuda a superar la pérdida de nuestros seres queridos",
-            "El remedio natural para curar la calvicie de uno de los herbolarios más reconocidos en Harvard.",
-            "5 consejos de la reconocida escritora J. K. Rowling para iniciarse como escritor de fantasía."
-        ]
-    },
-    "curiosidad": {
-        "description": "Este elemento siempre atrae a cualquier persona, lo deja con la inquietud de querer investigar más.",
-        "examples": [
-            "Lo Que Nunca Debes Comer En Un Avión",
-            "20 palabras prohibidas que nunca debes mencionar frente a tus hijos.",
-            "Escalofriante costumbre de los monjes tibetanos que nunca debes hacer.",
-            "Se casa con su jefe después de haber sido humillada 5 años en el trabajo."
-        ]
-    },
-    "advertencia": {
-        "description": "Generan una sensación de urgencia o peligro, que provoque una respuesta inmediata.",
-        "examples": [
-            "CUIDADO: Si tu cuerpo está presentando algunos de estos síntomas... tienes que ir con el médico",
-            "ADVERTENCIA: lo que las agencias de viajes no quieren que sepas cuando te vas de viaje.",
-            "¡DETENTE! y descubre lo que no sabías de la culinaria ancestral asiática.",
-            "CUIDADO: lo que el pediatra de tus hijos no quiere que sepas sobre la diabetes infantil, descúbrelo ahora mismo."
-        ]
-    },
-    "emocional": {
-        "description": "No hay nada más poderoso que las emociones, hacen una conexión con tu audiencia, les habla al corazón.",
-        "examples": [
-            "7 formas de ganar dinero en menos de 24 horas... aun si tu negocio está por quebrar.",
-            "Confesiones de un ex-empleado de Starbucks que te darán las ideas para crear y tener tu propia cafetería.",
-            "Para todas las mamás primerizas que no logran cambiar el pañal de su bebé con éxito... aquí te decimos cómo hacerlo en 3 pasos.",
-            "¿La menstruación frustró tus vacaciones de verano? 10 tips supereficaces para disfrutar de tus vacaciones aun esos días.",
-            "Para Todas Los Emprendedores Que Están Frustrados Por Sus Ventas"
-        ]
-    },
-    "sensacionalista": {
-        "description": "La prensa amarilla los usa una y otra vez. Las revistas de chismes de la farándula también lo hacen.",
-        "examples": [
-            "ATERRADOR método que utiliza el ejército israelí que te hará perder 7 kilos en tan solo una semana.",
-            "¡Mandar nudes es lo de hoy! Peligroso pero efectivo método para ligar en las redes sociales.",
-            "¡Nunca antes visto! Asesino serial consigue abrir una canal en YouTube para dar clases de cocina.",
-            "La más reciente prueba de amor que está matando a los adolescentes."
-        ]
-    },
-    "contraste": {
-        "description": "Moldea la percepción que tenemos de las cosas. Crea un cortocircuito en la mente de las personas.",
-        "examples": [
-            "¿Quieres una piel joven y tersa? Descubre los secretos de una mujer de 60 años para tener la piel de bebé",
-            "Descubre cómo el mal aliento me ayudó a enamorar a la mujer de mis sueños",
-            "Esta técnica de lectura rápida me ayudó a pasar mis exámenes sin tener que pasar meses estudiando.",
-            "¿La menstruación frustró tus vacaciones de verano? 10 tips súper eficaces para disfrutar de tus vacaciones aún esos días."
-        ]
-    },
-    "hazte_una_pregunta": {
-        "description": "Preguntar sobre algo que le interesa, hace que se detengan y reflexionen.",
-        "examples": [
-            "¿Cuál De Estos Problemas De Salud Quieres Solucionar?",
-            "¿Te gustaría ser más persuasivo y convincente para vender más y mejor?",
-            "¿Quieres aprender un nuevo idioma en poco tiempo y de forma sencilla?",
-            "¿Cometes estos errores garrafales en tus textos digitales? Aquí te decimos 4 apps que te ayudarán a tener buena ortografía..."
+# Función para generar titulares
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula):
+    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
+        "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).'"
+    )
+
+    # 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}. "
+        f"Asegúrate de utilizar la fórmula de titular seleccionada: {selected_formula['description']}. "
+        f"Usa los ejemplos proporcionados para guiar tu creatividad."
+    )
+
+    chat_session = model.start_chat(
+        history=[
+            {
+                "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
+
+# Configurar la interfaz de usuario con Streamlit
+st.set_page_config(page_title="Enchanted Hooks", layout="wide")
+
+# Centrar el título y el subtítulo
+st.markdown("<h1 style='text-align: center;'>Enchanted Hooks</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center;'>Imagina poder conjurar títulos que no solo informan, sino que encantan. Esta app es tu varita mágica en el mundo del copywriting, transformando cada concepto en un titular cautivador que deja a todos deseando más.</h4>", unsafe_allow_html=True)
+
+# Añadir CSS personalizado para el botón
+st.markdown("""
+    <style>
+    div.stButton > button {
+        background-color: #FFCC00;
+        color: black;
+        width: 90%;
+        height: 60px;
+        font-weight: bold;
+        font-size: 22px;
+        text-transform: uppercase;
+        border: 1px solid #000000;
+        border-radius: 8px;
+        display: block;
+        margin: 0 auto;
     }
-}
+    div.stButton > button:hover {
+        background-color: #FFD700;
+        color: black;
+    }
+    </style>
+""", unsafe_allow_html=True)
 
-# Función para generar titulares
-def generate_headlines(number_of_headlines, angle, product_mention, product):
-    mention_instruction = get_mention_instruction(product_mention, product)
-    
-    # Instrucción con el ángulo seleccionado
-    prompt = f"""
-    Genera {number_of_headlines} titulares persuasivos usando el ángulo: {angle}. 
-    El ángulo se basa en los siguientes puntos:
-    {angles[angle]['description']}
-    
-    Ejemplos: {', '.join(angles[angle]['examples'])}
-    
-    {mention_instruction}
-    """
-    response = genai.generate_text(prompt=prompt, temperature=0.7, max_output_tokens=1000)
-    return response.result
-
-# Aplicación Streamlit para interfaz de usuario
-st.title("Generador de Titulares Persuasivos")
-st.write("Genera titulares persuasivos basados en las fórmulas de copywriting y ángulos emocionales.")
-
-# Parámetros de personalización
-st.header("Selecciona un ángulo")
-selected_angle = st.selectbox("Selecciona un ángulo:", list(angles.keys()))
-
-# Producto de ejemplo
-product = st.text_input("Introduce el producto:", "suplemento para perder peso")
-
-# Opción de mención
-product_mention = st.selectbox("Selecciona el tipo de mención:", ["Directa", "Indirecta", "Metafórica"])
-
-# Número de titulares a generar
-number_of_headlines = st.slider("Número de titulares:", 1, 10, 5)
-
-if st.button("Generar Titulares"):
-    headlines = generate_headlines(number_of_headlines, selected_angle, product_mention, product)
-    st.write("Titulares generados:")
-    st.write(headlines)
+# Crear columnas
+col1, col2 = st.columns([1, 2])  
+
+# Columnas de entrada
+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")
+    number_of_headlines = st.selectbox("Número de Titulares", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
+
+    # Crear un único acordeón para fórmula y 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_formula_key = st.selectbox(
+            "Selecciona una fórmula para tus titulares",
+            options=list(headline_formulas.keys())
+        )
+        selected_formula = headline_formulas[selected_formula_key]
+
+        # Mostrar ejemplos de la fórmula seleccionada
+        st.markdown(f"**Ejemplos de la fórmula seleccionada ({selected_formula_key}):**")
+        for example in selected_formula['examples']:
+            st.write(example)
+
+        if selected_formula_key == 'fórmula_3':
+            st.markdown(f"**Explicación de la fórmula AIDA:**")
+            st.markdown(selected_formula['explanation'])
+        elif selected_formula_key == 'fórmula_1':
+            st.markdown(f"**Preguntas clave para la fórmula 1:**")
+            for question in selected_formula['questions']:
+                st.write(question)
+
+    # Botón de enviar
+    submit = st.button("Generar Titulares")
+
+# Mostrar los titulares generados
+if submit:
+    if target_audience and product and selected_formula:
+        try:
+            # Obtener la respuesta del modelo
+            generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula)
+            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, el producto y selecciona una fórmula.")