Update app.py

app.py

@@ -65,6 +65,7 @@ def generate_coupon_code(length=10):
     return ''.join(random.choice(characters) for _ in range(length))
 
 
+
 @app.route("/generate_custom_dish", methods=["POST"])
 def generate_custom_dish():
     try:
@@ -99,7 +100,28 @@ def generate_custom_dish():
         result = sf.Custom_Dish__c.create(custom_dish)
 
         if result.get('success'):
-            return redirect(url_for("menu", category="Customized Dish"))  # Redirect to the menu with custom dish
+            # After creating the custom dish, add it to the Cart_Item__c
+            email = session.get('user_email')  # Assuming you have the user's email in session
+            
+            # Create the cart item with custom dish details and set other fields to default values
+            cart_item = {
+                'Name': dish_name,
+                'Price__c': price,
+                'Quantity__c': 1,  # Assuming a default quantity of 1 for the custom dish
+                'Add_Ons__c': '',  # Set Add_ons__c to empty
+                'Add_Ons_Price__c': 0,  # Set Add_Ons_Price__c to 0
+                'Instructions__c': '',  # Use the custom dish description as the instructions
+                'Customer_Email__c': email  # Associate the custom dish with the logged-in user
+            }
+
+            # Insert the custom dish as a Cart_Item__c record in Salesforce
+            cart_result = sf.Cart_Item__c.create(cart_item)
+
+            if cart_result.get('success'):
+                return redirect(url_for("cart"))  # Redirect to the cart page after adding to the cart
+            else:
+                return jsonify({"success": False, "error": "Failed to add custom dish to the cart"}), 500
+
         else:
             return jsonify({"success": False, "error": "Failed to create custom dish in Salesforce"}), 500
 
@@ -107,6 +129,7 @@ def generate_custom_dish():
         return jsonify({"success": False, "error": str(e)}), 500
 
 
+
 import re
 @app.route("/edit_profile", methods=["GET", "POST"])
 def edit_profile():