Update app.py

app.py

@@ -877,6 +877,11 @@ elif page == "💡 Recomendación de Artículos":
     # Inicializar session state para la cesta si no existe
     if 'new_basket' not in st.session_state:
         st.session_state['new_basket'] = []  # Inicializar como una lista vacía
+    if 'recommendations_df' not in st.session_state:
+        st.session_state['recommendations_df'] = None
+
+    if 'selected_descriptions' not in st.session_state:
+        st.session_state['selected_descriptions'] = []
 
     def add_to_basket(rec_code):
         if rec_code not in st.session_state['new_basket']:
@@ -901,8 +906,9 @@ elif page == "💡 Recomendación de Artículos":
     article_dict = dict(zip(available_articles['DESCRIPCION'], available_articles['ARTICULO']))
 
     # Permitir seleccionar las descripciones, pero trabajar con los códigos
-    selected_descriptions = st.multiselect("Selecciona los artículos", available_articles['DESCRIPCION'].unique())
-    
+    selected_descriptions = st.multiselect("Selecciona los artículos",available_articles['DESCRIPCION'].unique(),default=st.session_state['selected_descriptions'])
+
+    st.session_state['selected_descriptions'] = selected_descriptions
 
     if selected_descriptions:
         st.write("### Visualiza la selección de artículos:")
@@ -932,59 +938,60 @@ elif page == "💡 Recomendación de Artículos":
         st.write(f"Artículos seleccionados (códigos): {new_basket}")
         
         if new_basket:
-            # Procesar la lista para recomendar utilizando tu función 'recomienda_tf'
-            recommendations_df = recomienda_tf(new_basket, cestas,productos)
-
-            if not recommendations_df.empty:
-                st.success("### Según tu cesta, te recomendamos que consideres añadir estos artículos:")
-
-                # Mostrar los artículos recomendados con imágenes y relevancia
-                for idx, row in recommendations_df.iterrows():
-                    rec_code = row['ARTICULO']
-                    rec_desc = row['DESCRIPCION']
-                    rec_relevance = row['RELEVANCIA']  # Usar la relevancia calculada
-                    rec_img_url = f"https://www.saneamiento-martinez.com/imagenes/articulos/{rec_code}_1.JPG"
-                    
-                    # Verificar si la imagen existe antes de mostrar el artículo
-                    if image_exists(rec_img_url):
-                        rec_col1, rec_col2, rec_col3, rec_col4 = st.columns([1, 3, 1, 1])  # Añadir una columna para la relevancia
-                        with rec_col1:
-                            st.image(rec_img_url, width=100)
-                        with rec_col2:
-                            st.write(f"**{rec_desc}** (Código: {rec_code})")
-                        with rec_col3:
-                            st.metric(label="Relevancia",value =f"{rec_relevance * 100:.2f}") # Mostrar la relevancia con 4 decimales
-                        with rec_col4:
-                    # Botón para añadir artículo recomendado a la cesta
-                            if rec_code not in st.session_state['new_basket']:
-                                st.button("➕", key=f"add_{rec_code}", on_click=add_to_basket, args=(rec_code,))
-                            else:
-                                st.write("✓")
-
-                # Mostrar una sección de previsualización de la nueva cesta antes de agregarla al histórico
-                st.write("### Previsualiza la cesta completa antes de agregarla:")
-                for code in st.session_state['new_basket']:
-                    # Mostrar la descripción del artículo en la previsualización
-                    article_desc = productos[productos['ARTICULO'] == code]['DESCRIPCION'].values[0]
-                    st.write(f"- {article_desc} (Código: {code})")
-
-
-                # Botón para añadir la cesta nueva al histórico
-                if st.button("📦 Añadir cesta al histórico"):
-                    if st.session_state['new_basket']:
-                        # Usar la función retroalimentacion para añadir la cesta
-                        retroalimentacion(cestas, st.session_state['new_basket'])
-                        st.success("✓ La cesta ha sido añadida al histórico.")
-                        # Limpiar la cesta después de añadirla al histórico
-                        st.session_state['new_basket'] = []
-                        st.rerun()  # Limpiar las recomendaciones también
+
+            st.session_state['recommendations_df'] = recomienda_tf(new_basket, cestas, productos)
+            st.experimental_rerun()
+            
+
+    # Display recommendations from session state
+    if st.session_state['recommendations_df'] is not None and not st.session_state['recommendations_df'].empty:
+        st.success("### Según tu cesta, te recomendamos que consideres añadir estos artículos:")
+
+        for idx, row in st.session_state['recommendations_df'].iterrows():
+            rec_code = row['ARTICULO']
+            rec_desc = row['DESCRIPCION']
+            rec_relevance = row['RELEVANCIA']
+            rec_img_url = f"https://www.saneamiento-martinez.com/imagenes/articulos/{rec_code}_1.JPG"
+            
+            if image_exists(rec_img_url):
+                rec_col1, rec_col2, rec_col3, rec_col4 = st.columns([1, 3, 1, 1])
+                with rec_col1:
+                    st.image(rec_img_url, width=100)
+                with rec_col2:
+                    st.write(f"**{rec_desc}** (Código: {rec_code})")
+                with rec_col3:
+                    st.metric(label="Relevancia", value=f"{rec_relevance * 100:.2f}")
+                with rec_col4:
+                    # Disable button if item is already in basket
+                    button_disabled = rec_code in st.session_state['new_basket']
+                    if not button_disabled:
+                        st.button("➕", 
+                                key=f"add_{rec_code}", 
+                                on_click=add_to_basket, 
+                                args=(rec_code,),
+                                disabled=button_disabled)
                     else:
-                        st.warning("⚠️ No hay artículos en la cesta para añadir.")                               
+                        st.write("✓")
+
+        # Preview section
+        st.write("### Previsualiza la cesta completa antes de agregarla:")
+        for code in st.session_state['new_basket']:
+            article_desc = productos[productos['ARTICULO'] == code]['DESCRIPCION'].values[0]
+            st.write(f"- {article_desc} (Código: {code})")
+
+        if st.button("📦 Añadir cesta al histórico"):
+            if st.session_state['new_basket']:
+                retroalimentacion(cestas, st.session_state['new_basket'])
+                st.success("✓ La cesta ha sido añadida al histórico.")
+                # Reset all session states
+                st.session_state['new_basket'] = []
+                st.session_state['recommendations_df'] = None
+                st.session_state['selected_descriptions'] = []
+                st.experimental_rerun()
             else:
-                st.warning("⚠️ No se encontraron recomendaciones para la cesta proporcionada.")
-        else:
-            st.warning("⚠️ Por favor selecciona al menos un artículo y define su cantidad.")
-
+                st.warning("⚠️ No hay artículos en la cesta para añadir.")
+    elif st.session_state['recommendations_df'] is not None:
+        st.warning("⚠️ No se encontraron recomendaciones para la cesta proporcionada.")
 
 
 # elif page == "💡 Recomendación de Artículos":