Update app.py

app.py

@@ -121,31 +121,36 @@ def login():
 def menu():
     selected_category = request.args.get("category", "All")
     user_id = session.get('user_id')
-    print(f"Cookies on /menu: {request.cookies}")
-    print(f"Session check in /menu: user_id={user_id}")
+    user_email = session.get('user_email')  # Fetch the user's email
+    print(f"Session check in /menu: user_id={user_id}, user_email={user_email}")
     
-    # Get the selected category from the query parameter, default is "All"
-    selected_category = request.args.get("category", "All")
-    print(f"Selected category: {selected_category}")
     if not user_id:
         print("Session missing, redirecting to login.")
         return redirect(url_for('login'))
     
-
     try:
-        query = """
+        # Fetch the user's Referral__c and Reward_Points__c
+        user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
+        user_result = sf.query(user_query)
+
+        if not user_result['records']:
+            print("User not found!")
+            return redirect(url_for('login'))
+
+        referral_code = user_result['records'][0].get('Referral__c', 'N/A')  # Default to 'N/A' if empty
+        reward_points = user_result['records'][0].get('Reward_Points__c', 0)  # Default to 0 if empty
+
+        # Query to fetch menu items
+        menu_query = """
             SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c 
             FROM Menu_Item__c
         """
-        # Query to fetch menu items
-        result = sf.query(query)
-
-        # Fetch all food items from the query result
+        result = sf.query(menu_query)
         food_items = result['records'] if 'records' in result else []
 
-        # Dynamically determine categories based on the fetched data
+        # Filter categories dynamically
         categories = {item.get("Veg_NonVeg__c").capitalize() for item in food_items if item.get("Veg_NonVeg__c")}
-        categories = {"Veg", "Non-Veg"}  # Explicitly overwrite to ensure valid categories only
+        categories = {"Veg", "Non-Veg"}  # Ensure valid categories only
 
         # Filter food items based on the selected category
         if selected_category == "Veg":
@@ -157,15 +162,19 @@ def menu():
         print(f"Error fetching menu data: {str(e)}")
         food_items = []
         categories = {"All", "Veg", "Non-Veg"}  # Default categories on error
+        referral_code = 'N/A'
+        reward_points = 0
 
-    # Render the menu page with the filtered data
+    # Render the menu page with the fetched data
     return render_template(
         "menu.html",
         food_items=food_items,
         categories=sorted(categories),  # Sort categories alphabetically if needed
         selected_category=selected_category,
-
+        referral_code=referral_code,
+        reward_points=reward_points
     )
+
 @app.route("/cart", methods=["GET"])
 def cart():
     email = session.get('user_email')  # Get logged-in user's email