Update app.py

app.py

@@ -72,24 +72,22 @@ def login():
 def menu():
     selected_category = request.args.get("category", "All")
     user_id = session.get('user_id')
-
     if not user_id:
-        print(f"Session data: {session}")  # Moved print statement before return
-        return redirect(url_for('login'))
-
+           print("User not logged in. Redirecting to login.")
+         return redirect(url_for('login'))  # Redirect to login if user_id is not in session
+        print(f"Session data: {session}")  # This will not be executed after return
     try:
         query = "SELECT Name, Price__c, Image1__c, Category__c, Description__c FROM Menu_Item__c"
         result = sf.query(query)
         food_items = result['records'] if 'records' in result else []
         categories = {item['Category__c'] for item in food_items if 'Category__c' in item}
-
+        selected_category = request.args.get("category", "All")
         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 data: {e}")
         food_items = []
         categories = []
-        print(f"Error fetching data: {e}")
-
     return render_template("menu.html", food_items=food_items, categories=categories, selected_category=selected_category)