Update app.py

app.py

@@ -1,4 +1,5 @@
 from flask import Flask, render_template, request, jsonify, redirect, url_for, session
+from flask import Flask, request, session, redirect, url_for, render_template
 from flask_session import Session  # Import the Session class
 from flask.sessions import SecureCookieSessionInterface  # Import the class
 from salesforce import get_salesforce_connection
@@ -40,7 +41,7 @@ def generate_referral_code(length=8):
 
 @app.route("/")
 def home():
-    # Check if Hugging Face provided user details
+    # Fetch user details from URL parameters
     user_email = request.args.get("email")
     user_name = request.args.get("name")
 
@@ -48,8 +49,12 @@ def home():
         session["user_email"] = user_email
         session["user_name"] = user_name
         print(f"User logged in via Hugging Face: {user_email} - {user_name}")
+
+        # Ensure session is saved before redirecting
+        session.modified = True
         return redirect(url_for("menu"))  # Redirect to menu directly
 
+    return render_template("login.html") 
 
 
 
@@ -261,65 +266,18 @@ def login():
             return render_template("login.html", error=f"Error: {str(e)}")
 
     return render_template("login.html")
-
-
-@app.route("/menu", methods=["GET", "POST"])
+@app.route("/menu")
 def menu():
-    selected_category = request.args.get("category", "All")
-    user_id = session.get('user_id')
-    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'))
-    
-    try:
-        # Fetch the user's Referral__c and Reward_Points__c
-        user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
-        user_result = sf.query(user_query)
-
-        if not user_result['records']:
-            print("User not found!")
-            return redirect(url_for('login'))
+    user_email = session.get("user_email")
+    user_name = session.get("user_name")
 
-        referral_code = user_result['records'][0].get('Referral__c', 'N/A')  # Default to 'N/A' if empty
-        reward_points = user_result['records'][0].get('Reward_Points__c', 0)  # Default to 0 if empty
+    if not user_email or not user_name:
+        return redirect(url_for("home"))  # Redirect to home if session is missing
 
-        # Query to fetch menu items
-        menu_query = """
-            SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c 
-            FROM Menu_Item__c
-        """
-        result = sf.query(menu_query)
-        food_items = result['records'] if 'records' in result else []
+    return render_template("menu.html", user_name=user_name)
 
-        # Filter categories dynamically
-        categories = {item.get("Veg_NonVeg__c").capitalize() for item in food_items if item.get("Veg_NonVeg__c")}
-        categories = {"Veg", "Non-Veg"}  # 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
-        referral_code = 'N/A'
-        reward_points = 0
-
-    # Render the menu page with the fetched data
-    return render_template(
-        "menu.html",
-        food_items=food_items,
-        categories=sorted(categories),  # Sort categories alphabetically if needed
-        selected_category=selected_category,
-        referral_code=referral_code,
-        reward_points=reward_points
-    )
 
 @app.route("/cart", methods=["GET"])
 def cart():