app.py

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

load_dotenv()
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

story_formulas = {
    "PASA": {
        "description": """
        La fórmula P.A.S.A. es una de las estructuras más efectivas en copywriting, diseñada para crear una conexión emocional profunda que lleva a la acción.

        TEORÍA:
        Esta fórmula se basa en la psicología del dolor y el alivio, siguiendo el principio de que las personas están más motivadas a evitar el dolor que a buscar el placer.

        ESTRUCTURA:
        1. Problema:
           - Identifica un dolor específico y real
           - Debe ser inmediatamente reconocible por la audiencia
           - Se presenta desde la perspectiva del lector
           - Usa lenguaje empático y comprensivo

        2. Agitación:
           - Profundiza en las consecuencias emocionales
           - Explora ramificaciones futuras
           - Aumenta la tensión psicológica
           - Crea un punto de inflexión emocional

        3. Solución:
           - Presenta la respuesta de forma natural
           - Enfoca en beneficios transformacionales
           - Construye credibilidad gradualmente
           - Demuestra el cambio positivo

        4. Acción:
           - Propone el siguiente paso lógico
           - Minimiza la fricción y resistencia
           - Crea sentido de urgencia
           - Ofrece garantías implícitas
        """,
        "examples": [
            {
                "title": "App de Meditación",
                "target": "Profesionales estresados",
                "story": """
                El insomnio se había convertido en su compañero más fiel. Sara miraba el reloj: 3 AM, otra vez.

                Los días se volvían borrosos, su concentración desaparecía. Las reuniones importantes se convertían en una lucha por mantener los ojos abiertos, mientras su productividad se desplomaba.

                Todo cambió cuando descubrió MeditaApp. Sus ejercicios de respiración guiada y meditaciones nocturnas la ayudaron a encontrar ese interruptor mental que necesitaba para desconectar.

                Descarga MeditaApp hoy y recupera tus noches. Tus primeros 7 días son gratuitos, porque creemos que una noche de buen sueño no tiene precio.
                """
            },
            {
                "title": "Curso de Finanzas Personales",
                "target": "Jóvenes profesionales",
                "story": """
                Cada notificación del banco le provocaba un nudo en el estómago. Miguel evitaba revisar su cuenta, sabiendo que los números rojos lo esperaban.

                Las deudas crecían mientras sus sueños se alejaban. La idea de comprar una casa o empezar un negocio parecía cada vez más inalcanzable, y la ansiedad financiera lo consumía día y noche.

                Entonces encontró FinanzasMaster. El curso no solo le enseñó a manejar su dinero, sino a construir un futuro financiero sólido. En seis meses, eliminó sus deudas y comenzó a ver crecer sus ahorros.

                Únete a FinanzasMaster hoy. Transforma tu relación con el dinero con nuestro programa probado de 12 semanas. Matrícula con 50% de descuento esta semana.
                """
            }
        ]
    },
    "GHA": {
        "description": """
        La fórmula G.H.A. (Gancho-Historia-Acción) es una estructura narrativa poderosa que utiliza el storytelling para crear conexiones emocionales y motivar a la acción.

        TEORÍA:
        Esta fórmula aprovecha el poder de las historias para captar la atención, generar empatía y motivar el cambio, basándose en la psicología narrativa y el poder de la identificación personal.

        ESTRUCTURA:
        1. Gancho:
           - Abre con un elemento sorpresa o intriga
           - Genera curiosidad inmediata
           - Plantea una pregunta o situación provocativa
           - Conecta emocionalmente desde la primera línea

        2. Historia:
           - Desarrolla una narrativa personal o relatable
           - Incluye elementos de conflicto y superación
           - Mantiene la tensión narrativa
           - Incorpora detalles sensoriales y emocionales

        3. Acción:
           - Surge naturalmente de la resolución
           - Conecta la moraleja con la solución
           - Motiva desde la inspiración
           - Propone un camino claro hacia el cambio
        """,
        "examples": [
            {
                "title": "Programa de Fitness",
                "target": "Personas que quieren comenzar a ejercitarse",
                "story": """
                "Nunca seré una persona fitness", me repetía cada vez que veía esas fotos perfectas en Instagram. Mi realidad eran 3 minutos de ejercicio y 20 de excusas.

                Un día, mientras subía las escaleras a mi departamento, tuve que detenerme en el segundo piso, jadeando. Mi hijo de 5 años me miró preocupado y me preguntó: "¿Mami, estás enferma?". Ese momento cambió todo.

                FitLife no me prometió un cuerpo de revista, me ofreció algo mejor: la capacidad de jugar con mi hijo sin quedarme sin aliento. Empecé con 10 minutos diarios, luego 15, y ahora corremos juntos en el parque.

                Comienza tu historia de transformación hoy. Únete a FitLife y descubre cómo 15 minutos pueden cambiar tu vida. Primera semana gratis, porque tu bienestar no puede esperar.
                """
            }
        ]
    }
}

def generate_story(formula_type, target_audience, product, action, mood, length, story_topic=None):
    """Función unificada para generar historias"""
    model = genai.GenerativeModel("gemini-2.0-flash")

    if formula_type not in story_formulas:
        raise ValueError("Fórmula no válida")

    # System prompt base
    system_prompt = f"""You are a world-class copywriter, specialized in crafting persuasive stories that emotionally connect with readers and drive them to action.

FORMAT RULES:
- Story must be structured in clear paragraphs
- Each section must flow naturally into the next
- No explicit section labels
- Maintain consistent tone throughout
- Include smooth transitions between parts
- Story must feel complete and cohesive

IMPORTANT STRUCTURE INSTRUCTIONS:
- The selected formula MUST be applied to the entire story
- Review and follow the structure detailed in story_formulas[formula_type]["description"]
- Use the examples in story_formulas[formula_type]["examples"] as inspiration
- Formula provides the framework, tone provides the emotional layer
- Both must work together seamlessly

FORMAT EXAMPLE:
[First paragraph introducing the situation]

[Second paragraph developing tension/conflict]

[Third paragraph presenting transformation]

[Final paragraph with natural call to action]

IMPORTANT:
- Each story must be unique and memorable
- Avoid clichés and generic scenarios
- Maintain a credible and authentic voice
- Adapt language to target audience
- Focus on emotional transformation
- Follow formula structure while maintaining selected tone
- Never explicitly label sections
- Integrate product and call-to-action organically

STORY STRUCTURE GUIDELINES:
1. Study the formula description carefully
2. Review provided examples for style and flow
3. Maintain similar paragraph length and structure
4. Use consistent emotional intensity
5. Mirror the narrative progression
6. Adapt content for target audience while keeping formula essence

TONE AND STYLE:
- Selected tone must remain consistent
- Tone modifies HOW the story is told
- Formula provides structure, tone provides emotion
- Both elements must complement each other
- Maintain voice appropriate for target audience

KEY ELEMENTS TO INCLUDE:
- Emotional hooks that grab attention
- Relatable situations and characters
- Sensory details and vivid descriptions
- Natural problem-solution progression
- Organic product integration
- Compelling but subtle call-to-action"""

    # Iniciar el prompt con las instrucciones del sistema
    story_instruction = f"{system_prompt}\n\n"

    # Agregar la descripción de la fórmula seleccionada
    story_instruction += f"""
SELECTED FORMULA STRUCTURE:
{story_formulas[formula_type]['description']}

SUCCESSFUL EXAMPLES OF THIS FORMULA:
"""

    # Agregar ejemplos de la fórmula
    for example in story_formulas[formula_type]['examples']:
        story_instruction += f"""
TITLE: {example['title']}
TARGET: {example['target']}
STORY:
{example['story']}
"""

    # Agregar instrucciones específicas para la generación
    story_instruction += f"""
YOUR TASK:
Create a compelling {length}-word story in Spanish with a {mood} tone for {target_audience} about {product} that leads to this action: {action}.

SPECIFIC REQUIREMENTS:
1. Follow the formula structure exactly as shown in the description
2. Maintain the {mood} tone throughout the story
3. Write for {target_audience} using appropriate language and references
4. Integrate {product} naturally into the narrative
5. Lead to the call-to-action: {action}
"""

    # Agregar instrucciones específicas para GHA si aplica
    if formula_type == "GHA" and story_topic:
        story_instruction += f"""
6. Incorporate this specific story topic: {story_topic}
"""

    story_instruction += """
FINAL REMINDERS:
1. Write in clear, engaging Spanish
2. Use proper paragraph structure
3. Maintain narrative flow
4. Create emotional connection
5. Make call-to-action feel natural

Generate the story now:
"""

    response = model.generate_content([story_instruction])
    if response and response.parts:
        return response.parts[0].text.strip()
    raise ValueError("No se pudo generar la historia")
    
# Inicializar la aplicación Streamlit
st.set_page_config(page_title="Generador de Historias", page_icon=":pencil:", layout="wide")

# Título y subtítulo
st.markdown("<h1 style='text-align: center;'>Story Genius Maker</h1>", unsafe_allow_html=True)
st.markdown("<h3 style='text-align: center;'>Teje historias inolvidables en segundos, guiado por la magia de la inteligencia artificial que da vida a tus ideas en relatos cautivadores.</h3>", unsafe_allow_html=True)

# Acordeón para elegir la fórmula
with st.expander("Selecciona la fórmula para generar tu historia"):
    # Obtener las fórmulas disponibles del diccionario story_formulas y crear nombres de visualización
    available_formulas = list(story_formulas.keys())
    formula_display_names = []
    formula_mapping = {}

    # Crear mapeo entre nombres de visualización y claves internas
    for formula in available_formulas:
        display_name = '.'.join(list(formula)) if formula.isupper() else formula
        formula_display_names.append(display_name)
        formula_mapping[display_name] = formula

    # Radio button con nombres de visualización
    formula = st.radio(
        "Selecciona la fórmula para generar tu historia:",
        formula_display_names
    )

    # Obtener la clave interna correspondiente al nombre seleccionado
    selected_formula = formula_mapping[formula]

# Crear las columnas para el diseño (40% para la izquierda, 60% para la derecha)
col1, col2 = st.columns([2, 3])

# Cambiar el comportamiento del slider cuando se seleccione "GHA"
with col1:
    target_audience = st.text_input("Público objetivo",
                                   placeholder="¿A quién está dirigido tu mensaje?")
    product = st.text_input("Producto/Servicio",
                           placeholder="¿Qué estás ofreciendo?")
    action = st.text_area("Llamado a la acción",
                         placeholder="¿Qué acción específica debe tomar tu audiencia?")

    # Personalización adicional
    with st.expander("Personaliza tu historia"):
        mood = st.selectbox("Tono de la historia:",
                           ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])

        # Ajustar el slider según la fórmula seleccionada
        if selected_formula == "GHA":
            length = st.slider("Longitud de la historia (palabras):",
                             min_value=50, max_value=200, value=150, step=10)
        else:
            length = st.slider("Longitud de la historia (palabras):",
                             min_value=50, max_value=150, value=100, step=10)

    # Mostrar campo adicional solo para GHA
    story_topic = None
    if selected_formula == "GHA":
        story_topic = st.text_area(
            "De qué quieres que trate la historia",
            placeholder="Explica si hay algo específico sobre lo que te gustaría contar (puede ser una vivencia personal, película, cuento, personaje ficticio, etc.)."
        )

    # Botón para generar contenido
    submit = st.button("Generar mi historia")

# Mostrar el contenido generado en la columna derecha (col2)
with col2:
    if submit:
        if target_audience and product and action:
            try:
                # Verificar si es GHA y necesita story_topic
                if selected_formula == "GHA":
                    if story_topic:
                        response = generate_story(selected_formula, target_audience, product,
                                               action, mood, length, story_topic)
                    else:
                        st.error("Por favor, completa todos los campos requeridos para la fórmula G.H.A.")
                        response = ""
                else:
                    response = generate_story(selected_formula, target_audience, product,
                                           action, mood, length)

                if response:
                    st.subheader("Historia generada:")
                    st.write(response)
            except ValueError as e:
                st.error(f"Error: {str(e)}")
        else:
            st.error("Por favor, completa todos los campos requeridos (Público objetivo, Producto y Acción).")