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

# Configurar la página con el sidebar colapsado
st.set_page_config(
    page_title="Tu App",
    page_icon="🚀",
    layout="wide",
    initial_sidebar_state="collapsed"
)

st.markdown("""<style>...""", unsafe_allow_html=True)

# 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 titulares con 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 titulares 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 titulares 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 titular debe mantener un tono conversacional y evitar parecer una fórmula obvia.
    """,
    "examples": [
        # A (dato sorprendente) + I (conexión) + D (prueba social) + A (paso simple)
        "¿Sabías que el 83% de los emprendedores exitosos dedican menos de 2 horas al día a reuniones? Lo fascinante es que usan un método japonés de gestión que les permite triplicar su productividad, y puedes empezar hoy mismo.",

        # A (contraintuitivo) + I (beneficio) + D (resultado) + A (inicio)
        "La mayoría no sabe que existe una técnica de ventas basada en videojuegos, está revolucionando el mercado B2B, genera resultados inmediatos y puedes aprenderla en 20 minutos.",

        # A (disruptivo) + I (detalle) + D (resultado)
        "Contrario a lo que piensas, el momento más productivo del día no es por la mañana, sino durante una ventana de tiempo inesperada que duplicará tu capacidad de concentración.",

        # A (descubrimiento) + I (solución) + D (prueba)
        "Me sorprendió descubrir que los mejores vendedores nunca memorizan scripts, usan una técnica secreta de improvisación.",

        # I (método único) + D (transformación) + A (acción)
        "Esta técnica revolucionaria de aprendizaje está transformando cómo los profesionales dominan nuevas habilidades, pruébala hoy con solo 10 minutos.",

        # I (innovación) + D (beneficio) + A (inicio)
        "El método del empresario silencioso cambia las reglas del networking moderno, empieza con un simple ejercicio.",

        # A (pregunta gancho) + D (resultado) + A (implementación)
        "¿Sabías que existe un ritual zen que está transformando la productividad en Silicon Valley? Implementalo mañana mismo.",

        # A (analogía) + D (promesa) + A (descubrimiento)
        "Como el bambú japonés, este método crece silenciosamente hasta que explota en resultados, descubre cómo en 5 minutos.",

        # A (mayoría) + I (validación) + A (acción)
        "La mayoría desconoce el mejor momento para tomar decisiones importantes, la ciencia lo confirma, implementa este descubrimiento hoy.",

        # A (mini-cambio) + I (resultado) + A (inicio)
        "Un pequeño cambio en tu rutina de email desencadenará mejoras exponenciales, comienza ahora."
    ]
},
    "4U": {
    "description": """
        La fórmula 4U se aplica de manera flexible y estratégica, combinando 1-4 elementos para crear titulares impactantes y naturales:

        1. **Útil** (Useful):
           Beneficios prácticos y tangibles:
           - "Cómo conseguir..." + resultado específico
           - "La guía paso a paso para..."
           - "El método probado que..."
           - "Descubre la forma de..."
           - "Aprende a..." + beneficio concreto
           - "La solución definitiva para..."
           - "El sistema que te permite..."
           - "La estrategia que garantiza..."

        2. **Urgente** (Urgent):
           Motivadores de acción inmediata:
           - "Última oportunidad para..."
           - "Solo disponible hasta..."
           - "Antes de que sea tarde..."
           - "Mientras aún hay tiempo..."
           - "No esperes a que..."
           - "Aprovecha ahora..."
           - "La oferta expira en..."
           - "Date prisa antes de que..."

        3. **Único** (Unique):
           Diferenciadores memorables:
           - "El método poco conocido..."
           - "La técnica contraintuitiva..."
           - "El descubrimiento sorprendente..."
           - "La estrategia secreta..."
           - "El sistema revolucionario..."
           - "El enfoque innovador..."
           - "La solución inesperada..."
           - "El método exclusivo..."

        4. **Ultra-específico** (Ultra-specific):
           Detalles precisos y medibles:
           - Números exactos: "27 técnicas..."
           - Tiempos concretos: "13 minutos..."
           - Resultados medibles: "63% más..."
           - Pasos definidos: "3 pasos..."
           - Datos precisos: "8.3 veces más..."
           - Métricas claras: "2.5x más rápido..."
           - Cantidades específicas: "97 personas..."
           - Plazos definidos: "21 días..."

        Combinaciones estratégicas:
        - Útil + Ultra-específico: Para audiencias prácticas y orientadas a resultados
        - Urgente + Único: Para ofertas especiales y lanzamientos
        - Único + Ultra-específico: Para diferenciación basada en datos
        - Útil + Urgente: Para motivar acción inmediata
        - Útil + Único + Ultra-específico: Para establecer autoridad y credibilidad
        - Útil + Urgente + Ultra-específico: Para ofertas con beneficios medibles
        - Único + Urgente + Ultra-específico: Para lanzamientos exclusivos
        - Útil + Único + Urgente: Para ofertas transformadoras
        - Las 4U juntas: Para máximo impacto en ofertas premium

        Cada titular debe mantener un tono conversacional y evitar parecer una fórmula obvia.
    """,
        "examples": [
    # Útil + Ultra-específico (Para audiencias prácticas y orientadas a resultados)
    "Aprende 347 palabras en alemán memorizando solo 12 minutos al día mientras cocinas",
    "Cultiva 27 tipos de hierbas aromáticas en 1.5 metros cuadrados de balcón generando 180 euros mensuales",
    "Automatiza 89% de tus tareas administrativas dedicando 31 minutos cada lunes",

    # Urgente + Único (Para ofertas especiales y lanzamientos)
    "Un cartero rural comparte su sistema de gestión del tiempo antes de su retiro definitivo",
    "La última clase del profesor más longevo de Oxford revela su método de memorización",
    "El manuscrito perdido de un monje escriba sale a la luz tras 80 años en el Vaticano",

    # Único + Ultra-específico (Para diferenciación basada en datos)
    "Un grupo de 1457 abuelas italianas entrena una IA para reconocer pasta al dente con 99.7% de precisión",
    "La técnica de un bibliotecario jubilado cataloga 47893 libros en 73 días usando fichas de colores",
    "Un taxista noruego fotografía 12437 auroras boreales usando un iPhone 6 y tres filtros caseros",

    # Útil + Urgente (Para motivar acción inmediata)
    "Aprende a crear tu huerto urbano antes de que suban los precios de las verduras",
    "Asegura tu plaza en el programa de ahorro energético antes del aumento de tarifas",
    "Registra tu marca personal mientras el dominio premium sigue disponible",

    # Útil + Único + Ultra-específico (Para establecer autoridad y credibilidad)
    "La técnica de un cartero rural ayuda a 1893 personas a organizar su tiempo usando solo 17 minutos cada mañana",
    "Un bibliotecario retirado genera 12437 euros vendiendo libros antiguos por menos de 3 euros cada uno",
    "Una abuela italiana entrena algoritmos que reducen el tiempo de cocción un 82% manteniendo el sabor tradicional",

    # Útil + Urgente + Ultra-específico (Para ofertas con beneficios medibles)
    "Aprende 478 palabras en japonés en 21 días antes del aumento de precio del programa",
    "Automatiza 95% de tu contabilidad en 7 días previo a la nueva normativa fiscal",
    "Genera 7 fuentes de ingresos pasivos en 30 días antes del cambio en las regulaciones",

    # Único + Urgente + Ultra-específico (Para lanzamientos exclusivos)
    "El sistema de un monje copista digitaliza 7345 manuscritos en 31 días últimas plazas disponibles",
    "La fórmula de un taxista fotografía 893 auroras boreales perfectas registro exclusivo esta semana",
    "El método de un bibliotecario jubilado organiza 47893 archivos en 73 días acceso limitado",

    # Útil + Único + Urgente (Para ofertas transformadoras)
    "La técnica de un cartero rural transforma tu productividad antes del fin de semana",
    "El sistema de una abuela italiana revoluciona tu forma de cocinar registro beta por 24 horas",
    "El método de un bibliotecario maximiza tu capacidad de aprendizaje plazas limitadas"
],
    "consejos_uso": [
        "Asegura que cada elemento U refuerce a los demás",
        "Mantén un equilibrio entre urgencia y credibilidad",
        "Usa datos específicos para aumentar la confianza",
        "Adapta el orden de los elementos según tu audiencia",
        "Evita exageraciones que puedan generar desconfianza",
        "Combina los elementos de forma natural y fluida",
        "Asegúrate de que cada elemento añada valor real",
        "Prueba diferentes combinaciones para encontrar las más efectivas"
        ]
    }
}

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,
        "top_k": 360,
        "max_output_tokens": 8196,
    }

    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.

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

FORMAT EXAMPLE:
1. Titular 1.

2. Titular 2.

3. Titular 3.

4. Titular 4.

5. Titular 5.

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

    # Modificar el headlines_instruction para enfatizar los ejemplos
    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 de {product} y enfócate en despertar interés genuino.\n\n"
        f"IMPORTANTE: Estudia cuidadosamente estos ejemplos modelo de la fórmula seleccionada. "
        f"Cada ejemplo representa el estilo, estructura y tono exacto que debes seguir:\n\n"
    )

    # Agregar 5 ejemplos aleatorios con análisis
    import random
    random_examples = random.sample(selected_formula['examples'], min(5, len(selected_formula['examples'])))

    headlines_instruction += "EJEMPLOS MODELO A SEGUIR:\n"
    for i, example in enumerate(random_examples, 1):
        headlines_instruction += f"{i}. {example}\n"

    headlines_instruction += "\nINSTRUCCIONES ESPECÍFICAS:\n"
    headlines_instruction += "1. Mantén la misma estructura y longitud que los ejemplos anteriores\n"
    headlines_instruction += "2. Usa el mismo tono y estilo de escritura\n"
    headlines_instruction += "3. Replica los patrones de construcción de frases\n"
    headlines_instruction += "4. Conserva el nivel de especificidad y detalle\n"
    headlines_instruction += f"5. Adapta el contenido para {target_audience} manteniendo la esencia de los ejemplos\n\n"

    headlines_instruction += f"FÓRMULA A SEGUIR:\n{selected_formula['description']}\n\n"

    headlines_instruction += "GENERA AHORA:\n"
    headlines_instruction += f"Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura de los ejemplos mostrados."

    chat_session = model.start_chat(
        history=[
            {
                "role": "user",
                "parts": [headlines_instruction],
            },
        ]
    )

    response = chat_session.send_message("Genera los titulares siguiendo exactamente el estilo de los ejemplos mostrados.")
    return response.text

# Configurar la interfaz de usuario con Streamlit
st.set_page_config(page_title="Enchanted Hooks", layout="wide")

# Configurar el estado inicial del sidebar si no existe
if 'sidebar_state' not in state:
    state.sidebar_state = 'collapsed'

# Leer el contenido del archivo manual.md
with open("manual.md", "r", encoding="utf-8") as file:
    manual_content = file.read()

# Mostrar el contenido del manual en el sidebar
st.sidebar.markdown(manual_content)

# Ocultar elementos de la interfaz
st.markdown("""
    <style>
        
        /* Reducir espacio superior */
        .block-container {
            padding-top: 1rem;
            padding-bottom: 5rem;
        }

        /* Ajustar espaciado del título */
        h1 {
            margin-top: -2rem;
            padding-top: 1rem;
        }

        /* Ajustar espaciado del subtítulo */
        h4 {
            margin-top: 0.5rem;
            padding-top: 0rem;
        }
    </style>
""", unsafe_allow_html=True)

# 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=2.0, value=1.0, 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: 15px; 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.")