Update app.py

app.py

@@ -78,7 +78,57 @@ def login():
             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"Cookies on /menu: {request.cookies}")
+    print(f"Session check in /menu: user_id={user_id}")
+    
+    # Get the selected category from the query parameter, default is "All"
+    selected_category = request.args.get("category", "All")
+    print(f"Selected category: {selected_category}")
+    if not user_id:
+        print("Session missing, redirecting to login.")
+        return redirect(url_for('login'))
+    
 
+    try:
+        query = """
+            SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c 
+            FROM Menu_Item__c
+        """
+        # Query to fetch menu items
+        result = sf.query(query)
+
+        # Fetch all food items from the query result
+        food_items = result['records'] if 'records' in result else []
+
+        # Dynamically determine categories based on the fetched data
+        categories = {item.get("Veg_NonVeg__c").capitalize() for item in food_items if item.get("Veg_NonVeg__c")}
+        categories = {"Veg", "Non-Veg"}  # Explicitly overwrite to ensure valid categories only
+
+        # Filter food items based on the 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)}")
+        food_items = []
+        categories = {"All", "Veg", "Non-Veg"}  # Default categories on error
+
+    # Render the menu page with the filtered data
+    return render_template(
+        "menu.html",
+        food_items=food_items,
+        categories=sorted(categories),  # Sort categories alphabetically if needed
+        selected_category=selected_category,
+
+    )
+    
+"""
 @app.route("/menu", methods=["GET", "POST"])
 def menu():
     selected_category = request.args.get("category", "All")
@@ -102,6 +152,7 @@ def menu():
         food_items = []
         categories = []
     return render_template("menu.html", food_items=food_items, categories=categories, selected_category=selected_category)
+"""
 
 
 @app.route("/cart", methods=["GET"])