Update app.py

app.py

@@ -82,12 +82,12 @@ def generate_custom_dish():
         query = f"SELECT Id, Name, Total_Ordered__c FROM Custom_Dish__c WHERE Name = '{dish_name}'"
         result = sf.query(query)
 
-        # If the dish exists, use the existing record ID and increment Total_Ordered__c
+        # If there are no records returned, result['records'] will be empty
         if result['records']:
             # Custom dish already exists
             custom_dish = result['records'][0]  # Get the existing custom dish record
-            custom_dish_id = custom_dish['Id']
-            total_ordered = custom_dish.get('Total_Ordered__c', 0)
+            custom_dish_id = custom_dish.get('Id')  # Safe way to access 'Id'
+            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
@@ -100,12 +100,11 @@ def generate_custom_dish():
             # Update the existing custom dish in Salesforce
             update_result = sf.Custom_Dish__c.update(custom_dish_id, update_data)
 
-            # Check if update was successful
-            if not update_result.get('success'):
+            if not update_result.get('success', False):  # Ensure 'success' exists
                 return jsonify({"success": False, "error": "Failed to update custom dish in Salesforce"}), 500
 
             # Set up cart item using the existing custom dish details
-            price = custom_dish['Price__c']  # Use the existing price
+            price = custom_dish.get('Price__c', 0)  # Safe way to access 'Price__c'
             email = session.get('user_email')  # Assuming you have the user's email in session
 
             # Create the cart item with custom dish details
@@ -125,7 +124,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'):
+            if cart_result.get('success', False):
                 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
@@ -156,7 +155,7 @@ def generate_custom_dish():
             # Insert the custom dish into Salesforce
             result = sf.Custom_Dish__c.create(custom_dish)
 
-            if result.get('success'):
+            if result.get('success', False):
                 # 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
 
@@ -169,14 +168,13 @@ def generate_custom_dish():
                     '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'):
+                if cart_result.get('success', False):
                     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