Update app.py

app.py

@@ -7,13 +7,13 @@ import random
 # Cargar las variables de entorno
 load_dotenv()
 
-# Configurar la API de GenAI
+# Configurar la API de Google
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Función para obtener una mención del producto de manera probabilística
 def get_random_product_mention():
     mentions = ["Indirecta", "Metafórica"]
-    probabilities = [0.50, 0.50]
+    probabilities = [0.50, 0.50]  
     return random.choices(mentions, probabilities)[0]
 
 # Crear la instrucción de mención basada en la opción seleccionada
@@ -24,101 +24,166 @@ def get_mention_instruction(product_mention, product):
         return f"Introduce el producto '{product}' usando una metáfora, conectándolo simbólicamente a la solución que necesita el lector."
     return ""
 
-# Función para generar los beneficios (bullets) basados en el enfoque
-def generate_benefits(focus_points, product, target_audience, creativity, num_bullets):
+# Fórmulas con ejemplos y explicaciones
+benefits_formulas = {
+    "plantilla": {
+        "description": """
+            Crea bullets de 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."
+        ]
+    },
+    "formula_suprema_istvanova": {
+        "description": """
+            La fórmula Suprema de Istvanova: Números + Adjetivo + Palabra Clave + Razón + Promesa.
+            Diseñada para crear beneficios específicos y atractivos que conecten emocionalmente con tu audiencia. Cada elemento cumple una función clave:
+            1. **Números**: Establecen una expectativa clara y medible.
+            2. **Adjetivo**: Añaden emoción y atractivo al mensaje.
+            3. **Palabra Clave**: Reflejan el beneficio principal o tema clave.
+            4. **Razón**: Justifican el beneficio o valor del producto.
+            5. **Promesa**: Muestran el resultado o beneficio que el lector puede esperar.
+        """,
+        "examples": [
+            "5 métodos simples para duplicar tus ventas en menos de 30 días.",
+            "8 trucos secretos para que tu perro obedezca en una semana.",
+            "10 técnicas fáciles que mejorarán tu relación antes de un mes."
+        ]
+    },
+    "formula_aida": {
+        "description": """
+            La fórmula AIDA para beneficios: Atención + Interés + Deseo + Acción.
+            Construye beneficios que sigan estos pasos:
+            - **Atención**: Captura de inmediato el interés del lector.
+            - **Interés**: Detalla cómo el beneficio soluciona un problema específico.
+            - **Deseo**: Despierta el anhelo por lograr el resultado prometido.
+            - **Acción**: Inspira al lector a actuar para obtener el beneficio.
+        """,
+        "examples": [
+            "Descubre cómo atraer clientes automáticamente, sin experiencia previa.",
+            "Imagina tener un cuerpo en forma disfrutando lo que te gusta comer.",
+            "Empieza hoy mismo y experimenta resultados en solo 7 días."
+        ]
+    }
+}
+
+# Función para generar bullets de beneficios
+def generate_benefits(number_of_benefits, target_audience, product, temperature, selected_formula):
     product_mention = get_random_product_mention()
     mention_instruction = get_mention_instruction(product_mention, product)
 
-    # Instrucción del sistema para el modelo
-    system_instruction = (
-        "You are a world-class copywriter with expertise in crafting emotional and persuasive benefits that connect with the target audience. "
-        "Generate highly creative and engaging benefits that resonate with the audience's desires and challenges."
+    # Crear la configuración del modelo
+    generation_config = {
+        "temperature": temperature,  
+        "top_p": 0.65,        # Considerar un poco menos de palabras probables
+        "top_k": 360,        # Aumentar las palabras candidatas para más variedad
+        "max_output_tokens": 8196,  # Mantenerlo igual, pero puedes aumentar si deseas más detalle
+        "response_mime_type": "text/plain",
+    }
+
+    model = genai.GenerativeModel(
+        model_name="gemini-1.5-flash",
+        generation_config=generation_config,
+        system_instruction="You are a world-class copywriter, with expertise in crafting benefits that connect emotionally and address the desires, problems, and motivations of a target audience. Your task is to generate compelling and specific benefit bullets in Spanish based on a given formula. Always respond with a numbered list format, and ensure each benefit is relevant, concise, and action-oriented. Do not include explanations or categories in your output."
     )
 
-    # Base del prompt
-    prompt_base = f"""
-Eres un experto en copywriting y tu objetivo es crear {{num_bullets}} bullets persuasivos que conecten emocionalmente con la audiencia {{target_audience}}.
-Cada bullet debe abordar sus problemas, deseos o situaciones, mostrando cómo se pueden mejorar o solucionar gracias a una solución específica.
-La idea es resaltar el valor que aporta la solución, sin que suene como una venta forzada.
-Además, menciona el producto utilizando el siguiente enfoque: {{mention_instruction}}.
-Usa enfoques creativos para conectar los beneficios del producto con lo que realmente le importa a la audiencia.
-Crea {{num_bullets}} bullets persuasivos que muestren cómo el producto puede transformar la situación de la audiencia.
-"""
-
-    benefits = []
-    for point in focus_points[:num_bullets]:
-        specific_prompt = prompt_base.format(
-            num_bullets=num_bullets,
-            target_audience=target_audience,
-            mention_instruction=mention_instruction
-        ) + f"\n\nEnfoque: {point}\n"
-
-        # Configuración del modelo y generación de contenido
-        response = genai.generate_text(
-            model="gemini-1.5-flash",
-            prompt=specific_prompt,
-            temperature=creativity,
-            top_p=0.65,
-            top_k=280,
-            max_output_tokens=2048,
-        )
-
-        if response and response.generations:
-            bullet = response.generations[0].text.strip()
-            benefits.append(bullet)
-        else:
-            benefits.append("No se pudo generar un beneficio para este enfoque.")
-
-    return benefits
+    # Crear un mensaje para el modelo, destacando la audiencia, el producto, la fórmula seleccionada y los ejemplos
+    benefits_instruction = (
+        f"Tu tarea es crear {number_of_benefits} beneficios específicos y atractivos diseñados para {target_audience}. "
+    f"El objetivo es conectar emocionalmente y destacar cómo {product} puede mejorar la vida del lector. Asegúrate de que cada beneficio sea "
+    f"llamativo, persuasivo y relevante, siguiendo la estructura de la fórmula seleccionada, que puedes ver a continuación: "
+    f"\n\n{selected_formula['description']}\n\n"
+    f"Revisa los siguientes ejemplos de cómo esta fórmula puede ser utilizada con éxito:\n"
+    f"- {selected_formula['examples'][0]}\n"
+    f"- {selected_formula['examples'][1]}\n"
+    f"- {selected_formula['examples'][2]}\n\n"
+    f"Usa estos ejemplos como inspiración y asegúrate de incluir la siguiente mención en los beneficios generados: {mention_instruction}. "
+    f"No expliques las fórmulas ni la estructura en la salida, solo proporciona los beneficios. "
+    f"Recuerda, el objetivo es inspirar acción y generar deseo en el lector."
+    )
 
-# Configuración de Streamlit
-st.set_page_config(page_title="Impact Bullet Generator", layout="wide")
+    chat_session = model.start_chat(
+        history=[
+            {
+                "role": "user",
+                "parts": [benefits_instruction],
+            },
+        ]
+    )
 
-# Leer el contenido del archivo manual.md
-with open("manual.md", "r", encoding="utf-8") as file:
-    manual_content = file.read()
+    response = chat_session.send_message("Genera los beneficios")  # Enviar mensaje para obtener la respuesta
+    return response.text  # Regresar la respuesta directamente
 
-# Mostrar el contenido del manual en el sidebar
-st.sidebar.markdown(manual_content)
+# Configurar la interfaz de usuario con Streamlit
+st.set_page_config(page_title="Bullet Benefits Generator", layout="wide")
 
 # Centrar el título y el subtítulo
-st.markdown("<h1 style='text-align: center;'>Impact Bullet Generator</h1>", unsafe_allow_html=True)
-st.markdown("<h4 style='text-align: center;'>Transforma los pensamientos de tu audiencia en beneficios persuasivos que inspiren a la acción.</h4>", unsafe_allow_html=True)
+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("""
+    <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])
+col1, col2 = st.columns([1, 2])  
 
 # Columnas de entrada
 with col1:
-    target_audience = st.text_input("Público objetivo:", placeholder="Ejemplo: Estudiantes universitarios")
-    product = st.text_input("Producto relacionado:", placeholder="Ejemplo: Curso de productividad")
-    
-    # Slider para la creatividad
-    creativity = st.slider("Creatividad (Temperatura)", min_value=0.0, max_value=1.0, value=0.7, step=0.1)
-    
-    # Slider para el número de bullets
-    num_bullets = st.slider("Número de Bullets", min_value=1, max_value=10, value=5, step=1)
-    
-    focus_points = st.multiselect(
-        "Selecciona los enfoques que deseas utilizar:",
-        ["Curiosidad", "CASI Imposible", "Autoridad y Credibilidad", "Contraste"],
-        default=[]
-    )
+    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)
+
+    # Crear un único acordeón para fórmula y creatividad
+    with st.expander("Personaliza tus beneficios"):
+        temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+        
+        selected_formula_key = st.selectbox(
+            "Selecciona una fórmula para tus beneficios",
+            options=list(benefits_formulas.keys())
+        )
+        selected_formula = benefits_formulas[selected_formula_key]
 
+    # Botón de enviar
     submit = st.button("Generar Beneficios")
 
 # Mostrar los beneficios generados
-with col2:
-    if submit:
-        if focus_points and product and target_audience:
-            benefits = generate_benefits(focus_points, product, target_audience, creativity, num_bullets)
-            formatted_benefits = '<br style="line-height: 2;">'.join(benefits)
-            
-            st.markdown(f"""
+if submit:
+    if target_audience and product and selected_formula:
+        try:
+            # Obtener la respuesta del modelo
+            generated_benefits = generate_benefits(number_of_benefits, target_audience, product, temperature, selected_formula)
+            col2.markdown(f"""
                 <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">
-                    <h4>Mira los beneficios generados:</h4>
-                    <p style="line-height: 2;">{formatted_benefits}</p>
+                    <h4>Explora los beneficios generados:</h4>
+                    <p>{generated_benefits}</p>
                 </div>
             """, unsafe_allow_html=True)
-        else:
-            st.error("Por favor, proporciona al menos un enfoque, un producto y un público objetivo para generar beneficios.")
+        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.")