Update app.py

app.py

@@ -225,14 +225,16 @@ elif menu == "Current":
         "Colacaos": {
             "Colacao": 1.50,
         },
-        "Comidas": {
-            "Barrita con aceite": 0.65,
-            "Barrita con tomate": 1.30,
-            "Palmera de chocolate": 1.50, 
-            "Palmera de chocolate blanco": 1.50,
-            "Yogurt": 1.00,
-            "Pincho de tortilla": 1.50    
+        "Palmeras": {
+            "Palmera chocolate": 1.50, 
+            "Palmera chocolate blanco": 1.50,
         },
+        "Barrita aceite": 0.65,
+        "Barrita tomate": 1.30,
+        "Napolitana": 0.65,
+        "Croissant": 0.65,
+        "Yogurt": 1.00,
+        "Tortilla": 1.50,
         "Nada": 0.00,
     }
     
@@ -279,13 +281,47 @@ elif menu == "Current":
             drinks.append(row['Drinks'])
             foods.append(row['Food'])
 
-        # Convert every type of "café" to just "café"
+        # Simplify items
         drinks = ["Café" if x in item_prices["Cafés"].keys() else x for x in drinks]
+        drinks = ["Infusión" if x in item_prices["Infusiones"].keys() else x for x in drinks]
+        foods = ["Palmera" if x in item_prices["Palmeras"].keys() else x for x in foods]
 
         # Display selections
         st.write(f"Drinks: {drinks}")
         st.write(f"Foods: {foods}")
 
+        # Count items
+        item_count = {
+            "Café": len([x for x in drinks if x == "Café"]),
+            "Infusión": len([x for x in drinks if x == "Infusión"]),
+            "Colacao": len([x for x in drinks if x == "Colacao"]),
+            "Barrita aceite": len([x for x in drinks if x == "Barrita aceite"]),
+            "Barrita tomate": len([x for x in drinks if x == "Barrita tomate"]),
+            "Napolitana": len([x for x in drinks if x == "Napolitana"]),
+            "Palmera": len([x for x in drinks if x == "Palmeras"]),
+            "Tortilla": len([x for x in drinks if x == "Tortilla"]),
+            "Croissant": len([x for x in drinks if x == "Croissant"]),
+            "Yogurt": len([x for x in drinks if x == "Yogurt"]),
+        }
+
+        # Make associations
+        item_association = {}
+        food_count = item_count["Barrita aceite"] + item_count["Barrita tomate"] + item_count["Napolitana"] + item_count["Croissant"]
+        if item_count["Café"] >= food_count:
+            item_association["Desayuno + Café (tomate)"] = item_count["Barrita tomate"]
+            item_association["Desayuno + Café (aceite)"] = item_count["Barrita aceite"]
+            item_association["Desayuno + Café (napolitana)"] = item_count["Napolitana"]
+            item_association["Desayuno + Café (croissant)"] = item_count["Croissant"]
+            item_association["Café"] = item_count["Café"] - food_count
+        else:
+            raise NotImplementedError
+        item_association["Colacaos"] = item_count["Colacao"]
+        item_association["Yogurts"] = item_count["Yogurt"]
+        item_association["Tortillas"] = item_count["Tortilla"]
+        item_association["Palmeras"] = item_count["Palmera"]
+        item_association["Infusiones"] = item_count["Infusión"]
+
+        st.write(item_association)
 
         
             # drinks = row['Drinks'].split(", ") if isinstance(row['Drinks'], str) else []