Update app.py

app.py

@@ -1,122 +1,59 @@
-from dotenv import load_dotenv
-import streamlit as st
-import os
-import google.generativeai as genai
-from bullet_formulas import bullet_formulas
-from style import styles
-from prompts import create_instruction
-from angles import bullet_angles
-
-# Cargar las variables de entorno
-load_dotenv()
-
-# Configurar la API de Google
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-
-# Función para generar bullets de beneficios
-@st.cache_resource
-def get_model(temperature):
-    generation_config = {
-        "temperature": temperature,
-    }
-    return genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
-
-# Update the generate_benefits function to include selected_angle
-def generate_benefits(number_of_benefits, target_audience, product, temperature, selected_formula, selected_angle):
-    if not target_audience or not product:
-        return "Por favor, completa todos los campos requeridos."
+system_prompt = """You are a world-class expert copywriter, experienced in creating benefits that emotionally connect and address the desires, problems, and motivations of the target audience.
+
+OBJECTIVE:
+- Generate convincing and specific benefit bullets in Spanish
+- Connect emotionally with the audience
+- Address real desires, problems, and motivations
+- Maintain natural and conversational language
+- Orient each benefit towards action
+
+FORMAT RULES:
+- Each benefit must start with "• "
+- One benefit per line
+- No numbers at the beginning
+- No explanations or categories
+- Add a line break between each benefit
+- Never include : symbols in bullets
+- Each benefit must be a complete and concise phrase
+
+BENEFIT STRUCTURE:
+- Must be relevant to target audience
+- Must show a specific result
+- Must include an emotional element
+- Must eliminate an objection or pain point
+- Must inspire immediate action
+
+EJEMPLO DE FORMATO:
+• Transforma tu negocio con estrategias probadas que duplican tus ingresos en 90 días, sin sacrificar tu tiempo en familia.
+
+• Domina las técnicas más efectivas para conquistar tu mercado, mientras mantienes el equilibrio entre trabajo y vida personal.
+
+• Implementa sistemas automatizados que hacen crecer tu empresa incluso mientras duermes, eliminando la necesidad de trabajar más horas.
+
+IMPORTANT:
+- Each benefit must be unique and specific
+- Avoid repetitions and generalities
+- Maintain a persuasive but honest tone
+- Adapt language to audience comprehension level
+- Focus on tangible and measurable results
+"""
     
-    model = get_model(temperature)
-    benefits_instruction = create_instruction(
-        number_of_benefits, 
-        target_audience, 
-        product, 
-        selected_formula,
-        selected_angle
-    )
+def create_instruction(number_of_benefits, target_audience, product, selected_formula, selected_angle):
+    angle_instruction = ""
+    if selected_angle["description"] != "Generate the bullet without any specific angle":
+        angle_instruction = f"\nApply this angle: {selected_angle['description']}\nStyle: {selected_angle['style']}\nUse these keywords as inspiration: {', '.join(selected_angle['keywords'])}"
     
-    response = model.generate_content([benefits_instruction], generation_config={"temperature": temperature})
-    return response.parts[0].text if response and response.parts else "Error generating content."
-
-# Configurar la interfaz de usuario con Streamlit
-st.set_page_config(page_title="Bullet Benefits Generator", 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)
-
-# Ocultar elementos de la interfaz
-st.markdown(styles["main_layout"], unsafe_allow_html=True)
-
-# Centrar el título y el subtítulo
-st.markdown("<h1 style='text-align: center;'>Bullet Benefits Generator</h1>", unsafe_allow_html=True)
-st.markdown("<h4 style='text-align: center;'>Transforma características en beneficios irresistibles que conectan emocionalmente con tu audiencia.</h4>", unsafe_allow_html=True)
-
-# Añadir CSS personalizado para el botón
-st.markdown(styles["button"], unsafe_allow_html=True)
-
-# Crear columnas
-col1, col2 = st.columns([1, 2])  
-
-# Columnas de entrada
-# Inside the column section
-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_benefits = st.selectbox("Número de Beneficios", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
-
-    # Fix indentation - remove extra spaces
-    with st.expander("Opciones avanzadas"):
-        selected_formula_key = st.selectbox(
-            "Tipo de bullet",  # Added label
-            options=list(bullet_formulas.keys()),
-            label_visibility="visible"
-        )
-        selected_formula = bullet_formulas[selected_formula_key]
-
-        selected_angle_key = st.selectbox(
-            "Ángulo del bullet",  # Added label
-            options=list(bullet_angles.keys()),
-            label_visibility="visible"
-        )
-        selected_angle = bullet_angles[selected_angle_key]
-
-        temperature = st.slider(
-            "Nivel de creatividad",  # Added label
-            min_value=0.0, 
-            max_value=2.0, 
-            value=1.0, 
-            step=0.1,
-            label_visibility="visible"
-        )
-
-    submit = st.button("GENERAR LOS BULLETS POINTS")
-
-# Mostrar los beneficios generados
-# Update the submit section to include selected_angle
-if submit:
-    if target_audience and product and selected_formula:
-        with st.spinner('Generando beneficios...'):
-            generated_benefits = generate_benefits(
-                number_of_benefits, 
-                target_audience, 
-                product, 
-                temperature, 
-                selected_formula,
-                selected_angle
-            )
-            if not isinstance(generated_benefits, str):
-                col2.error("Error al generar beneficios")
-            else:
-                col2.markdown(f"""
-                    <div style="{styles['results_container']}">
-                        <h3>Beneficios Generados:</h3>
-                        <p>{generated_benefits}</p>
-                    </div>
-                """, unsafe_allow_html=True)
-
-    else:
-        col2.warning("Por favor, completa todos los campos antes de generar beneficios.")
+    return (
+        f"{system_prompt}\n\n"
+        f"Your task is to create {number_of_benefits} irresistible benefits designed for {target_audience}. "
+        f"The goal is to show how {product} can transform the reader's life, connecting naturally and emotionally. "
+        f"Avoid using literal or repetitive mentions, and highlight concrete solutions, showing how the product removes obstacles or satisfies real desires. "
+        f"{angle_instruction}\n"
+        f"IMPORTANT: Keep bullets short and direct, maximum 100 characters per bullet. "
+        f"Use the selected formula as a guide:\n\n{selected_formula['description']}\n\n"
+        f"Get inspired by these examples:\n"
+        f"- {selected_formula['examples'][0]}\n"
+        f"- {selected_formula['examples'][1]}\n"
+        f"- {selected_formula['examples'][2]}\n\n"
+        f"Your goal is to inspire desire and action, avoiding explanations or categories in the response."
+    )