Update app.py

app.py

@@ -207,16 +207,61 @@ elif menu == "Current":
 
     # Define item prices
     item_prices = {
-        "Café con leche": 1.20, "Colacao": 1.00, "Descafeinado con leche": 1.20, "Cortado": 1.20, 
-        "Aguasusia": 1.20, "Aguasusia susia": 1.20, "Café descafeinado con leche desnatada": 1.20, 
-        "Italiano": 1.20, "Café con soja": 1.20, "Té": 1.00, "Manzanilla": 1.00, "Nada": 0.00,
-        "Barrita con aceite": 1.00, "Barrita con tomate": 1.00, "Palmera de chocolate": 1.00, 
-        "Palmera de chocolate blanco": 1.00, "Yogurt": 1.00, "Pincho de tortilla": 1.00
+        "Cafés": {
+            "Café con leche": 1.20,
+            "Descafeinado con leche": 1.20,
+            "Cortado": 1.20, 
+            "Aguasusia": 1.20,
+            "Aguasusia susia": 1.20,
+            "Descafeinado con leche desnatada": 1.20, 
+            "Italiano": 1.20,
+            "Café con soja": 1.20,
+            "Café sin lactosa": 1.20,
+        },
+        "Infusiones": {
+            "Té": 0.90,
+            "Manzanilla": 0.90,    
+        },
+        "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    
+        }
+        "Nada": 0.00,
     }
+    
+    # # Define item prices
+    # item_prices = {
+    #     "Café con leche": 1.20,
+    #     "Colacao": 1.50,
+    #     "Café Descafeinado con leche": 1.20,
+    #     "Cortado": 1.20, 
+    #     "Café Aguasusia": 1.20,
+    #     "Café Aguasusia susia": 1.20,
+    #     "Café descafeinado con leche desnatada": 1.20, 
+    #     "Café Italiano": 1.20,
+    #     "Café con soja": 1.20,
+    #     "Té": 0.90,
+    #     "Manzanilla": 0.90,
+    #     "Nada": 0.00,
+    #     "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
+    # }
     # Define combined prices for special combinations
     combo_prices = {
         "desayuno + café (aceite)": 1.85,
-        "desayuno + café (tomate)": 2.50
+        "desayuno + café (tomate)": 2.50,
+        "desayuno + café (napolitana)": 1.85,
     }
 
     # Use session state to persist ticket generation status
@@ -228,40 +273,54 @@ elif menu == "Current":
         ticket = []
 
         # Iterate over each user's selections
+        drinks = []
+        foods = []
         for _, row in current_df.iterrows():
-            drinks = row['Drinks'].split(", ") if isinstance(row['Drinks'], str) else []
-            food = row['Food'].split(", ") if isinstance(row['Food'], str) else []
-
-            used_drinks = set()
-            used_food = set()
-
-            # Handle combinations of café + barrita con aceite
-            for drink in drinks:
-                if "café" in drink.lower() and "Barrita con aceite" in food:
-                    ticket.append({"Item": "desayuno + café (aceite)", "Price": combo_prices["desayuno + café (aceite)"]})
-                    used_drinks.add(drink)
-                    used_food.add("Barrita con aceite")
-                    break
-
-            # Handle combinations of café + barrita con tomate
-            for drink in drinks:
-                if "café" in drink.lower() and "Barrita con tomate" in food and drink not in used_drinks:
-                    ticket.append({"Item": "desayuno + café (tomate)", "Price": combo_prices["desayuno + café (tomate)"]})
-                    used_drinks.add(drink)
-                    used_food.add("Barrita con tomate")
-                    break
-
-            # Add remaining individual drinks not used in combinations
-            for drink in drinks:
-                if drink not in used_drinks and drink in item_prices:
-                    ticket.append({"Item": drink, "Price": item_prices[drink]})
-                    used_drinks.add(drink)
-
-            # Add remaining individual food not used in combinations
-            for f in food:
-                if f not in used_food and f in item_prices:
-                    ticket.append({"Item": f, "Price": item_prices[f]})
-                    used_food.add(f)
+            drinks.append(row['Drinks'])
+            foods.append(row['Food'])
+
+        # Convert every type of "café" to just "café"
+        drinks = ["Café" for x in drinks if x in item_prices["Cafés"].keys() else x]
+
+        # Display selections
+        st.write(f"Drinks: {drinks}")
+        st.write(f"Foods: {foods}")
+
+
+        
+            # drinks = row['Drinks'].split(", ") if isinstance(row['Drinks'], str) else []
+            # food = row['Food'].split(", ") if isinstance(row['Food'], str) else []
+
+            # used_drinks = set()
+            # used_food = set()
+
+            # # Handle combinations of café + barrita con aceite
+            # for drink in drinks:
+            #     if "café" in drink.lower() and "Barrita con aceite" in food:
+            #         ticket.append({"Item": "desayuno + café (aceite)", "Price": combo_prices["desayuno + café (aceite)"]})
+            #         used_drinks.add(drink)
+            #         used_food.add("Barrita con aceite")
+            #         break
+
+            # # Handle combinations of café + barrita con tomate
+            # for drink in drinks:
+            #     if "café" in drink.lower() and "Barrita con tomate" in food and drink not in used_drinks:
+            #         ticket.append({"Item": "desayuno + café (tomate)", "Price": combo_prices["desayuno + café (tomate)"]})
+            #         used_drinks.add(drink)
+            #         used_food.add("Barrita con tomate")
+            #         break
+
+            # # Add remaining individual drinks not used in combinations
+            # for drink in drinks:
+            #     if drink not in used_drinks and drink in item_prices:
+            #         ticket.append({"Item": drink, "Price": item_prices[drink]})
+            #         used_drinks.add(drink)
+
+            # # Add remaining individual food not used in combinations
+            # for f in food:
+            #     if f not in used_food and f in item_prices:
+            #         ticket.append({"Item": f, "Price": item_prices[f]})
+            #         used_food.add(f)
 
         # Create a DataFrame to display the ticket
         ticket_df = pd.DataFrame(ticket)