Update app.py

app.py

@@ -314,24 +314,25 @@ def login():
 @app.route("/menu", methods=["GET", "POST"])
 def menu():
     selected_category = request.args.get("category", "All")
-    user_id = session.get('user_name')
+    user_id = session.get('user_id')
     user_email = session.get('user_email')
 
-    # Check if user_email exists in session, if not get it from request arguments
     if not user_email:
         user_email = request.args.get("email")
         user_name = request.args.get("name")
 
-        # If email is found in the request, save it to the session
         if user_email:
             session['user_email'] = user_email
-            session['user_name'] = user_name  # Store the name in session
+            session['user_name'] = user_name  # Store name in session
         else:
             return redirect(url_for("login"))
     else:
-        # If the email exists in session, get the name from the session
-        user_name = session.get('user_name')
+        user_name = session.get('user_name')  # Get name from session if it's already stored
+
+    # Get the first letter of the user's name (make it uppercase for consistency)
+    first_letter = user_name[0].upper() if user_name else "A"
 
+    # Continue with the rest of the logic...
     try:
         # Fetch user referral and reward points
         user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
@@ -381,7 +382,7 @@ def menu():
         referral_code = 'N/A'
         reward_points = 0
 
-    # Pass the user's name (user_name) to the template
+    # Pass the user's first letter (first_letter) to the template
     return render_template(
         "menu.html",
         ordered_menu=ordered_menu,
@@ -389,7 +390,8 @@ def menu():
         selected_category=selected_category,
         referral_code=referral_code,
         reward_points=reward_points,
-        user_name=user_name  # Pass name to the template
+        user_name=user_name,  # Pass name to the template
+        first_letter=first_letter  # Pass first letter to the template
     )