Update app.py

app.py

@@ -407,17 +407,24 @@ def menu():
                 item['Total_Ordered__c'] = 0  # Default value
 
         # Sort items by Total_Ordered__c in descending order and pick top 4 as best sellers
-        best_sellers = sorted(food_items, key=lambda x: x.get("Total_Ordered__c", 0), reverse=True)[:4]
+        best_sellers = sorted(food_items, key=lambda x: x.get("Total_Ordered__c", 0), reverse=True)
         print(f"Best sellers: {[item['Name'] for item in best_sellers]}")  # Debugging
+        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"]]
 
-        # Define the order of sections, adding "Best Sellers" at the top
-        section_order = ["Best Sellers", "Starters","Biryanis","Curries","Breads","Apetizer", "Desserts", "Soft Drinks"]
-        ordered_menu = {section: [] for section in section_order}
+        # Take only the top 4 best sellers after filtering
+        best_sellers = best_sellers[:4]
 
-        # Add best sellers to ordered_menu if there are any
+        # Ensure "Best Sellers" is added only if there are items after filtering
         if best_sellers:
             ordered_menu["Best Sellers"] = best_sellers
 
+        # Define the order of sections, adding "Best Sellers" at the top
+        section_order = ["Best Sellers", "Starters","Biryanis","Curries","Breads","Apetizer", "Desserts", "Soft Drinks"]
+        ordered_menu = {section: [] for section in section_order}
+
         # Filter and organize menu items based on category and section
         for item in food_items:
             section = item.get("Section__c", "Others")  # Default to "Others" if missing