Update app.py

app.py

@@ -259,9 +259,21 @@ def menu():
     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}")
     
-    if not user_id:
-        print("Session missing, redirecting to login.")
-        return redirect(url_for('login'))
+    if not user_email:
+        user_email = request.args.get("email")
+        user_name = request.args.get("name")
+
+        if user_email:
+            session['user_email'] = user_email
+            session['user_name'] = user_name  # If needed
+
+            print(f"✅ User session set: {user_email}, {user_name}")
+        else:
+            print("❌ No email in URL, redirecting to login.")
+            return redirect(url_for("login"))
+
+    print(f"Session check in /menu: user_email={user_email}")
+
     
     try:
         # Fetch the user's Referral__c and Reward_Points__c
@@ -696,7 +708,7 @@ def update_quantity():
 @app.route("/checkout", methods=["POST"])
 def checkout():
     email = session.get('user_email')
-    user_id = session.get('user_id')
+    user_id = session.get('user_name')
 
     if not email or not user_id:
         return jsonify({"success": False, "message": "User not logged in"})