Update app.py

app.py

@@ -79,31 +79,18 @@ def generate_custom_dish():
             return jsonify({"success": False, "error": "Both fields are required"}), 400
 
         # Check if the custom dish already exists (case-sensitive check)
-        query = f"SELECT Id, Name, Total_Ordered__c FROM Custom_Dish__c WHERE Name = '{dish_name}'"
+        query = f"SELECT Id, Name FROM Custom_Dish__c WHERE Name = '{dish_name}'"
         result = sf.query(query)
 
-        # If there are no records returned, result['records'] will be empty
+        # If the dish exists, use the existing record ID, otherwise create a new one
         if result['records']:
             # Custom dish already exists
+            custom_dish_id = result['records'][0]['Id']
             custom_dish = result['records'][0]  # Get the existing custom dish record
-            total_ordered = custom_dish.get('Total_Ordered__c', 0)  # Safe way to access 'Total_Ordered__c'
-
-            # Increment the Total_Ordered__c by 1
-            new_total_ordered = total_ordered + 1
-
-            # Update the existing custom dish with the new Total_Ordered__c value
-            update_data = {
-                'Total_Ordered__c': new_total_ordered
-            }
-
-            # Update the existing custom dish in Salesforce
-            update_result = sf.Custom_Dish__c.update(custom_dish_id, update_data)
-
-            if not update_result.get('success', False):  # Ensure 'success' exists
-                return jsonify({"success": False, "error": "Failed to update custom dish in Salesforce"}), 500
 
+            # No need to create a new dish, just add it to the cart
             # Set up cart item using the existing custom dish details
-            price = custom_dish.get('Price__c', 0)  # Safe way to access 'Price__c'
+            price = custom_dish['Price__c']  # Use the existing price
             email = session.get('user_email')  # Assuming you have the user's email in session
 
             # Create the cart item with custom dish details
@@ -123,7 +110,7 @@ def generate_custom_dish():
             # Insert the cart item as a Cart_Item__c record in Salesforce
             cart_result = sf.Cart_Item__c.create(cart_item)
 
-            if cart_result.get('success', False):
+            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
@@ -148,13 +135,13 @@ def generate_custom_dish():
                 'Image2__c': item_image_url2,
                 'Description__c': description,
                 'Veg_NonVeg__c': category,
-                'Total_Ordered__c': 1  # Set Total_Ordered__c to 1 for new custom dishes
+                'Total_Ordered__c': 0
             }
 
             # Insert the custom dish into Salesforce
             result = sf.Custom_Dish__c.create(custom_dish)
 
-            if result.get('success', False):
+            if result.get('success'):
                 # 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
 
@@ -173,7 +160,7 @@ def generate_custom_dish():
                 # 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', False):
+                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