Update app.py

app.py

@@ -273,59 +273,56 @@ def login():
     return render_template("menu.html", user_name=user_name)
 @app.route("/menu", methods=["GET", "POST"])
 def menu():
-    # Capture user details from URL (Hugging Face parameters)
-    if "email" in request.args and "name" in request.args:
-        session["user_email"] = request.args.get("email")
-        session["user_name"] = request.args.get("name")
-        session.modified = True
-        print(f"User auto-logged in: {session['user_email']} - {session['user_name']}")
+    selected_category = request.args.get("category", "All")
+    user_id = session.get('user_id')
+    user_email = session.get('user_email')  
 
-    user_email = session.get("user_email")
-    user_name = session.get("user_name")
+    print_session()  # Debugging session info
 
-    if not user_email or not user_name:
-        print("User not logged in. Redirecting to login page.")
-        return redirect(url_for("login"))  # Redirect if session data is missing
+    if not user_id or not user_email:
+        print("❌ Session missing, redirecting to login.")
+        return redirect(url_for('login'))
 
     try:
-        # Fetch referral code and reward points from Salesforce
-        user_query = f"""
-            SELECT Referral__c, Reward_Points__c 
-            FROM Customer_Login__c 
-            WHERE Email__c = '{user_email}'
-        """
+        # ✅ Fetch user referral & reward points
+        user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
         user_result = sf.query(user_query)
-        user_data = user_result.get("records", [{}])[0]
 
-        referral_code = user_data.get("Referral__c", "N/A")
-        reward_points = user_data.get("Reward_Points__c", 0)
+        if not user_result['records']:
+            print("❌ User not found in Salesforce, redirecting to login.")
+            return redirect(url_for('login'))
+
+        referral_code = user_result['records'][0].get('Referral__c', 'N/A')  # Default 'N/A'
+        reward_points = user_result['records'][0].get('Reward_Points__c', 0)  # Default 0
 
-        # Fetch menu items
+        # ✅ Fetch menu items
         menu_query = """
             SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c 
             FROM Menu_Item__c
         """
-        menu_result = sf.query(menu_query)
-        food_items = menu_result.get("records", [])
+        result = sf.query(menu_query)
+        food_items = result['records'] if 'records' in result else []
 
-        # Extract and filter categories
-        categories = {"All", "Veg", "Non-Veg"}  # Default categories
-        selected_category = request.args.get("category", "All")
+        # ✅ 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"}  # Ensure valid categories only
 
+        # ✅ Filter items based on selected category
         if selected_category == "Veg":
             food_items = [item for item in food_items if item.get("Veg_NonVeg__c") in ["Veg", "both"]]
         elif selected_category == "Non-Veg":
             food_items = [item for item in food_items if item.get("Veg_NonVeg__c") in ["Non veg", "both"]]
 
     except Exception as e:
-        print(f"Error fetching menu data: {str(e)}")
+        print(f"❌ Error fetching menu data: {str(e)}")
+        print(traceback.format_exc())  # Show full error traceback
         food_items = []
-        referral_code = "N/A"
+        categories = {"All", "Veg", "Non-Veg"}
+        referral_code = 'N/A'
         reward_points = 0
 
     return render_template(
         "menu.html",
-        user_name=user_name,
         food_items=food_items,
         categories=sorted(categories),
         selected_category=selected_category,