Update app.py (#4)

- Update app.py (1122bd8e3c78b7e01eec3ee7b5476903e1e3e945)


Co-authored-by: Yaswanth Telagamsetti <Yaswanth56@users.noreply.huggingface.co>

app.py

@@ -57,22 +57,13 @@ def generate_referral_code(length=8):
     referral_code = ''.join(random.choice(characters) for _ in range(length))
     return referral_code
 
-@app.route("/")
-def home():
-    # Fetch user details from URL parameters
-    user_email = request.args.get("email")
-    user_name = request.args.get("name")
-
-    if user_email and user_name:
-        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
+@app.route("/menu", methods=["GET", "POST"])
+def menu():
+    user_email = session.get('user_email')
+    if not user_email:
+        return redirect(url_for("login"))
+    # Rest of your code for fetching menu and handling orders...
 
-    return render_template("login.html") 
 
 
 from datetime import datetime
@@ -131,6 +122,7 @@ def logout():
     session.pop('email', None)
     session.pop('rewardPoints', None)
     session.pop('coupon', None)
+    session.clear()
 
     # Create the redirect response
     response = redirect("https://biryanihub-dev-ed.develop.my.salesforce-sites.com/PublicLogin")