Update app.py

app.py

@@ -44,17 +44,20 @@ def home():
     # Fetch user details from URL parameters
     user_email = request.args.get("email")
     user_name = request.args.get("name")
+    table_number = request.args.get("table")  # Capture table number
 
     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}")
+        session["table_number"] = table_number  # Store table number in session
+        print(f"User logged in: {user_email} - {user_name} - Table: {table_number}")
 
         # Ensure session is saved before redirecting
         session.modified = True
         return redirect(url_for("menu"))  # Redirect to menu directly
 
-    return render_template("redirect_page.html") 
+    return render_template("redirect_page.html")
+
 
 
 from datetime import datetime
@@ -990,6 +993,7 @@ def update_quantity():
 def checkout():
     email = session.get('user_email')
     user_id = session.get('user_name')
+    table_number = session.get('table_number')  # Retrieve table number
 
     if not email or not user_id:
         return jsonify({"success": False, "message": "User not logged in"})
@@ -1011,31 +1015,27 @@ def checkout():
 
         total_price = sum(item['Price__c'] for item in cart_items)
         discount = 0
+
         # Fetch the user's existing coupons
         coupon_query = sf.query(f"""
             SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
         """)
 
-        has_coupons = bool(coupon_query["records"])  # Check if user has any coupons
+        has_coupons = bool(coupon_query["records"])
 
         if selected_coupon:
-            # Case 3: User selected a valid coupon → Apply discount & remove coupon
             discount = total_price * 0.10  # 10% discount
             referral_coupon_id = coupon_query["records"][0]["Id"]
             existing_coupons = coupon_query["records"][0]["Coupon_Code__c"].split("\n")
 
             # Remove only the selected coupon
             updated_coupons = [coupon for coupon in existing_coupons if coupon.strip() != selected_coupon]
-
-            # Convert list back to a string with newlines
             updated_coupons_str = "\n".join(updated_coupons).strip()
 
-            # Update the Referral_Coupon__c record with the remaining coupons
             sf.Referral_Coupon__c.update(referral_coupon_id, {
                 "Coupon_Code__c": updated_coupons_str
             })
         else:
-            # Case 1 & Case 2: User has no coupons or has coupons but didn’t select one → Add 10% to reward points
             reward_points_to_add = total_price * 0.10
 
             # Fetch current reward points
@@ -1049,17 +1049,13 @@ def checkout():
                 current_reward_points = customer.get("Reward_Points__c") or 0
                 new_reward_points = current_reward_points + reward_points_to_add
 
-                print(f"Updating reward points: Current = {current_reward_points}, Adding = {reward_points_to_add}, New = {new_reward_points}")
-
-                # Update reward points in Salesforce
                 sf.Customer_Login__c.update(customer["Id"], {
                     "Reward_Points__c": new_reward_points
                 })
-                print(f"Successfully updated reward points for {email}")
 
         total_bill = total_price - discount
 
-        # ✅ Store **all details** including Add-Ons, Instructions, Price, and Image
+        # ✅ Store all order details
         order_details = "\n".join(
             f"{item['Name']} x{item['Quantity__c']} | Add-Ons: {item.get('Add_Ons__c', 'None')} | "
             f"Instructions: {item.get('Instructions__c', 'None')} | "
@@ -1067,19 +1063,18 @@ def checkout():
             for item in cart_items
         )
 
-        # Fetch Customer ID from Customer_Login__c based on email
+        # Fetch Customer ID from Customer_Login__c
         customer_query = sf.query(f"""
             SELECT Id FROM Customer_Login__c
             WHERE Email__c = '{email}'
         """)
         
-        # Assuming the customer exists
         customer_id = customer_query["records"][0]["Id"] if customer_query["records"] else None
         
         if not customer_id:
             return jsonify({"success": False, "message": "Customer record not found in Salesforce"})
 
-        # Store the order details in Order__c
+        # ✅ Store table number in Order__c
         order_data = {
             "Customer_Name__c": user_id,
             "Customer_Email__c": email,
@@ -1087,24 +1082,23 @@ def checkout():
             "Discount__c": discount,
             "Total_Bill__c": total_bill,
             "Order_Status__c": "Pending",
-            "Customer2__c": customer_id,  # Correcting to use the Salesforce Customer record ID
-            "Order_Details__c": order_details  # ✅ Now includes **all details**
+            "Customer2__c": customer_id,
+            "Order_Details__c": order_details,
+            "Table_Number__c": table_number  # ✅ Store table number
         }
 
         sf.Order__c.create(order_data)
-        
 
         # ✅ Delete cart items after order is placed
         for item in cart_items:
             sf.Cart_Item__c.delete(item["Id"])
-       
-
 
         return jsonify({"success": True, "message": "Order placed successfully!"})
 
     except Exception as e:
         print(f"Error during checkout: {str(e)}")
         return jsonify({"success": False, "error": str(e)})
+
 @app.route("/order", methods=["GET"])
 def order_summary():
     email = session.get('user_email')  # Fetch logged-in user's email