Update app.py

app.py

@@ -1,8 +1,7 @@
 from flask import Flask, render_template, request, jsonify, redirect, url_for, session
-import random
-from salesforce import get_salesforce_connection
 import os
 from flask_session import Session
+from salesforce import get_salesforce_connection
 
 # Initialize Flask app and Salesforce connection
 print("Starting app...")
@@ -17,7 +16,7 @@ print("Salesforce connection established.")
 app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q")  # Replace with a secure key
 app.config["SESSION_TYPE"] = "filesystem"
 
-# Check session initialization
+# Initialize session
 Session(app)
 print("Session configured.")
 
@@ -49,7 +48,7 @@ def login():
     if request.method == "POST":
         email = request.form.get("email")
         password = request.form.get("password")
-        print(f"Login attempt with email: {email}")  # Correctly indented print statement
+        print(f"Login attempt with email: {email}")  # Debug log
 
         try:
             query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
@@ -58,22 +57,26 @@ def login():
             if result["records"]:
                 session['user_id'] = result["records"][0]['Id']
                 session['user_email'] = email
-                print(f"Session variables: {session}")  # Moved print statement before return
+                print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
                 return redirect(url_for("menu"))
             else:
+                print("Invalid credentials!")
                 return render_template("login.html", error="Invalid credentials!")
         except Exception as e:
+            print(f"Error during login: {str(e)}")
             return render_template("login.html", error=f"Error: {str(e)}")
 
     return render_template("login.html")
 
-
 @app.route("/menu", methods=["GET", "POST"])
 def menu():
     selected_category = request.args.get("category", "All")
     user_id = session.get('user_id')
+    print(f"Session check in /menu: user_id={user_id}")
+
     if not user_id:
-         return redirect(url_for('login'))  # Redirect to login if user_id is not in session
+        print("Session missing, redirecting to login.")
+        return redirect(url_for('login'))
     
     try:
         query = "SELECT Name, Price__c, Image1__c, Category__c, Description__c FROM Menu_Item__c"
@@ -83,6 +86,7 @@ def menu():
         if selected_category != "All":
             food_items = [item for item in food_items if item.get("Category__c") == selected_category]
     except Exception as e:
+        print(f"Error fetching menu data: {str(e)}")
         food_items = []
         categories = []
     return render_template("menu.html", food_items=food_items, categories=categories, selected_category=selected_category)