Update app.py

app.py

@@ -29,16 +29,57 @@ def get_model(temperature):
     }
     return genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
 
+# Modificar la función generate_buyer_persona para manejar diferentes combinaciones
 def generate_buyer_persona(product, skills, target_audience, temperature):
-    if not product or not skills:
-        return "Por favor, completa los campos de producto y habilidades."
+    # Verificar qué información tenemos disponible
+    has_product = bool(product.strip())
+    has_skills = bool(skills.strip())
+    has_target = bool(target_audience.strip())
+    
+    # Si no tenemos ni producto ni habilidades, no podemos generar un perfil
+    if not has_product and not has_skills:
+        return "Por favor, completa al menos uno de los campos: producto/servicio o habilidades."
     
     model = get_model(temperature)
-    instruction = create_instruction(
-        product_service=product,
-        skills=skills,
-        target_audience=target_audience
-    )
+    
+    # Crear instrucción adaptada a la información disponible
+    if has_product and has_skills:
+        # Caso 1: Tenemos producto y habilidades
+        instruction = create_instruction(
+            product_service=product,
+            skills=skills,
+            target_audience=target_audience if has_target else ""
+        )
+    elif has_product:
+        # Caso 2: Solo tenemos producto
+        instruction = f"""
+        Crea un perfil detallado de cliente ideal para el siguiente producto/servicio: {product}.
+        {f'El público objetivo es: {target_audience}.' if has_target else 'Infiere el público objetivo más adecuado.'}
+        
+        Incluye:
+        - Demografía (edad, género, ubicación, nivel educativo, ingresos)
+        - Psicografía (valores, intereses, estilo de vida)
+        - Necesidades y deseos
+        - Puntos de dolor que el producto/servicio resuelve
+        - Objeciones comunes y cómo abordarlas
+        - Canales preferidos para recibir información
+        """
+    else:
+        # Caso 3: Solo tenemos habilidades
+        instruction = f"""
+        Crea un perfil detallado de cliente ideal para alguien con las siguientes habilidades: {skills}.
+        {f'El público objetivo es: {target_audience}.' if has_target else 'Infiere el público objetivo más adecuado.'}
+        
+        Sugiere productos o servicios que podrían ofrecerse con estas habilidades.
+        
+        Incluye:
+        - Demografía (edad, género, ubicación, nivel educativo, ingresos)
+        - Psicografía (valores, intereses, estilo de vida)
+        - Necesidades y deseos
+        - Puntos de dolor que podrían resolverse con estas habilidades
+        - Objeciones comunes y cómo abordarlas
+        - Canales preferidos para recibir información
+        """
     
     # Añadir instrucción explícita para respuesta en español
     instruction += "\n\nIMPORTANTE: La respuesta debe estar completamente en español."
@@ -46,6 +87,62 @@ def generate_buyer_persona(product, skills, target_audience, temperature):
     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."
 
+# Modificar la sección de verificación en la columna de resultados
+with col2:
+    # Verificar si se ha enviado el formulario
+    if 'submitted' in st.session_state and st.session_state.submitted:
+        # Verificar si tenemos al menos producto o habilidades
+        if st.session_state.producto or st.session_state.habilidades:
+            with st.spinner("Creando tu Cliente Ideal Soñado..."):
+                # Generar el perfil del cliente
+                perfil_cliente = generate_buyer_persona(
+                    st.session_state.producto,
+                    st.session_state.habilidades,
+                    st.session_state.publico_objetivo,
+                    st.session_state.creatividad
+                )
+                # Guardar en session_state
+                st.session_state.perfil_cliente = perfil_cliente
+                # Resetear el estado de envío
+                st.session_state.submitted = False
+                
+            # Mostrar resultados
+            if not isinstance(st.session_state.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>
+                        {st.session_state.perfil_cliente}
+                    </div>
+                """, unsafe_allow_html=True)
+                
+                # Opción para descargar
+                st.download_button(
+                    label="Descargar Perfil",
+                    data=st.session_state.perfil_cliente,
+                    file_name="cliente_ideal.txt",
+                    mime="text/markdown"
+                )
+        else:
+            st.warning("Por favor, completa al menos uno de los campos: producto/servicio o habilidades.")
+    # Mostrar resultados anteriores si existen
+    elif st.session_state.perfil_cliente:
+        st.markdown(f"""
+            <div style="{styles['results_container']}">
+                <h3>Tu Cliente Ideal</h3>
+                {st.session_state.perfil_cliente}
+            </div>
+        """, unsafe_allow_html=True)
+        
+        # Opción para descargar
+        st.download_button(
+            label="Descargar Perfil",
+            data=st.session_state.perfil_cliente,
+            file_name="cliente_ideal.md",
+            mime="text/markdown"
+        )
+
 # Modificar la función update_profile para que no use spinner
 def update_profile():
     # Solo actualizar la variable de sesión