Update app.py

app.py

@@ -878,7 +878,7 @@ elif page == "💡 Recomendación de Artículos":
     )
     st.markdown("""<p style='text-align: center; color: #5D6D7E;'>Obtén recomendaciones personalizadas para tus clientes basadas en su cesta de compra.</p>""", unsafe_allow_html=True)
 
-    st.write("### Selecciona los artículos y asigna las cantidades para la cesta:")
+    st.write("### Selecciona los artículos para la cesta:")
 
     # Añadir separador para mejorar la segmentación visual
     st.divider()
@@ -892,14 +892,13 @@ elif page == "💡 Recomendación de Artículos":
     # Permitir seleccionar las descripciones, pero trabajar con los códigos
     selected_descriptions = st.multiselect("Selecciona los artículos", available_articles['DESCRIPCION'].unique())
     
-    quantities = {}
 
     if selected_descriptions:
-        st.write("### Selecciona los artículos, las unidades, y visualiza la imagen:")
+        st.write("### Visualiza la selección de artículos:")
 
         for description in selected_descriptions:
             code = article_dict[description]  # Usar el código del artículo
-            col1, col2, col3 = st.columns([1, 2, 2])  # Ajustar proporciones para que las imágenes y textos se alineen
+            col1, col2 = st.columns([1, 2])  # Ajustar proporciones para que las imágenes y textos se alineen
 
             with col1:
                 # Mostrar la imagen del artículo
@@ -908,20 +907,12 @@ elif page == "💡 Recomendación de Artículos":
 
             with col2:
                 # Mostrar la descripción del artículo
-                st.write(f"**{description}**")
-            
-            with col3:
-                # Caja de número para la cantidad, asociada al código
-                quantities[code] = st.number_input(f"Cantidad {code}", min_value=0, step=1, key=code)
+                st.write(f"**{description}**(Código: {code})")
 
     # Añadir un botón estilizado "Calcular" con icono
     if st.button("🛒 Obtener Recomendaciones"):
         # Crear una lista de artículos basada en los códigos y cantidades
-        new_basket = []
-        for code in quantities:
-            quantity = quantities[code]
-            if quantity > 0:
-                new_basket.extend([code] * quantity)  # Añadir el código tantas veces como 'quantity'
+        new_basket = [article_dict[description] for description in selected_descriptions]  # Usar los códigos en lugar de las descripciones
         
         if new_basket:
             # Procesar la lista para recomendar utilizando tu función 'recomienda_tf'