Update menu.py

menu.py

@@ -36,9 +36,21 @@ def get_valid_video_path(item_name, video_url=None):
 
 @menu_blueprint.route("/menu", methods=["GET", "POST"])
 def menu():
-    # Correct filter selection logic
-    selected_category = request.args.get("category", "All")
+    # Get raw query params
+    category_param = request.args.get("category", "All")
     is_veg = request.args.get("veg") == "on"
+
+    # Priority logic for setting selected_category
+    if category_param == "Customized Dish":
+        selected_category = "Customized Dish"
+    elif is_veg:
+        selected_category = "Veg"
+    elif category_param == "Non veg":
+        selected_category = "Non veg"
+    else:
+        selected_category = "All"
+
+    # Boolean flags to pass to template
     is_custom = selected_category == "Customized Dish"
     is_non_veg = selected_category == "Non veg"
 
@@ -70,7 +82,7 @@ def menu():
     cart_count_result = sf.query(cart_query)
     cart_item_count = cart_count_result.get('totalSize', 0)
 
-    # Menu items from Menu_Item__c
+    # Query menu items
     menu_query = """
         SELECT Name, Price__c, Description__c, Image1__c, Image2__c, 
                Veg_NonVeg__c, Section__c, Total_Ordered__c, Video1__c,
@@ -122,7 +134,7 @@ def menu():
         best_sellers = [item for item in best_sellers if not item['is_menu_item']]
     ordered_menu["Best Sellers"] = best_sellers[:4]
 
-    # Organize items
+    # Organize into sections
     added_item_names = set()
     for item in all_items:
         section = item['Section__c']