Update app.py

app.py

@@ -158,6 +158,11 @@ def generate_custom_dish():
 
     except Exception as e:
         return jsonify({"success": False, "error": str(e)}), 500
+@app.route("/set_allergies", methods=["POST"])
+def set_allergies():
+    allergies = request.form.getlist('allergies')
+    session['allergies'] = allergies  # Store allergies in session
+    return redirect(url_for('menu'))
 
                 
 
@@ -488,10 +493,9 @@ def menu():
 
         referral_code = user_result['records'][0].get('Referral__c', 'N/A')
         reward_points = user_result['records'][0].get('Reward_Points__c', 0)
-
-        # Query to fetch Menu_Item__c records including Total_Ordered__c for best sellers
+        # Fetch menu items
         menu_query = """
-            SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c, Total_Ordered__c 
+            SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c, Total_Ordered__c, Allergens__c
             FROM Menu_Item__c
         """
         result = sf.query(menu_query)
@@ -502,6 +506,17 @@ def menu():
             if 'Total_Ordered__c' not in item or item['Total_Ordered__c'] is None:
                 item['Total_Ordered__c'] = 0  # Default value
 
+        # Filter out items that contain allergens the user has
+        filtered_items = []
+        for item in food_items:
+            allergens = item.get('Allergens__c', '').split(',')  # Assuming allergens are comma-separated
+            if any(allergen in allergens for allergen in allergies):
+                continue  # Skip item if it contains any of the user's allergens
+            filtered_items.append(item)
+
+
+       
+
         # Query to fetch Custom_Dish__c records created within the last 7 days with Total_Ordered__c > 10
         custom_dish_query = """
             SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c, Total_Ordered__c 
@@ -512,7 +527,7 @@ def menu():
         custom_dishes = custom_dish_result['records'] if 'records' in custom_dish_result else []
 
         # Merge both Menu_Item__c and Custom_Dish__c records into the ordered menu
-        all_items = food_items + custom_dishes
+        all_items = filtered_items + custom_dishes
 
         # 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"]