Update menu.py

menu.py

@@ -8,7 +8,9 @@ sf = get_salesforce_connection()
 
 @menu_blueprint.route("/menu", methods=["GET", "POST"])
 def menu():
-    selected_category = request.args.get("category", "All")
+    # Get the current toggle status from the query string (False if not set)
+    veg_selected = request.args.get("veg", "off") == "on"
+    customized_dish_selected = request.args.get("customized_dish", "off") == "on"
     user_email = session.get('user_email')
 
     if not user_email:
@@ -62,6 +64,16 @@ def menu():
         # Merge both Menu_Item__c and Custom_Dish__c records into the ordered menu
         all_items = food_items + custom_dishes
 
+        # Apply filtering based on toggles
+        if veg_selected:
+            all_items = [item for item in all_items if item.get("Veg_NonVeg__c") in ["Veg", "both"]]
+        if customized_dish_selected:
+            all_items = [item for item in all_items if item.get("Section__c") == "Customized dish"]
+
+        # Default to all items if both toggles are off
+        if not veg_selected and not customized_dish_selected:
+            all_items = food_items + custom_dishes  # Show everything
+
         # Define the order of sections, adding "Best Sellers" at the top
         section_order = ["Best Sellers", "Starters", "Biryanis", "Curries", "Breads", "Customized dish", "Apetizer", "Desserts", "Soft Drinks"]
         ordered_menu = {section: [] for section in section_order}
@@ -69,11 +81,6 @@ def menu():
         # Sort items by Total_Ordered__c in descending order and pick top 4 as best sellers
         best_sellers = sorted(all_items, key=lambda x: x.get("Total_Ordered__c", 0), reverse=True)
 
-        if selected_category == "Veg":
-            best_sellers = [item for item in best_sellers if item.get("Veg_NonVeg__c") in ["Veg", "both"]]
-        elif selected_category == "Non veg":
-            best_sellers = [item for item in best_sellers if item.get("Veg_NonVeg__c") in ["Non veg", "both"]]
-
         # Take only the top 4 best sellers after filtering
         best_sellers = best_sellers[:4]
 
@@ -84,7 +91,7 @@ def menu():
         # Create a set to track item names already added to prevent duplicates
         added_item_names = set()
 
-        # Filter and organize menu items based on category and section (to avoid duplicates)
+        # Filter and organize menu items based on section (to avoid duplicates)
         for item in all_items:
             section = item.get("Section__c", "Others")  # Default to "Others" if missing
             if section not in ordered_menu:
@@ -94,20 +101,13 @@ def menu():
             if item['Name'] in added_item_names:
                 continue
 
-            # Apply category filters
-            if selected_category == "Veg" and item.get("Veg_NonVeg__c") not in ["Veg", "both"]:
-                continue
-            if selected_category == "Non veg" and item.get("Veg_NonVeg__c") not in ["Non veg", "both"]:
-                continue
-
             ordered_menu[section].append(item)
             added_item_names.add(item['Name'])  # Add item to the set of added items
-            print(f"Added item to {section}: {item['Name']}")  # Debugging
 
         # Remove empty sections
         ordered_menu = {section: items for section, items in ordered_menu.items() if items}
-        print(f"Final ordered menu: {ordered_menu.keys()}")  # Debugging
 
+        # Set categories for filtering
         categories = ["All", "Veg", "Non veg"]
 
     except Exception as e:
@@ -117,17 +117,20 @@ def menu():
         referral_code = 'N/A'
         reward_points = 0
 
-    # Pass the user's first letter (first_letter) to the template
+    # Pass the necessary variables to the template
     return render_template(
         "menu.html",
         ordered_menu=ordered_menu,
         categories=categories,
-        selected_category=selected_category,
+        veg_selected=veg_selected,
+        customized_dish_selected=customized_dish_selected,
+        selected_category="All",  # Default category is All
         referral_code=referral_code,
         reward_points=reward_points,
         user_name=user_name,  # Pass name to the template
         first_letter=first_letter  # Pass first letter to the template
     )
+
 @menu_blueprint.route('/api/addons', methods=['GET'])
 def get_addons():
     item_name = request.args.get('item_name')