Update app.py

app.py

@@ -20,8 +20,8 @@ if 'habilidades' not in st.session_state:
     st.session_state.habilidades = ""
 if 'creatividad' not in st.session_state:
     st.session_state.creatividad = 1.0
-if 'incluir_niveles_conciencia' not in st.session_state:
-    st.session_state.incluir_niveles_conciencia = False
+if 'nivel_conciencia' not in st.session_state:
+    st.session_state.nivel_conciencia = "Ninguno"
 
 # Función para generar el perfil de cliente ideal
 @st.cache_resource
@@ -31,7 +31,7 @@ def get_model(temperature):
     }
     return genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
 
-def generate_buyer_persona(product, skills, target_audience, temperature, include_consciousness_levels=False):
+def generate_buyer_persona(product, skills, target_audience, temperature, consciousness_level="Ninguno"):
     if not product or not skills:
         return "Por favor, completa los campos de producto y habilidades."
     
@@ -40,7 +40,7 @@ def generate_buyer_persona(product, skills, target_audience, temperature, includ
         product_service=product,
         skills=skills,
         target_audience=target_audience,
-        include_consciousness_levels=include_consciousness_levels
+        consciousness_level=consciousness_level
     )
     
     # Añadir instrucción explícita para respuesta en español
@@ -121,23 +121,25 @@ with col1:
                                key="creatividad_slider")
         st.session_state.creatividad = creatividad
         
-        # Opción para incluir análisis de niveles de conciencia
-        incluir_niveles = st.checkbox(
-            "Incluir análisis de los 5 niveles de conciencia", 
-            value=st.session_state.incluir_niveles_conciencia,
-            help="Analiza al cliente ideal desde los 5 niveles de conciencia de Eugene Schwartz"
+        # Selector de nivel de conciencia
+        nivel_conciencia = st.selectbox(
+            "Nivel de conciencia del cliente ideal",
+            ["Ninguno", "Desconocido", "Consciente del problema", "Consciente de la solución", 
+             "Consciente del producto", "Consciente de la compra"],
+            index=0,
+            help="Selecciona el nivel de conciencia en el que se encuentra tu cliente ideal"
         )
-        st.session_state.incluir_niveles_conciencia = incluir_niveles
+        st.session_state.nivel_conciencia = nivel_conciencia
         
-        if incluir_niveles:
-            st.info("""
-            **Los 5 niveles de conciencia:**
-            1. **Desconocido**: No sabe que tiene un problema
-            2. **Consciente del problema**: Reconoce el problema pero no sabe cómo resolverlo
-            3. **Consciente de la solución**: Conoce posibles soluciones pero no sabe cuál elegir
-            4. **Consciente del producto**: Conoce tu producto pero no está convencido
-            5. **Consciente de la compra**: Está listo para comprar pero necesita un incentivo final
-            """)
+        if nivel_conciencia != "Ninguno":
+            niveles_info = {
+                "Desconocido": "No sabe que tiene un problema. Tu mensaje debe crear conciencia del problema.",
+                "Consciente del problema": "Reconoce el problema pero no sabe cómo resolverlo. Tu mensaje debe mostrar la urgencia del problema.",
+                "Consciente de la solución": "Conoce posibles soluciones pero no sabe cuál elegir. Tu mensaje debe presentar las soluciones disponibles.",
+                "Consciente del producto": "Conoce tu producto pero no está convencido. Tu mensaje debe probar la superioridad de tu oferta.",
+                "Consciente de la compra": "Está listo para comprar pero necesita un incentivo final. Tu mensaje debe facilitar la decisión de compra."
+            }
+            st.info(f"**{nivel_conciencia}**: {niveles_info.get(nivel_conciencia, '')}")
     
     # Botón para generar
     submit = st.button("GENERAR PERFIL DE CLIENTE IDEAL", on_click=update_profile)
@@ -154,7 +156,7 @@ with col2:
                     st.session_state.habilidades,
                     st.session_state.publico_objetivo,
                     st.session_state.creatividad,
-                    st.session_state.incluir_niveles_conciencia
+                    st.session_state.nivel_conciencia
                 )
                 # Guardar en session_state
                 st.session_state.perfil_cliente = perfil_cliente
@@ -186,8 +188,8 @@ with col2:
                 st.download_button(
                     label="Descargar Perfil",
                     data=st.session_state.perfil_cliente,
-                    file_name="cliente_ideal.md",
-                    mime="text/markdown"
+                    file_name="cliente_ideal.txt",
+                    mime="text/plain"
                 )
         else:
             st.warning("Por favor, completa los campos de producto y habilidades antes de generar el perfil.")