Update app.py

app.py

@@ -118,28 +118,49 @@ def generate_custom_dish():
             if not result.get('success'):
                 return jsonify({"success": False, "error": "Failed to create custom dish in Salesforce"}), 500
 
-        # After ensuring the dish exists, add it to the Cart_Item__c
+        # Query to check if the cart item already exists for the user
         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,
-            'Base_Price__c': price,
-            'Image1__c': item_image_url,
-            '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
-            'Customer_Email__c': email  # Associate the custom dish with the logged-in user
-        }
+        existing_cart_item_query = f"SELECT Id, Name, Quantity__c, Price__c, Base_Price__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Name = '{dish_name}'"
+        existing_cart_item_result = sf.query(existing_cart_item_query)
+
+        if existing_cart_item_result['totalSize'] > 0:
+            # If the cart item exists, update the quantity and price
+            existing_cart_item = existing_cart_item_result['records'][0]
+            new_quantity = existing_cart_item['Quantity__c'] + 1
+            new_price = existing_cart_item['Base_Price__c'] * new_quantity
+
+            # Update the cart item with new quantity and price
+            updated_cart_item = {
+                'Id': existing_cart_item['Id'],
+                'Quantity__c': new_quantity,
+                'Price__c': new_price
+            }
 
-        # Insert the custom dish as a Cart_Item__c record in Salesforce
-        cart_result = sf.Cart_Item__c.create(cart_item)
+            # Update the existing Cart_Item__c record in Salesforce
+            cart_update_result = sf.Cart_Item__c.update(updated_cart_item)
 
-        if cart_result.get('success'):
-            return redirect(url_for("cart"))  # Redirect to the cart page after adding to the cart
+            if not cart_update_result.get('success'):
+                return jsonify({"success": False, "error": "Failed to update cart item in Salesforce"}), 500
         else:
-            return jsonify({"success": False, "error": "Failed to add custom dish to the cart"}), 500
+            # If the cart item does not exist, create a new one
+            cart_item = {
+                'Name': dish_name,
+                'Price__c': price,
+                'Base_Price__c': price,
+                'Image1__c': item_image_url,
+                'Quantity__c': 1,  # Set the quantity to 1 for the new cart item
+                'Add_Ons__c': '',  # Set Add_ons__c to empty
+                'Add_Ons_Price__c': 0,  # Set Add_ons_Price__c to 0
+                '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 not cart_result.get('success'):
+                return jsonify({"success": False, "error": "Failed to add custom dish to the cart"}), 500
+
+        return redirect(url_for("cart"))  # Redirect to the cart page after adding to the cart
 
     except Exception as e:
         return jsonify({"success": False, "error": str(e)}), 500
@@ -147,7 +168,6 @@ def generate_custom_dish():
 
 
 
-
 import re
 @app.route("/edit_profile", methods=["GET", "POST"])
 def edit_profile():