Update app.py

app.py

@@ -126,43 +126,81 @@ def generate_custom_dish():
         cart_item_query = f"SELECT Id, Quantity__c, Price__c, Base_Price__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Name = '{dish_name}'"
         cart_item_result = sf.query(cart_item_query)
 
-        if cart_item_result['totalSize'] > 0:
-            # If the custom dish is already in the cart, update the quantity and price
-            cart_item = cart_item_result['records'][0]
-            new_quantity = cart_item['Quantity__c'] + 1  # Increase quantity by 1
-            new_price = price * new_quantity  # Update price based on new quantity
+    #     if cart_item_result['totalSize'] > 0:
+    #         # If the custom dish is already in the cart, update the quantity and price
+    #         cart_item = cart_item_result['records'][0]
+    #         new_quantity = cart_item['Quantity__c'] + 1  # Increase quantity by 1
+    #         new_price = price * new_quantity  # Update price based on new quantity
             
-            # Update the cart item in Salesforce
-            updated_cart_item = {
-                'Quantity__c': new_quantity,
-                'Price__c': new_price
-            }
+    #         # Update the cart item in Salesforce
+    #         updated_cart_item = {
+    #             'Quantity__c': new_quantity,
+    #             'Price__c': new_price
+    #         }
             
-            cart_item_update = sf.Cart_Item__c.update(cart_item['Id'], updated_cart_item)
+    #         cart_item_update = sf.Cart_Item__c.update(cart_item['Id'], updated_cart_item)
+
+    #     else:
+    #         # If the custom dish is not in the cart, create a new cart item
+    #         cart_item = {
+    #             'Name': dish_name,
+    #             'Price__c': price,
+    #             'Base_Price__c': price,
+    #             'Image1__c': item_image_url,
+    #             'Quantity__c': 1,  # Default quantity is 1
+    #             '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)
+
+    #     # Redirect to the cart page after successfully adding or updating the cart item
+    #     return redirect(url_for("cart"))
+
+    # except Exception as e:
+    #     return jsonify({"success": False, "error": str(e)}), 500
 
-        else:
-            # If the custom dish is not in the cart, create a new cart item
-            cart_item = {
-                'Name': dish_name,
-                'Price__c': price,
-                'Base_Price__c': price,
-                'Image1__c': item_image_url,
-                'Quantity__c': 1,  # Default quantity is 1
-                '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
-            }
+                
+  try:
+    if cart_item_result['totalSize'] > 0:
+        # If the custom dish is already in the cart, update the quantity and price
+        cart_item = cart_item_result['records'][0]
+        
+        # Make sure quantity is an integer (rounded if necessary)
+        new_quantity = int(cart_item['Quantity__c']) + 1  # Increase quantity by 1 and ensure it's an integer
+        new_price = price * new_quantity  # Update price based on new quantity
+        
+        # Update the cart item in Salesforce
+        updated_cart_item = {
+            'Quantity__c': new_quantity,
+            'Price__c': new_price
+        }
+        
+        cart_item_update = sf.Cart_Item__c.update(cart_item['Id'], updated_cart_item)
 
-            # Insert the custom dish as a Cart_Item__c record in Salesforce
-            cart_result = sf.Cart_Item__c.create(cart_item)
+    else:
+        # If the custom dish is not in the cart, create a new cart item
+        cart_item = {
+            'Name': dish_name,
+            'Price__c': price,
+            'Base_Price__c': price,
+            'Image1__c': item_image_url,
+            'Quantity__c': 1,  # Default quantity is 1
+            '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
+        }
 
-        # Redirect to the cart page after successfully adding or updating the cart item
-        return redirect(url_for("cart"))
+        # Insert the custom dish as a Cart_Item__c record in Salesforce
+        cart_result = sf.Cart_Item__c.create(cart_item)
 
-    except Exception as e:
-        return jsonify({"success": False, "error": str(e)}), 500
+    # Redirect to the cart page after successfully adding or updating the cart item
+    return redirect(url_for("cart"))
 
-                
+except Exception as e:
+    return jsonify({"success": False, "error": str(e)}), 500