Update app.py

app.py

@@ -63,9 +63,6 @@ def generate_coupon_code(length=10):
     """Generates a random alphanumeric coupon code"""
     characters = string.ascii_uppercase + string.digits  # A-Z, 0-9
     return ''.join(random.choice(characters) for _ in range(length))
-
-
-
 @app.route("/generate_custom_dish", methods=["POST"])
 def generate_custom_dish():
     try:
@@ -89,7 +86,7 @@ def generate_custom_dish():
                    "Non veg" if any(word in description.lower() for word in non_veg_keywords) else \
                    "both"
 
-        # Query to check if the dish already exists in Salesforce
+        # Query to check if the dish already exists in Salesforce (Custom_Dish__c object)
         existing_dish_query = f"SELECT Id, Name, Price__c, Image1__c, Image2__c, Description__c, Veg_NonVeg__c FROM Custom_Dish__c WHERE Name = '{dish_name}'"
         existing_dish_result = sf.query(existing_dish_query)
 
@@ -112,51 +109,43 @@ def generate_custom_dish():
                 'Total_Ordered__c': 0
             }
 
-            # Insert the custom dish into Salesforce
+            # Insert the custom dish into Salesforce (Custom_Dish__c object)
             result = sf.Custom_Dish__c.create(custom_dish)
 
             if not result.get('success'):
                 return jsonify({"success": False, "error": "Failed to create custom dish in Salesforce"}), 500
 
-        # Query to check if the cart item already exists for the user
+        # After ensuring the dish exists, check if it's already in the Cart_Item__c
         email = session.get('user_email')  # Assuming you have the user's email in session
-        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 the cart item exists, update the quantity and price
-        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
+        
+        # Query to check if the custom dish already exists in the cart for the logged-in user
+        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
+            
+            # 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)
 
-            # Update the existing Cart_Item__c record in Salesforce
-            try:
-                cart_update_result = sf.Cart_Item__c.update(existing_cart_item['Id'], updated_cart_item)
-
-                # Check the result; Salesforce may return different types of responses
-                if isinstance(cart_update_result, dict) and cart_update_result.get('success'):
-                    return jsonify({"success": True, "message": "Cart item updated successfully"}), 200
-                else:
-                    return jsonify({"success": False, "error": "Failed to update cart item in Salesforce"}), 500
-
-    except Exception as e:
-        return jsonify({"success": False, "error": str(e)}), 500
-
+            if not cart_item_update.get('success'):
+                return jsonify({"success": False, "error": "Failed to update cart item"}), 500
         else:
-            # If the cart item does not exist, create a new one
+            # 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,  # Set the quantity to 1 for the new cart item
+                '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
@@ -168,11 +157,13 @@ def generate_custom_dish():
             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
+        # 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
 
+