Update app.py

app.py

@@ -4,14 +4,16 @@ import os
 import google.generativeai as genai
 import random
 
+# Cargar las variables de entorno
 load_dotenv()
 
+# Configurar la API de GenAI
 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
@@ -21,40 +23,28 @@ def get_mention_instruction(product_mention, product):
     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 ""
-    
+
 # Función para generar los beneficios (bullets) basados en el enfoque
 def generate_benefits(focus_points, product, target_audience, creativity, num_bullets):
-    model = genai.GenerativeModel("gemini-1.5-flash")
+    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 hooks, headlines, and subject lines that immediately "
-        "capture the reader's attention, prompting them to open the email or continue reading. Your skill lies in deeply understanding "
-        "the emotions, desires, and challenges of a specific audience. You are also an expert in creating benefits that connect symptoms "
-        "with problems, allowing you to design personalized strategies that resonate and motivate action. You know how to use proven "
-        "structures to attract your target audience, generating interest and creating a powerful connection. Your task is to generate "
-        "unusual, creative, and fascinating subject lines or headlines that spark curiosity and encourage the reader to engage further. "
-        "Respond in Spanish and use a numbered list format. Important: Only answer with subject lines, never include explanations or categories."
+        "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."
     )
 
-    # Modificación en el prompt para integrar menciones del producto
-prompt_base = f"""
+    # 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.
-Piensa en cómo puedes captar la atención de la audiencia con frases impactantes, como si estuvieras hablando directamente con ellos.
-La idea es resaltar el valor que aporta la solución, sin hacer que el producto suene como una venta forzada.
-Además, asegúrate de mencionar el producto utilizando el siguiente enfoque: {{mention_instruction}}.
-Los bullets deben ser claros, directos y generar curiosidad, invitando a la acción de manera natural.
-Evita la jerga técnica y mantén la simplicidad, enfocándote en cómo el producto se integra de manera natural en la vida de la audiencia para mejorarla.
+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 resolver o transformar una situación para la audiencia.
+Crea {{num_bullets}} bullets persuasivos que muestren cómo el producto puede transformar la situación de la audiencia.
 """
 
-# Generación del prompt específico con mención
-def generate_benefits(focus_points, product, target_audience, creativity, num_bullets):
-    product_mention = get_random_product_mention()
-    mention_instruction = get_mention_instruction(product_mention, product)
-
     benefits = []
     for point in focus_points[:num_bullets]:
         specific_prompt = prompt_base.format(
@@ -63,20 +53,18 @@ def generate_benefits(focus_points, product, target_audience, creativity, num_bu
             mention_instruction=mention_instruction
         ) + f"\n\nEnfoque: {point}\n"
 
-        response = genai.GenerativeModel(
-            model_name="gemini-1.5-flash",
-            generation_config={
-                "temperature": creativity,
-                "top_p": 0.65,
-                "top_k": 280,
-                "max_output_tokens": 2048,
-                "response_mime_type": "text/plain",
-            },
-            system_instruction=system_instruction
-        ).generate_content([specific_prompt])
-
-        if response and response.parts:
-            bullet = response.parts[0].text.strip()
+        # 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.")
@@ -84,7 +72,7 @@ def generate_benefits(focus_points, product, target_audience, creativity, num_bu
     return benefits
 
 # Configuración de Streamlit
-st.set_page_config(page_title="Quick Prompt", layout="wide")
+st.set_page_config(page_title="Impact Bullet Generator", layout="wide")
 
 # Leer el contenido del archivo manual.md
 with open("manual.md", "r", encoding="utf-8") as file:
@@ -120,16 +108,17 @@ with col1:
     submit = st.button("Generar Beneficios")
 
 # Mostrar los beneficios generados
-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)
-        
-        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, proporciona al menos un enfoque, un producto y un público objetivo para generar beneficios.")
+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"""
+                <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:
+            st.error("Por favor, proporciona al menos un enfoque, un producto y un público objetivo para generar beneficios.")