Update app.py

app.py

@@ -2,7 +2,6 @@ from dotenv import load_dotenv
 import streamlit as st
 import os
 import google.generativeai as genai
-import random
 
 # Cargar las variables de entorno
 load_dotenv()
@@ -10,97 +9,28 @@ load_dotenv()
 # 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 = ["Directa", "Indirecta", "Metafórica"]
-    probabilities = [0.35, 0.25, 0.40]
-    return random.choices(mentions, probabilities)[0]
-
-# Crear la instrucción de mención basada en la opción seleccionada
-def get_mention_instruction(product_mention, product):
-    if product_mention == "Directa":
-        return f"Introduce directamente el producto '{product}' como la solución clara al problema que enfrenta el lector, de manera conversacional, no forzada."
-    elif product_mention == "Indirecta":
-        return f"Referencia sutilmente el producto '{product}' como una posible solución al problema del lector sin nombrarlo explícitamente."
-    elif product_mention == "Metafórica":
-        return f"Introduce el producto '{product}' usando una metáfora, conectándolo simbólicamente a la solución que necesita el lector."
-    return ""
-
-# Ejemplos de bullets
-benefit_types = {
-    "directos": [
-        "El armario del baño es el mejor lugar para guardar medicamentos, ¿verdad? Incorrecto. Es el peor. Los hechos están en la página 10.",
-        "El mejor tiempo verbal que le da a tus clientes la sensación de que ya te han comprado.",
-        "La historia de un joven emprendedor que transformó su vida aplicando esta técnica simple pero poderosa."
-    ],
-    "misterios": [
-        "Los misterios de cómo algunas personas parecen tener éxito sin esfuerzo, mientras otras luchan. La clave está en esta pequeña diferencia.",
-        "Los misterios de cómo una técnica sencilla te permite reducir el estrés al instante, sin necesidad de dejar tu trabajo o cambiar tu estilo de vida."
-    ],
-    "leyendas": [
-        "La leyenda de aquellos que dominaron la productividad con un solo hábito. ¿Te atreves a descubrirlo?",
-        "La verdad que nunca te han contado en la escuela, o en casa, sobre cómo ganarte la vida con la música."
-    ],
-    "historias_personales": [
-        "La historia de un padre ocupado que, con solo 10 minutos al día, logró transformar su salud y bienestar.",
-        "¿Sabías que muchas personas están usando este método y han mejorado su bienestar en solo 7 días?"
-    ],
-    "preguntas_retoricas": [
-        "¿Cuándo es una buena idea decirle a una chica que te gusta? Si no se lo dices en ese momento, despídete de conocerla íntimamente."
-    ],
-}
-
-# Generar el resultado utilizando el modelo con la instrucción de bullets específica
-def generate_bullets(number_of_bullets, target_audience, product, call_to_action, temperature):
-    product_mention = get_random_product_mention()
-    mention_instruction = get_mention_instruction(product_mention, product)
-
-    # Configuración del modelo
-    generation_config = {
-        "temperature": temperature,
-        "top_p": 0.85,
-        "top_k": 128,
-        "max_output_tokens": 2048,
-        "response_mime_type": "text/plain",
-    }
-
-    # Crear la instrucción del sistema
-    system_instruction = (
-        f"Eres un experto copywriter especializado en escribir bullets atractivos, curiosos e inusuales para {target_audience} sobre {product} que promueven la acción de {call_to_action}. "
-        f"Tu tarea es ayudarme a escribir {number_of_bullets} bullets que destaquen los beneficios de {product}, escritos en un formato de lista numerada y cada bullet ocupa un reglón sin importar su longitud. "
-        "Recuerda que un buen bullet debe tener:\n\n"
-        f"1. Acción: Establece el tono y el punto de partida del mensaje, en este caso, 'Cocina como un superhéroe' crea una imagen poderosa e inspiradora que engancha a {target_audience} y los motiva a actuar de manera proactiva.\n"
-        f"2. Beneficio: Describe lo que se gana o se obtiene al seguir al conseguir {product} mencionada, en este ejemplo, 'al preparar comidas nutritivas y deliciosas' enfatiza el valor añadido que la audiencia experimentará al realizar esa acción.\n"
-        f"3. Resultado: Muestra el desenlace concreto o el impacto directo, y es crucial ser específico. En lugar de un término general como 'en tiempo récord', se debe optar por algo más preciso como 'en tan solo 15 minutos'. Este nivel de especificidad le da un sentido claro de urgencia y eficiencia a conseguir {product}, haciendo que el mensaje sea más creíble y atractivo.\n"
-        "Important: Only answer bullets, never include explanations or categories, like this: 'Registrarme ahora y descubrir cómo encontrar un poco de paz en medio del caos. (Este bullet apela al deseo de Han Solo de encontrar un momento de tranquilidad en su vida agitada.).'\n"
-        f"Basate en estos ejemplos como respuesta, escribe {number_of_bullets} bullets enfocados a beneficios:"
-        "- 'Reduce el estrés y la ansiedad como un ancla que te mantiene firme en medio de la tormenta.'\n"
-        "- 'Aumenta tu energía y concentración para navegar con mayor seguridad y precisión.'\n"
-        "- 'Mejora tu flexibilidad y movilidad para adaptarte a cualquier situación con mayor agilidad.'\n"
-        "- 'Encuentra la paz interior como un faro que te guía hacia la calma en medio del caos.'\n"
-        "- 'Conecta contigo mismo para descubrir tu propio rumbo y navegar con mayor confianza.'\n"
-        "- 'Aprende técnicas para gestionar el tiempo y la energía para optimizar tu viaje y disfrutar de cada momento.'\n"
-        "Los bullets no deben de llevar explicaciones, ni parentesis, ni dos puntos."
-        "Asegúrate de que cada bullet siga la estructura de 'Acción + Beneficio + Resultado', y evita incluir explicaciones como 'Acción: Descubrir' o 'Resultado: Un oasis de paz en medio del caos'.\n"
-    )
-
-    # Configuración del modelo generativo
-    model = genai.GenerativeModel(
-        model_name="gemini-1.5-flash",
-        generation_config=generation_config,
+# Nueva función para generar beneficios basados en enfoques seleccionados por el usuario
+def generate_benefits(focus_points, product, target_audience):
+    # Prompt basado en las descripciones de los bullets
+    prompt = (
+        "Qué son los bullets: "
+        "Son pequeños anzuelos que capturan la atención sin que parezca que estás esforzándote en ello. "
+        "Hacen que hasta los más distraídos digan: \"Oye, esto suena interesante\". "
+        "Son como los tráilers de una película, pero en lugar de vender entradas, venden tus ideas. "
+        "Son tan parecidos a la lista del súper, que cuando los leen gastan dinero como si estuvieras comprando en el supermercado. "
+        "Considéralos como pequeños headlines, así que las formulas de estos funcionan para crearlos. "
+        "Ayudan a que tus textos no se vean como parrafadas, porque a nadie le gusta leer eso. "
+        "Un bullet bien escrito destaca en el texto. "
+        "Van enfocados en el beneficio emocional o práctico del producto."
     )
 
-    # Generar el resultado utilizando el modelo con la instrucción de bullets específica
-    try:
-        response = model.generate_content([system_instruction])
-        
-        # Extraer el texto de la respuesta
-        generated_bullets = response.candidates[0].content.parts[0].text.strip()  # Modificado aquí
-        
-        # Retornar el resultado
-        return generated_bullets
-    except Exception as e:
-        raise ValueError(f"Error al generar los bullets: {str(e)}")
+    # Generar beneficios con el producto y la audiencia como referencia
+    benefits = []
+    for point in focus_points:
+        benefits.append(
+            f"{prompt} Enfócate en: {point}. Producto: {product}. Público objetivo: {target_audience}."
+        )
+    return benefits
 
 # Configurar la interfaz de usuario con Streamlit
 st.set_page_config(page_title="Quick Prompt", layout="wide")
@@ -114,7 +44,7 @@ st.sidebar.markdown(manual_content)
 
 # 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 balas persuasivas que inspiren a la acción.</h4>", 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)
 
 # Añadir CSS personalizado para el botón
 st.markdown("""
@@ -144,33 +74,38 @@ 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")
-    call_to_action = st.text_input("¿Qué acción deseas que tomen?", placeholder="Ejemplo: Inscribirse al curso")
-    number_of_bullets = st.selectbox("Número de bullets", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
-    temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
-
-    # Botón de enviar
+    with st.expander("Enfoques para los beneficios"):
+        st.markdown("""
+        1. **Curiosidad**
+        - Su función es dejar al lector enganchado, queriendo saber más.
+        2. **CASI Imposible**
+        - Afirmaciones que parecen difíciles de creer, pero no son completamente irreales.
+        3. **Autoridad y Credibilidad**
+        - Establece confianza mediante hechos respaldados.
+        4. **Contraste**
+        - Confronta ideas para captar la atención.
+        """)
+
+    focus_points = st.multiselect(
+        "Selecciona los enfoques que deseas utilizar:",
+        ["Curiosidad", "CASI Imposible", "Autoridad y Credibilidad", "Contraste"],
+        default=[]
+    )
+    product = st.text_input("Producto relacionado:", placeholder="Ejemplo: Curso de productividad")
+    target_audience = st.text_input("Público objetivo:", placeholder="Ejemplo: Estudiantes universitarios")
     submit = st.button("Generar Beneficios")
 
 # Mostrar los beneficios generados
 if submit:
-    if target_audience and product and call_to_action:
-        try:
-            # Obtener la respuesta del modelo
-            generated_bullets = generate_bullets(number_of_bullets, target_audience, product, call_to_action, temperature)
-            
-            # Formatear la salida usando etiquetas <br> para saltos de línea
-            bullets_list = generated_bullets.split('\n')
-            formatted_bullets = '<br style="line-height: 2;">'.join([bullet for bullet in bullets_list if bullet])  # Agregar interlineado
-
-            col2.markdown(f"""
-                <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">
-                    <h4>Mira la magia en acción con estos bullets:</h4>
-                    <p style="line-height: 2;">{formatted_bullets}</p>
-                </div>
-            """, unsafe_allow_html=True)
-        except Exception as e:
-            col2.error(f"Error al generar los bullets: {str(e)}")  # Cambiar a mostrar el error en col2
+    if focus_points and product and target_audience:
+        benefits = generate_benefits(focus_points, product, target_audience)
+        formatted_benefits = '<br style="line-height: 2;">'.join(benefits)
+        
+        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>
+            </div>
+        """, unsafe_allow_html=True)
     else:
-        col2.error("Por favor, completa todos los campos.")  # Cambiar a mostrar el error en col2
+        col2.error("Por favor, proporciona al menos un enfoque, un producto y un público objetivo para generar beneficios.")