Update app.py

app.py

@@ -2,10 +2,8 @@ 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()
@@ -13,7 +11,7 @@ load_dotenv()
 # Configurar la API de Google
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Función para generar bullets de beneficios
+# Función para generar el perfil de cliente ideal
 @st.cache_resource
 def get_model(temperature):
     generation_config = {
@@ -21,39 +19,39 @@ def get_model(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."
+def generate_buyer_persona(product, skills, temperature):
+    if not product or not skills:
+        return "Por favor, completa los campos de producto y habilidades."
     
     model = get_model(temperature)
-    benefits_instruction = create_instruction(
-        number_of_benefits, 
-        target_audience, 
-        product, 
-        selected_formula,
-        selected_angle
+    instruction = create_instruction(
+        product_service=product,
+        skills=skills
     )
     
-    response = model.generate_content([benefits_instruction], generation_config={"temperature": temperature})
-    return response.parts[0].text if response and response.parts else "Error generating content."
+    response = model.generate_content([instruction], generation_config={"temperature": temperature})
+    return response.parts[0].text if response and response.parts else "Error generando el perfil de cliente ideal."
 
 # 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)
+st.set_page_config(page_title="Generador de Cliente Ideal", page_icon="👤", layout="wide")
+
+# Leer el contenido del archivo manual.md si existe
+try:
+    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)
+except FileNotFoundError:
+    st.sidebar.warning("Manual not found. Please create a manual.md file.")
+except Exception as e:
+    st.sidebar.error(f"Error loading manual: {str(e)}")
 
 # 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)
+st.markdown("<h1 style='text-align: center;'>Generador de Perfil de Cliente Ideal</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center;'>Crea un perfil detallado de tu cliente ideal basado en tu producto y habilidades.</h4>", unsafe_allow_html=True)
 
 # Añadir CSS personalizado para el botón
 st.markdown(styles["button"], unsafe_allow_html=True)
@@ -61,62 +59,44 @@ st.markdown(styles["button"], unsafe_allow_html=True)
 # Crear columnas
 col1, col2 = st.columns([1, 2])  
 
-# Columnas de entrada
-# Inside the column section
+# Columna 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_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.")
+    producto = st.text_input("¿Qué producto o servicio ofreces?", placeholder="Ejemplo: Curso de Inglés")
+    habilidades = st.text_input("¿Cuáles son tus habilidades principales?", placeholder="Ejemplo: Enseñanza, comunicación, diseño de contenidos")
+    
+    # Nivel de creatividad con slider
+    creatividad = st.slider("Nivel de creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1)
+    
+    # Botón para generar
+    submit = st.button("GENERAR PERFIL DE CLIENTE IDEAL")
+
+# Columna de resultados
+with col2:
+    if submit:
+        if producto and habilidades:
+            with st.spinner('Generando perfil de cliente ideal...'):
+                perfil_cliente = generate_buyer_persona(
+                    producto,
+                    habilidades,
+                    creatividad
+                )
+                
+                if not isinstance(perfil_cliente, str):
+                    st.error("Error al generar el perfil de cliente ideal")
+                else:
+                    st.markdown(f"""
+                        <div style="{styles['results_container']}">
+                            <h3>Tu Cliente Ideal</h3>
+                            {perfil_cliente}
+                        </div>
+                    """, unsafe_allow_html=True)
+                    
+                    # Opción para descargar
+                    st.download_button(
+                        label="Descargar Perfil",
+                        data=perfil_cliente,
+                        file_name="cliente_ideal.md",
+                        mime="text/markdown"
+                    )
+        else:
+            st.warning("Por favor, completa los campos de producto y habilidades antes de generar el perfil.")