src/services/story_generator.py

from dotenv import load_dotenv
import os
import google.generativeai as genai
import random
from src.data.story_formulas import story_formulas

class StoryGenerator:
    def __init__(self):
        try:
            load_dotenv()
            genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
            
            self.generation_config = {
                "temperature": 1.0,
                "top_p": 0.65,
                "top_k": 360,
                "max_output_tokens": 8196,
            }
            
            self.model = genai.GenerativeModel(
                model_name="gemini-2.0-flash",
                generation_config=self.generation_config
            )
        except Exception as e:
            raise ValueError(f"Error initializing StoryGenerator: {str(e)}")

    #Función para generar un titular basado en la historia    
    def generate_headline(self, story, formula_type, product, target_audience, mood, temperature):
        """Función para generar un titular basado en la historia"""
        system_prompt = """Eres un experto copywriter especializado en crear titulares persuasivos.

    IMPORTANTE:

    - Genera SOLO UN TITULAR

    - No incluyas explicaciones ni etiquetas

    - El titular debe estar entre comillas "titular"

    - No uses números ni viñetas

    - El titular debe fluir naturalmente con la historia

    - El titular DEBE estar directamente relacionado con el contenido de la historia

    - Evita titulares genéricos que podrían aplicar a cualquier historia

    - Extrae elementos específicos de la historia para crear un titular único

    """
    
        formula_prompts = {
        "GPS": f"""

        {system_prompt}



        INSTRUCCIONES PARA TITULAR GPS:

        Crea un titular que combine:

        1. Meta: Un resultado deseable para {target_audience}

        2. Periodo: Una situación cotidiana o momento del día

        3. Superación: Un conector con toque de humor (sin, incluso si, aunque, etc.)



        ESTRUCTURA REQUERIDA:

        [Meta deseable] + [Momento cotidiano] + [Conector humorístico]



        EJEMPLOS EFECTIVOS:

        - "Domina el arte de la inversión mientras te cepillas los dientes incluso si confundes Excel con PowerPoint"

        - "Aprende un nuevo idioma durante tus visitas al baño aunque solo sepas decir gracias y por favor"

        - "Conquista el miedo a hablar en público durante el desayuno a pesar de que te tiemblen hasta las pestañas"

        - "Desarrolla músculos preparando café cuando levantar la taza te parece ejercicio extremo"

        - "Domina la fotografía profesional en el supermercado aun con ese celular que sobrevivió tres caídas"



        VALIDACIONES:

        - ¿Incluye una meta clara y deseable?

        - ¿Menciona un momento cotidiano específico?

        - ¿Usa un conector humorístico natural?

        - ¿Mantiene el humor sin ser ridículo?

        - ¿Es relevante para {target_audience}?



        Historia a titular:

        {story}

        """,

        "AIDA": f"""

        {system_prompt}



        INSTRUCCIONES PARA TITULAR AIDA:

        Crea un titular que combine:

        1. Atención: Gancho sorprendente o contraintuitivo

        2. Interés: Desarrollo que intriga

        3. Deseo: Promesa transformadora

        4. Acción: Siguiente paso natural



        ESTRUCTURA REQUERIDA:

        [Gancho sorprendente] + [Desarrollo intrigante] + [Promesa]



        EJEMPLOS EFECTIVOS:

        - "¿Sabías que el 83% de los emprendedores exitosos dedican menos de 2 horas al día a reuniones?"

        - "La mayoría no sabe que existe una técnica de ventas basada en videojuegos"

        - "Contrario a lo que piensas, el momento más productivo del día no es por la mañana"

        - "Como el bambú japonés, este método crece invisible hasta explotar en resultados"

        - "Los atletas olímpicos entrenan menos horas pero logran más gracias a esta técnica"



        VALIDACIONES:

        - ¿El gancho genera curiosidad inmediata?

        - ¿El desarrollo mantiene el interés?

        - ¿La promesa es creíble y atractiva?

        - ¿Mantiene coherencia con la historia?

        - ¿Es relevante para {target_audience}?



        Historia a titular:

        {story}

        """,

        "4U": f"""

        {system_prompt}



        INSTRUCCIONES PARA TITULAR 4U:

        Combina estratégicamente:

        1. Útil: Beneficio práctico y tangible

        2. Urgente: Motivador de acción inmediata

        3. Único: Diferenciador memorable

        4. Ultra-específico: Detalles precisos y medibles



        ESTRUCTURA REQUERIDA:

        [Beneficio útil] + [Elemento único] + [Detalle ultra-específico] + [Urgencia]



        EJEMPLOS EFECTIVOS:

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

        - "Automatiza 89% de tus tareas administrativas dedicando 31 minutos cada lunes"

        - "La técnica de un cartero rural ayuda a 1893 personas a organizar su tiempo"

        - "Un bibliotecario retirado genera 12437 euros vendiendo libros antiguos"



        VALIDACIONES:

        - ¿Incluye un beneficio claro y medible?

        - ¿Tiene un elemento diferenciador único?

        - ¿Usa datos específicos y creíbles?

        - ¿Genera sensación de urgencia natural?

        - ¿Es relevante para {target_audience}?



        Historia a titular:

        {story}

        """
    }
        headline_prompt = formula_prompts.get(formula_type, f"""

    {system_prompt}



    INSTRUCCIONES ESPECÍFICAS:

    Genera un titular persuasivo y memorable que:

    - Conecte directamente con la historia narrada

    - Refleje la transformación o experiencia descrita

    - Use elementos específicos mencionados en el contenido

    - Mantenga un tono {mood} coherente con la narrativa

    - Sea conciso pero impactante

    - No mencione directamente "{product}"

    - Genere curiosidad mientras mantiene la relevancia



    IMPORTANTE:

    - El titular debe surgir naturalmente de la historia

    - Usa palabras clave y conceptos presentes en la narrativa

    - Evita titulares genéricos que podrían aplicar a cualquier historia



    Historia a titular:

    {story}

    """)
        response = model.generate_content([headline_prompt])
        if response and response.parts:
            headline_text = response.parts[0].text.strip()
            import re
            match = re.search(r'"([^"]*)"', headline_text)
            if match:
                return match.group(1)
            return headline_text
        return None
    def generate_story(self, formula_type, target_audience, product, action, mood, length, temperature, story_topic=None):
        try:
            self.generation_config["temperature"] = temperature
            
            if formula_type not in story_formulas:
                raise ValueError("Fórmula no válida")
            
            natural_instruction = f"""

            IMPORTANTE - NATURALIDAD EN LA HISTORIA:

            - NO menciones "{product}" de forma literal o forzada

            - Introduce el producto/servicio de manera sutil y orgánica en la narrativa

            - Evita que suene como una fórmula publicitaria

            - La solución debe surgir naturalmente de la historia

            - Usa variaciones y descripciones naturales en lugar del nombre exacto

            - La historia debe fluir como una conversación real, no como un anuncio

            """
        
            # Análisis del público objetivo
            audience_analysis = f"""

            ANÁLISIS DEL PÚBLICO OBJETIVO: {target_audience}

            

            INSTRUCCIONES DE ANÁLISIS:

            1. Dolores y Problemas:

            - Identifica los problemas cotidianos específicos de {target_audience}

            - Reconoce sus frustraciones y miedos más profundos

            - Analiza las consecuencias emocionales de estos problemas

            - Considera el impacto en su vida diaria

            

            2. Creencias y Valores:

            - Comprende sus creencias limitantes

            - Identifica sus aspiraciones y sueños

            - Reconoce sus valores fundamentales

            - Entiende sus motivaciones principales

            

            3. Situaciones de Vida:

            - Visualiza sus rutinas diarias

            - Identifica momentos de tensión o conflicto

            - Reconoce sus contextos sociales y profesionales

            - Comprende sus responsabilidades y presiones

            

            4. Puntos de Dolor Específicos:

            - Problemas prácticos: [analiza sus dificultades concretas]

            - Dolores emocionales: [identifica sus preocupaciones más profundas]

            - Miedos: [reconoce sus temores principales]

            - Frustraciones: [entiende sus obstáculos recurrentes]

            

            CONTEXTO NARRATIVO:

            - Si hay story_topic ("{story_topic if story_topic else 'No especificado'}"),

            úsalo como escenario o contexto para desarrollar la historia

            - El protagonista debe reflejar fielmente las características de {target_audience}

            - Los problemas y situaciones deben ser 100% reconocibles por la audiencia

            - La transformación debe abordar directamente sus dolores específicos

            """
        
            # Definir el system prompt dentro de la función
            system_prompt = """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 short paragraphs (2-4 lines maximum)

            - Number of paragraphs should adjust to requested length

            - Each section must flow naturally into the next

            - No explicit section labels

            - Include smooth transitions between parts

            - Story must feel complete and cohesive

            - Break text for better readability and impact

            

            FORMULA APPLICATION:

            - The selected formula MUST be applied to the entire story

            - Review and follow the structure from story_formulas[formula_type]["description"]

            - Use examples in story_formulas[formula_type]["examples"] as inspiration

            - Formula provides framework, tone provides emotional layer

            - Paragraphs can be more or fewer than formula steps

            - Structure should be present but not obvious

            

            CONTENT AND AUDIENCE:

            - Focus on relatable, everyday situations

            - Describe specific audience problems and obstacles

            - Use details that generate immediate identification

            - Story must faithfully reflect audience's real life

            - Problems and situations must be 100% recognizable

            - Base narrative on real audience pain points

            - Each story must be unique and memorable

            - Avoid clichés and generic scenarios

            

            TONE AND STYLE:

            - Mood only affects emotional tone, not structure

            - Emotions should arise from real situations

            - Keep focus on audience regardless of tone

            - Selected tone must remain consistent

            - Both formula and tone must complement each other

            - Maintain voice appropriate for target audience

            

            SPECIAL CONSIDERATIONS FOR GHA:

            - Story must develop around specified topic

            - Topic should be the main thread

            - Integrate product/service naturally within topic context

            - Keep topic as central narrative element

            - Importante:

            - 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

            - LANGUAGE INSTRUCTIONS:

            - Generate the story in Spanish

            - Use natural, fluent Spanish

            - Maintain cultural relevance for Spanish-speaking audiences

            - Ensure idioms and expressions are appropriate for Spanish

            - Keep all story content in Spanish, only system instructions in English

            - CALL TO ACTION GUIDELINES:

            In the end of each story create a natural and persuasive call to action that:

            

            Flows naturally from the story

            Focuses on transformation and benefits

            KEY ELEMENTS IN EVERY CTA:

            

            Transformational Benefit → What will change in their life

            Implicit Social Proof → Others have already achieved it

            Simple Next Step → A clear and achievable action

            Natural Urgency → Based on benefits, not scarcity

            - IMPORTANT:

            - The CTA must flow naturally from the story

            - Maintain the established emotional tone

            - Use language that inspires rather than pressures

            - Connect with the transformation described in the story

            """
            story_instruction = f"{system_prompt}\n\n{audience_analysis}\n\n"
        
            # Instrucciones específicas para GHA cuando hay story_topic
            if formula_type == "GHA" and story_topic:
                story_instruction += f"""

                INTEGRACIÓN DEL TEMA CENTRAL EN LA HISTORIA:

            

                TEMA PRINCIPAL: "{story_topic}"

            

                1. ESTRUCTURA NARRATIVA:

                - Primer párrafo: Introduce el tema "{story_topic}" de forma cautivadora

                - Desarrollo: Profundiza en la experiencia/situación relacionada con el tema

                - Clímax: Momento de transformación donde el producto se integra naturalmente

                - Cierre: Resolución que conecta el tema con la solución propuesta

            

                2. DESARROLLO DEL TEMA:

                - Usa el tema como hilo conductor de toda la historia

                - Describe situaciones específicas relacionadas con {story_topic}

                - Incluye detalles y experiencias únicas del tema

                - Mantén el foco en la perspectiva de {target_audience} sobre {story_topic}

            

                3. INTEGRACIÓN DEL PRODUCTO:

                - NO menciones {product} hasta haber desarrollado bien el tema

                - El producto debe surgir como solución natural a una situación del tema

                - Conecta los beneficios de {product} con aspectos específicos de {story_topic}

                - Mantén el protagonismo del tema por encima del producto

            

                4. ELEMENTOS OBLIGATORIOS:

                - Mínimo 3 referencias específicas a {story_topic}

                - Al menos 2 situaciones o momentos relacionados con el tema

                - Una conexión clara entre el tema y la necesidad del producto

                - Un cierre que refuerce la relación entre tema y solución

            

                5. VALIDACIONES:

                - ¿La historia gira realmente en torno a {story_topic}?

                - ¿Se mantiene el tema como elemento central?

                - ¿La integración del producto es natural dentro del contexto del tema?

                - ¿El llamado a la acción conecta con el tema desarrollado?

            

                RECORDATORIO IMPORTANTE:

                - La historia debe ser 100% sobre {story_topic}

                - El producto es secundario al tema

                - Cada párrafo debe contener elementos del tema

                - La transformación debe estar directamente relacionada con {story_topic}

                """
                
        # Añadir ejemplos específicos para el tipo de tema
        story_instruction += f"""

        EJEMPLOS DE INTEGRACIÓN DEL TEMA:

        

        Si el tema es una experiencia personal:

            INCORRECTO: "Un día cualquiera..."

            CORRECTO: "Aquella mañana en [situación específica del tema]..."

        

        Si el tema es una situación:

            INCORRECTO: "Las personas suelen..."

            CORRECTO: "En medio de [detalle específico del tema]..."

        

        Si el tema es un concepto:

            INCORRECTO: "Todos queremos mejorar..."

            CORRECTO: "Cuando te enfrentas a [aspecto específico del tema]..."

        

        ESTRUCTURA DEL PÁRRAFO INICIAL:

            1. Situación específica del {story_topic}

            2. Conexión emocional con {target_audience}

            3. Transición hacia el desarrollo

        """
        
        # Añadir instrucciones sobre el producto y beneficios
        story_instruction += f"""

        PRODUCT INTEGRATION GUIDELINES:

        - Evita menciones obvias de {product}

        - Enfócate en despertar interés genuino mostrando beneficios transformacionales

        - Integra el producto de manera natural en la narrativa

        - Destaca cómo el producto mejora la vida del protagonista

        - Mantén el tono {mood} al describir los beneficios

        - IMPORTANT REMINDER:

        - The entire story must be written in Spanish

        - Use natural expressions and cultural references appropriate for Spanish-speaking audiences

        - Ensure the story flows naturally in Spanish

        """
        
        # Si es GHA y hay story_topic, añadir instrucciones específicas
        if formula_type == "GHA" and story_topic:
            system_prompt += f"""

            INSTRUCCIONES ESPECÍFICAS PARA GHA:

            - El titular DEBE incorporar elementos del story_topic: "{story_topic}"

            - Mantén el enfoque principal en la transformación o experiencia relacionada con el tema

            - Usa palabras clave o conceptos específicos mencionados en la historia

            - Asegúrate de que el titular refleje la esencia del story_topic

            - La conexión entre el titular y el tema debe ser clara y natural

            CALL TO ACTION GUIDELINES:

            In the end of each story create a natural and persuasive call to action that:

        

            Flows naturally from the story

            Focuses on transformation and benefits

            KEY ELEMENTS IN EVERY CTA:

        

            Transformational Benefit → What will change in their life

            Implicit Social Proof → Others have already achieved it

            Simple Next Step → A clear and achievable action

            Natural Urgency → Based on benefits, not scarcity

        - IMPORTANT:

        - The CTA must flow naturally from the story

        - Maintain the established emotional tone

        - Use language that inspires rather than pressures

        - Connect with the transformation described in the story

        """
        
        # Instrucciones de estudio de ejemplos
        story_instruction += f"""

        IMPORTANTE: Estudia cuidadosamente estos ejemplos de la fórmula seleccionada.

        Cada ejemplo representa el estilo y estructura a seguir, adaptados al tono {mood}:

        """
        
        # Seleccionar ejemplos aleatorios
        random_examples = random.sample(story_formulas[formula_type]['examples'],
                                      min(3, len(story_formulas[formula_type]['examples'])))
        
        # Agregar ejemplos seleccionados
        story_instruction += "\nEJEMPLOS DE LA FÓRMULA A SEGUIR:\n"
        for i, example in enumerate(random_examples, 1):
            story_instruction += f"""

        Ejemplo {i}:

        Título: {example['title']}

        Audiencia: {example['target']}

        Acción: {example['action']}

        Historia:

        {example['story']}

        """
        
        # Agregar instrucciones específicas
        story_instruction += """

        INSTRUCCIONES ESPECÍFICAS:

        1. Mantén la misma estructura y longitud que los ejemplos anteriores

        2. Usa el mismo tono y estilo de escritura, adaptado al mood especificado

        3. Replica los patrones de construcción de frases y transiciones

        4. Conserva el nivel de especificidad y detalle en las descripciones

        5. Adapta el contenido para la audiencia manteniendo la esencia de los ejemplos

        6. Integra los beneficios del producto de manera sutil y convincente

        7. Asegura que la transformación del protagonista sea creíble y emotiva

        """
        
        # Agregar la descripción de la fórmula
        story_instruction += f"\nFÓRMULA A SEGUIR:\n{story_formulas[formula_type]['description']}\n\n"
        
        # Recordatorio final
        story_instruction += f"""

        RECORDATORIO FINAL:

        1. Sigue la estructura de la fórmula seleccionada

        2. Aplica el tono {mood} de manera consistente

        3. Mantén la coherencia narrativa

        4. Asegura que la historia refleje una transformación auténtica

        5. Integra {product} y sus beneficios de forma natural

        6. Culmina con un llamado a la acción convincente: {action}

        

        GENERA AHORA:

        Crea una historia persuasiva de {length} palabras en español que siga fielmente el estilo y estructura de los ejemplos mostrados,

        manteniendo un tono {mood} y enfocándote en la transformación que {product} puede traer a la vida de {target_audience}.

        

        IMPORTANT FINAL REMINDER:

        - Write the complete story in Spanish

        - Ensure natural flow and cultural relevance

        - Maintain authentic Spanish expressions and tone

        """
        
            # Asegúrate de que las instrucciones de naturalidad estén al inicio del prompt
            story_instruction = f"{natural_instruction}\n\n{story_instruction}"
            
            # Generar la historia
            response = self.model.generate_content([story_instruction])    
            # Generar la historia
            response = self.model.generate_content([story_instruction])
            if not response or not response.parts:
                raise ValueError("La API no generó una respuesta válida")
            
            story = response.parts[0].text.strip()
            if not story:
                raise ValueError("La historia generada está vacía")
            
            formula_types = ["GPS", "AIDA", "4U", "Númerica Suprema"]
            selected_formula = random.choice(formula_types)
            headline = self.generate_headline(story, selected_formula, product, target_audience, mood, temperature)
            
            if headline:
                return f"{headline}\n\n{story}"
            return story
                
        except Exception as e:
            raise ValueError(f"No se pudo generar la historia: {str(e)}")