Update app.py

app.py

@@ -11,6 +11,16 @@ load_dotenv()
 # Configurar la API de Google
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
+# Inicializar variables de estado en session_state si no existen
+if 'perfil_cliente' not in st.session_state:
+    st.session_state.perfil_cliente = None
+if 'producto' not in st.session_state:
+    st.session_state.producto = ""
+if 'habilidades' not in st.session_state:
+    st.session_state.habilidades = ""
+if 'creatividad' not in st.session_state:
+    st.session_state.creatividad = 1.0
+
 # Función para generar el perfil de cliente ideal
 @st.cache_resource
 def get_model(temperature):
@@ -32,6 +42,14 @@ def generate_buyer_persona(product, skills, 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."
 
+# Función para actualizar el perfil del cliente
+def update_profile():
+    st.session_state.perfil_cliente = generate_buyer_persona(
+        st.session_state.producto,
+        st.session_state.habilidades,
+        st.session_state.creatividad
+    )
+
 # Configurar la interfaz de usuario con Streamlit
 st.set_page_config(page_title="Generador de Cliente Ideal", page_icon="👤", layout="wide")
 
@@ -61,42 +79,51 @@ col1, col2 = st.columns([1, 2])
 
 # Columna de entrada
 with col1:
-    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")
+    producto = st.text_input("¿Qué producto o servicio ofreces?", 
+                            value=st.session_state.producto,
+                            placeholder="Ejemplo: Curso de Inglés",
+                            key="producto_input")
+    st.session_state.producto = producto
+    
+    habilidades = st.text_input("¿Cuáles son tus habilidades principales?", 
+                               value=st.session_state.habilidades,
+                               placeholder="Ejemplo: Enseñanza, comunicación, diseño de contenidos",
+                               key="habilidades_input")
+    st.session_state.habilidades = habilidades
     
     # 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)
+    creatividad = st.slider("Nivel de creatividad", 
+                           min_value=0.0, 
+                           max_value=2.0, 
+                           value=st.session_state.creatividad, 
+                           step=0.1,
+                           key="creatividad_slider")
+    st.session_state.creatividad = creatividad
     
     # Botón para generar
-    submit = st.button("GENERAR PERFIL DE CLIENTE IDEAL")
+    submit = st.button("GENERAR PERFIL DE CLIENTE IDEAL", on_click=update_profile)
 
 # 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"
-                    )
+    # Mostrar resultados si existen en el estado de la sesión
+    if st.session_state.perfil_cliente:
+        if not isinstance(st.session_state.perfil_cliente, str):
+            st.error("Error al generar el perfil de cliente ideal")
         else:
-            st.warning("Por favor, completa los campos de producto y habilidades antes de generar el perfil.")
+            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"
+            )
+    # Mostrar el mensaje de advertencia solo si se ha presionado el botón pero faltan campos
+    elif submit and (not producto or not habilidades):
+        st.warning("Por favor, completa los campos de producto y habilidades antes de generar el perfil.")