Update app.py

app.py

@@ -1,184 +1,250 @@
-from dotenv import load_dotenv
-import streamlit as st
-import os
-import google.generativeai as genai
-from cta_formulas import cta_formulas
-from styles import apply_styles
-from tone_formulas import tone_settings
-
-# Cargar variables de entorno
-load_dotenv()
-
-# Configurar API de Google Gemini
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-
-def get_gemini_response(product_service, target_audience, desired_action, formula_type, tone_type, temperature, postdata_theme=None, cta_count=5):
-    if not product_service or not target_audience or not desired_action:
-        return "Por favor, completa todos los campos requeridos."
-    
-    formula = cta_formulas[formula_type]
-    tone = tone_settings[tone_type]
-    
-    # Lista de fórmulas que ya incluyen P.D. en su estructura
-    formulas_with_pd = ["El Último Aviso (No tan Último)", "Cierra con Corazón", "Gancho Cachondo"]
-    
-    # Preparar información de postdata
-    postdata_instruction = ""
-    if postdata_theme:
-        # Solo aplicar la instrucción de postdata si la fórmula tiene PD en su estructura
-        if formula_type in formulas_with_pd:
-            postdata_instruction = f"""
-            POSTDATA THEME:
-            Use the following theme for the P.S. section: {postdata_theme}
-            
-            IMPORTANT: For formulas with P.S.2 sections, use a DIFFERENT theme or approach than the one used in P.S.1.
-            DO NOT repeat the same urgency factor, benefit, or discount information in both postdatas.
-            
-            Examples:
-            - If P.S.1 mentions "only 2 days left", P.S.2 should NOT mention time limits again
-            - If P.S.1 talks about "limited spots", P.S.2 should focus on a different benefit or feature
-            
-            Make sure each postdata adds unique value and persuasion elements.
-            """
-    
-    model = genai.GenerativeModel('gemini-2.0-flash')
-    full_prompt = f"""
-    You are an expert copywriter specialized in creating persuasive Calls to Action (CTAs).
-    Analyze (internally, don't include in output) the following information:
-
-    BUSINESS INFORMATION:
-    Product/Service: {product_service}
-    Target Audience: {target_audience}
-    Desired Action: {desired_action}
-    CTA Type: {formula_type}
-    Tone Style: {tone['style']}
-    Keywords to consider: {', '.join(tone['keywords'])}
-    {formula["description"]}
-    {postdata_instruction}
-
-    First, analyze (but don't show) these points:
-    1. TARGET AUDIENCE ANALYSIS:
-       - What motivates them to take action?
-       - What obstacles prevent them from acting?
-       - What immediate benefits are they seeking?
-       - What fears or doubts do they have?
-       - What language and tone resonates with them?
-
-    2. PERSUASION ELEMENTS:
-       - How to make the desired action more appealing?
-       - What emotional triggers will resonate most?
-       - How to create a sense of urgency naturally?
-       - What unique value proposition to emphasize?
-       - How to minimize perceived risk?
-
-    Based on your internal analysis, create {cta_count} different CTAs following EXACTLY the formula structure:
-    {formula["description"]}
-    
-    CRITICAL INSTRUCTIONS:
-    - Follow the exact formula structure shown in the description above
-    - Create {cta_count} different CTAs using the same formula pattern
-    - ALL CTAs MUST BE IN SPANISH
-    - DO NOT add postdata (P.S.) to formulas that don't include it in their structure
-    - When a formula includes multiple postdatas (P.S.1 and P.S.2), make sure they focus on DIFFERENT themes and don't repeat the same urgency factors or benefits
-    
-    EXAMPLES TO FOLLOW:
-    {formula["examples"]}
-    
-    Output EXACTLY in this format based on {formula_type}:
-    1. Follow format from {formula["examples"]}
-    
-    2. Follow format from {formula["examples"]}
-    
-    3. Follow format from {formula["examples"]}
-    """
-    
-    response = model.generate_content([full_prompt], generation_config={"temperature": temperature})
-    return response.parts[0].text if response and response.parts else "Error al generar contenido."
-
-# Configurar la aplicación Streamlit
-st.set_page_config(page_title="CTA Generator", page_icon="🎯", layout="wide")
-
-# Leer y mostrar el manual en el sidebar
-with open("manual.md", "r", encoding="utf-8") as file:
-    manual_content = file.read()
-st.sidebar.markdown(manual_content)
-
-# Aplicar estilos
-st.markdown(apply_styles(), unsafe_allow_html=True)
-
-# Título de la app
-st.markdown("<h1>Generador de CTAs Persuasivos</h1>", unsafe_allow_html=True)
-st.markdown("<h3>Crea llamados a la acción que motiven a tu audiencia a dar el siguiente paso.</h3>", unsafe_allow_html=True)
-
-# Remove the duplicate manual expander from here
-
-# Crear dos columnas
-col1, col2 = st.columns([0.4, 0.6])  # 40% for left column, 60% for right column
-
-# Columna izquierda para inputs
-with col1:
-    target_audience = st.text_area(
-        "¿Cuál es tu público objetivo?",
-        placeholder="Ejemplo: Emprendedores que buscan automatizar su negocio..."
-    )
-    
-    product_service = st.text_area(
-        "¿Cuál es tu producto o servicio?",
-        placeholder="Ejemplo: Curso de automatización con IA, Software de gestión..."
-    )
-    
-    desired_action = st.text_area(
-        "¿Qué acción quieres que realicen?",
-        placeholder="Ejemplo: Registrarse al webinar, Descargar la guía gratuita..."
-    )
-    
-    # Mover el botón aquí, antes del acordeón
-    generate_button = st.button("Generar CTAs")
-    
-    with st.expander("Opciones avanzadas"):
-        formula_type = st.selectbox(
-            "Tipo de CTA:",
-            options=list(cta_formulas.keys())
-        )
-        
-        tone_type = st.selectbox(
-            "Tono del CTA:",
-            options=list(tone_settings.keys()),
-        )
-        
-        # Nuevos campos para postdata
-        postdata_theme = st.text_input(
-            "Tema o enfoque para la postdata",
-            placeholder="Ejemplo: urgencia, beneficio, descuento"
-        )
-        
-        cta_count = st.number_input(
-            "Número de llamados a la acción",
-            min_value=1,
-            max_value=5,
-            value=3
-        )
-        
-        temperature = st.slider(
-            "Nivel de creatividad:",
-            min_value=0.0,
-            max_value=2.0,
-            value=1.0,
-            step=0.1,
-            help="Valores más altos generan CTAs más creativos pero menos predecibles."
-        )
-
-# Columna derecha para resultados
-with col2:
-    if generate_button and (response := get_gemini_response(
-        product_service,
-        target_audience,
-        desired_action,
-        formula_type,
-        tone_type,
-        temperature,
-        postdata_theme,
-        cta_count
-    )):
-        st.markdown("### Tus Llamados a la Acción")
-        st.write(response)
+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
+from formulas import headline_formulas
+from angles import angles
+
+# 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 dictionary has been moved to formulas/headline_formulas.py
+
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle):
+    # 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,
+    )
+    
+    # Angle dictionaries have been moved to angles/angle_data.py
+    
+    # Incluir las instrucciones del sistema en el prompt principal
+    system_prompt = f"""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
+
+IMPORTANT ANGLE INSTRUCTIONS:
+- The selected angle MUST be applied to EVERY headline
+- The angle modifies HOW the formula is expressed, not its structure
+- Think of the angle as a "tone overlay" on the formula
+- The formula provides the structure, the angle provides the style
+- Both must work together seamlessly
+
+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
+- Follow the selected angle style while maintaining formula structure"""
+
+    # Iniciar el prompt con las instrucciones del sistema
+    headlines_instruction = f"{system_prompt}\n\n"
+    
+    # Añadir instrucciones de ángulo solo si no es "NINGUNO"
+    if selected_angle != "NINGUNO":
+        headlines_instruction += f"""
+ÁNGULO PRINCIPAL: {selected_angle}
+INSTRUCCIONES DE ÁNGULO ESPECÍFICAS:
+{angles[selected_angle]["instruction"]}
+
+IMPORTANTE: El ángulo {selected_angle} debe aplicarse como una "capa de estilo" sobre la estructura de la fórmula:
+1. Mantén la estructura base de la fórmula intacta
+2. Aplica el tono y estilo del ángulo {selected_angle}
+3. Asegura que cada elemento de la fórmula refleje el ángulo
+4. El ángulo afecta al "cómo" se dice, no al "qué" se dice
+
+EJEMPLOS EXITOSOS DEL ÁNGULO {selected_angle}:
+"""
+        for example in angles[selected_angle]["examples"]:
+            headlines_instruction += f"- {example}\n"
+
+    headlines_instruction += (
+        f"\nTu tarea es crear {number_of_headlines} titulares irresistibles para {target_audience} "
+        f"que capturen la atención instantáneamente y generen curiosidad sobre {product}. "
+    )
+
+    if selected_angle != "NINGUNO":
+        headlines_instruction += f"IMPORTANTE: Cada titular DEBE seguir el ángulo {selected_angle} de manera clara y consistente.\n\n"
+
+    headlines_instruction += (
+        f"Evita menciones obvias de {product} y enfócate en despertar interés genuino"
+    )
+
+    if selected_angle != "NINGUNO":
+        headlines_instruction += f" usando el ángulo seleccionado"
+
+    headlines_instruction += ".\n\n"
+
+    headlines_instruction += (
+        f"IMPORTANTE: Estudia cuidadosamente estos ejemplos de la fórmula seleccionada. "
+        f"Cada ejemplo representa el estilo y estructura a seguir"
+    )
+
+    if selected_angle != "NINGUNO":
+        headlines_instruction += f", adaptados al ángulo {selected_angle}"
+
+    headlines_instruction += ":\n\n"
+
+    # Agregar 5 ejemplos aleatorios de la fórmula
+    random_examples = random.sample(selected_formula['examples'], min(5, len(selected_formula['examples'])))
+
+    headlines_instruction += "EJEMPLOS DE LA FÓRMULA 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"
+
+    # CORRECTO (con indentación):
+    if selected_angle != "NINGUNO":
+        headlines_instruction += f"""
+    RECORDATORIO FINAL:
+    1. Sigue la estructura de la fórmula seleccionada
+    2. Aplica el ángulo como una "capa de estilo"
+    3. Mantén la coherencia entre fórmula y ángulo
+    4. Asegura que cada titular refleje ambos elementos
+
+GENERA AHORA:
+Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura de los ejemplos mostrados.
+"""
+    else:
+        headlines_instruction += f"""
+GENERA AHORA:
+Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura de los ejemplos mostrados.
+"""
+
+    # Enviar el mensaje al modelo (sin condiciones de imagen)
+    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")
+
+# 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)
+
+# Load CSS from file
+with open("styles/main.css", "r") as f:
+    css = f.read()
+    
+# Apply the CSS
+st.markdown(f"<style>{css}</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)
+
+# 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, creatividad y ángulo
+    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())
+        )
+
+        # Automatically use the keys from the angles dictionary
+        # Make sure "NINGUNO" appears first, then the rest alphabetically
+        angle_keys = ["NINGUNO"] + sorted([key for key in angles.keys() if key != "NINGUNO"])
+        selected_angle = st.selectbox(
+            "Selecciona el ángulo para tus titulares",
+            options=angle_keys
+        )
+
+    selected_formula = headline_formulas[selected_formula_key]
+
+    # Botón de enviar
+    submit = st.button("Generar Titulares")
+
+# Mostrar los titulares generados
+if submit:
+    # Check if we have valid inputs
+    has_product = product.strip() != ""
+    has_audience = target_audience.strip() != ""
+    
+    # Valid combination: Product + Audience
+    valid_inputs = has_product and has_audience
+    
+    if valid_inputs and selected_formula:
+        try:
+            generated_headlines = generate_headlines(
+                number_of_headlines,
+                target_audience,
+                product,
+                temperature,
+                selected_formula,
+                selected_angle
+            )
+            col2.markdown(f"""
+                <div class="results-container">
+                    <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:
+        if not selected_formula:
+            col2.error("Por favor, selecciona una fórmula.")
+        else:
+            col2.error("Por favor, proporciona el público objetivo y el producto.")