app.py

from dotenv import load_dotenv
import streamlit as st
import os
import google.generativeai as genai

# Cargar las variables de entorno
load_dotenv()

# Configurar la API de Google
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

# Fórmulas con ejemplos y explicaciones
headline_formulas = {
    "GPS": {
        "description": """
            Crea bullets de beneficios que respondan estas tres preguntas clave:
            1. ¿Qué es lo que el lector quiere conseguir?
            2. ¿En qué periodo de tiempo quiere conseguirlo?
            3. ¿Cuál es la objeción principal del lector que le impide lograrlo?
        """,
        "examples": [
            "Obtén tus primeros 100 clientes en 30 días, sin gastar un solo centavo en publicidad.",
            "Pierde esos 5 kilos en solo 10 minutos al día, sin dejar de disfrutar tus comidas favoritas.",
            "Ahorra para salir de viaje en tres meses, sin sacrificar esas noches de cine."
        ]
    },
    "Númerica Suprema": {
    "description": """
        La Fórmula Suprema de Istvanova combina 5 elementos clave más artículos plurales para crear bullets persuasivos:

        1. **Artículos Plurales** (Art):
           - Los (para masculino plural)
           - Las (para femenino plural)
           - Dan naturalidad y autoridad al texto
           - Ejemplos: "Los 7 métodos...", "Las 3 técnicas..."

        2. **Números** (N):
           - Específicos y creíbles (3, 5, 7, 10...)
           - Crean estructura y expectativas claras
           - Se combinan con artículos: "Los 5...", "Las 3..."

        3. **Adjetivo** (A):
           - Emocionales y descriptivos
           - Conectan con deseos/miedos
           - Ejemplos: poderosos, simples, efectivos, revolucionarios

        4. **Palabra Clave** (P):
           - Término central del beneficio en plural
           - Fácil de entender y recordar
           - Ejemplos: métodos, estrategias, técnicas, secretos

        5. **Razón** (R):
           - Justifica el beneficio
           - Añade credibilidad
           - Conecta con la motivación del lector

        6. **Promesa** (P):
           - Resultado específico y medible
           - Timeframe realista
           - Beneficio final atractivo

        Formatos:
        - Corto: Art plural + N + A + P + P
        - Medio: Art plural + N + A + P + R + P
        - Largo: Art plural + N + A + P + R detallada + P específica
    """,
    "examples": [
        "Los 3 rituales probados para dormir mejor.",
        "Las 5 rutinas efectivas para fortalecer tu core.",
        "Los 7 hábitos esenciales para aumentar productividad.",
        "Las 3 técnicas comprobadas para dormir mejor basadas en neurociencia.",
        "Los 5 movimientos efectivos para fortalecer tu core sin equipamiento.",
        "Las 7 estrategias esenciales para aumentar productividad sin estrés.",
        "Los 3 métodos científicos para dormir mejor basados en los últimos descubrimientos de la neurociencia del sueño que transformarán tus noches.",
        "Las 5 secuencias efectivas para fortalecer tu core descubiertas por fisioterapeutas olímpicos que puedes hacer en casa.",
        "Los 7 sistemas revolucionarios para aumentar productividad desarrollados por CEOs que duplicarán tus resultados."
    ],
    "variaciones_estructura": {
        "básica": "Art plural + N + A + P + P",
        "intermedia": "Art plural + N + A + P + R + P",
        "avanzada": "Art plural + N + A + P + R detallada + P específica"
    },
    "uso_articulos_plurales": {
        "masculino_plural": {
            "artículo": "los",
            "ejemplos_palabras": "métodos, sistemas, pasos, secretos, trucos, hábitos"
        },
        "femenino_plural": {
            "artículo": "las",
            "ejemplos_palabras": "técnicas, estrategias, rutinas, tácticas, claves"
        }
    },
    "consejos_uso": [
        "Usa siempre la forma plural para mayor impacto",
        "Alterna entre 'los' y 'las' según la palabra clave",
        "Mantén coherencia en el género a lo largo del bullet",
        "Combina artículos con números de forma natural",
        "Asegura que la palabra clave esté en plural"
    ]
},
    "AIDA": {
    "description": """
        La fórmula AIDA se aplica de manera flexible y estratégica, combinando 1-4 elementos para crear bullets impactantes y naturales:

        1. **Atención** (A):
           Ganchos de apertura poderosos:
           - "¿Sabías que...?" + dato sorprendente
           - Mini-historia disruptiva
           - Idea contraintuitiva
           - Descubrimiento inesperado
           - Analogía poderosa
           - "La mayoría no sabe que..."
           - "Contrario a lo que piensas..."
           - "Me sorprendió descubrir que..."

        2. **Interés** (I):
           Desarrollo del gancho inicial:
           - Detalles específicos y relevantes
           - Conexión problema-solución inesperada
           - Beneficios únicos y memorables
           - Puente situación actual-resultado
           - "La razón es simple..."
           - "Lo fascinante es que..."
           - "Y lo mejor de todo..."
           - "Lo que hace la diferencia es..."

        3. **Deseo** (D):
           Amplificación emocional:
           - Imagen vivida del resultado
           - Experiencia personalizada
           - Prueba social natural
           - Toque de exclusividad
           - Conexión emocional profunda
           - "Imagina poder..."
           - "Piensa cómo sería..."
           - "Esto significa que podrás..."

        4. **Acción** (A):
           Cierre natural:
           - Siguiente paso simple
           - Baja fricción para comenzar
           - Gratificación inmediata
           - Primer paso sencillo
           - Seguridad fluida
           - "Pruébalo hoy mismo..."
           - "Comienza con un simple..."
           - "Solo necesitas..."

        Combinaciones estratégicas:
        - A + I: Para despertar curiosidad y explicar el valor
        - A + D: Para conectar problema con deseo
        - I + D: Para construir deseo desde la lógica
        - I + D + A: Para construir convicción y motivar
        - A + I + D: Para educar, intrigar y crear anhelo

        Cada bullet debe mantener un tono conversacional y evitar parecer una fórmula obvia.
    """,
    "examples": [
        "• Un estudio con deportistas de élite reveló algo sorprendente los atletas que menos entrenaban tenían mejores resultados. La clave está en un ritual de recuperación de 8 minutos que ahora puedes usar en casa.", # A (dato sorprendente) + I (conexión problema-solución)

        "• Mi cliente más exitoso solía dormir solo 4 horas por noche. Ahora duerme 8 y factura el doble, todo gracias a una rutina matutina que cambió su forma de trabajar.", # A (mini-historia) + I (beneficio único) + D (prueba social)

        "• Como el bambú chino, que parece no crecer durante años y de repente se dispara hacia arriba, tu negocio está a punto de experimentar ese momento de explosión.", # A (analogía poderosa) + D (imagen vivida)

        "• La técnica que transformó mi productividad apareció en el lugar más inesperado una clase de cocina italiana. Descubre cómo este método tan simple puede revolucionar tu forma de trabajar.", # A (descubrimiento inesperado) + I (beneficio único)

        "• Los expertos en productividad estaban equivocados. Las mañanas no son el momento más importante del día. Descubre cuándo ocurre realmente la magia y aprovecha ese momento desde hoy.", # A (contraintuitivo) + I (detalle específico) + A (siguiente paso)

        "• Imagina despertar cada mañana con tu bandeja de entrada vacía y tus tareas importantes ya organizadas. Este pequeño truco de 5 minutos lo hace posible.", # D (imagen vivida) + I (beneficio) + A (simplicidad)
    ]
},
    "3 en 1": {
        "description": """
            La fórmula 3 en 1 para empezar a conectar con las personas combina:
            1. **Característica**: Describe las propiedades tangibles del producto o servicio. Por ejemplo, "Este software tiene una función de automatización única".
            2. **Para qué**: El beneficio inmediato que resuelve un problema o cumple una necesidad. Por ejemplo, "Para que puedas enfocarte en tareas más importantes sin distracciones".
            3. **Con lo que**: El impacto emocional o práctico a largo plazo que obtiene el cliente. Por ejemplo, "Con lo que podrás disfrutar de un día más productivo y menos estresante".
            Crea bullets points que integren estos elementos para mostrar beneficios de forma clara, emocional y conectada al producto.
        """,
        "examples": [
            "Una batería de larga duración para que puedas trabajar todo el día con lo que mantienes tu productividad sin interrupciones.",
            "Un sistema de seguridad avanzado para que duermas tranquilo con lo que proteges lo que más valoras.",
            "Un diseño intuitivo para que aprendas rápido con lo que ahorras tiempo y esfuerzos innecesarios."
        ]
    }
}

# Función para generar titulares
def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula):

    # Crear la configuración del modelo
    generation_config = {
        "temperature": temperature,  
        "top_p": 0.65,        # Considerar un poco menos de palabras probables
        "top_k": 360,        # Aumentar las palabras candidatas para más variedad
        "max_output_tokens": 8196,  # Mantenerlo igual, pero puedes aumentar si deseas más detalle
    }

    model = genai.GenerativeModel(
        model_name="gemini-2.0-flash",
        generation_config=generation_config,
)
# Incluir las instrucciones del sistema en el prompt principal
system_prompt = """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.

OBJECTIVE:
- Generate persuasive and specific headlines in Spanish
- Spark immediate curiosity
- Address specific desires and pain points
- Maintain natural and magnetic language
- Motivate continued reading

FORMAT RULES:
- Each headline must start with number and period
- One headline per line
- No explanations or categories
- Add a line break between each headline
- Avoid unnecessary : symbols
- Each headline must be a complete and intriguing sentence

HEADLINE STRUCTURE:
- Must be relevant to target audience
- Must promise a specific result
- Must include an element of curiosity
- Must eliminate a main objection
- Must motivate continued reading

FORMAT EXAMPLE:
1. The uncomfortable truth about business success that no one dares to tell you (and how to use it to your advantage)

2. Why 89% of entrepreneurs fail in their first year (and how you can be in the 11% that succeeds)

3. The millionaires' secret method to double income working half the time

IMPORTANT:
- Each headline must be unique and memorable
- Avoid clichés and generalities
- Maintain an intriguing but credible tone
- Adapt language to the audience
- Focus on transformative benefits"""

# Crear un mensaje para el modelo
headlines_instruction = (
    f"{system_prompt}\n\n"
    f"Tu tarea es crear {number_of_headlines} titulares irresistibles diseñados para {target_audience}. "
    f"El objetivo es capturar la atención instantáneamente y generar curiosidad sobre {product}. "
    f"Evita menciones obvias del producto y enfócate en despertar interés genuino. "
    f"Usa la fórmula seleccionada como guía:\n\n{selected_formula['description']}\n\n"
    f"Inspírate en estos ejemplos:\n"
    f"1. {selected_formula['examples'][0]}\n"
    f"2. {selected_formula['examples'][1]}\n"
    f"3. {selected_formula['examples'][2]}\n\n"
    f"Tu objetivo es crear titulares tan intrigantes que sea imposible no seguir leyendo."
)

# Crear la sesión de chat
chat_session = model.start_chat(
    history=[
        {
            "role": "user",
            "parts": [headlines_instruction],
        },
    ]
)

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")

# 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)

# 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]

    # 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.")