Update app.py

app.py

@@ -768,7 +768,7 @@ def add_suggestion_to_cart():
 @app.route('/cart/add', methods=['POST'])
 def add_to_cart():
     data = request.json
-    item_name = data.get('itemName').strip()
+    item_name = data.get('itemName', '').strip()
     item_price = data.get('itemPrice')
     item_image = data.get('itemImage')
     addons = data.get('addons', [])
@@ -777,20 +777,28 @@ def add_to_cart():
     section = data.get('section')
     customer_email = session.get('user_email')
 
+    # Basic validation for required fields
     if not item_name or not item_price:
-        return jsonify({"success": False, "error": "Item name and price are required."})
+        return jsonify({"success": False, "error": "Item name and price are required."}), 400
+
+    if not customer_email:
+        return jsonify({"success": False, "error": "User email is required."}), 400
 
     try:
+        # Query to check if the item is already in the cart
         query = f"""
-            SELECT Id, Quantity__c, Add_Ons__c, Add_Ons_Price__c, Instructions__c FROM Cart_Item__c
+            SELECT Id, Quantity__c, Add_Ons__c, Add_Ons_Price__c, Instructions__c 
+            FROM Cart_Item__c
             WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'
         """
         result = sf.query(query)
         cart_items = result.get("records", [])
 
+        # Calculate the total price for the addons
         addons_price = sum(addon['price'] for addon in addons)
         new_addons = "; ".join([f"{addon['name']} (${addon['price']})" for addon in addons])
 
+        # If the item is already in the cart, update it
         if cart_items:
             cart_item_id = cart_items[0]['Id']
             existing_quantity = cart_items[0]['Quantity__c']
@@ -798,19 +806,23 @@ def add_to_cart():
             existing_addons_price = cart_items[0].get('Add_Ons_Price__c', 0)
             existing_instructions = cart_items[0].get('Instructions__c', "")
 
+            # Combine the new addons with the existing ones
             combined_addons = existing_addons if existing_addons != "None" else ""
             if new_addons:
                 combined_addons = f"{combined_addons}; {new_addons}".strip("; ")
 
+            # Combine existing instructions with new instructions
             combined_instructions = existing_instructions
             if instructions:
                 combined_instructions = f"{combined_instructions} | {instructions}".strip(" | ")
 
+            # Calculate total addons price
             combined_addons_list = combined_addons.split("; ")
             combined_addons_price = sum(
                 float(addon.split("($")[1][:-1]) for addon in combined_addons_list if "($" in addon
             )
 
+            # Update the cart item in Salesforce
             sf.Cart_Item__c.update(cart_item_id, {
                 "Quantity__c": existing_quantity + 1,
                 "Add_Ons__c": combined_addons,
@@ -821,12 +833,14 @@ def add_to_cart():
                 "Section__c": section
             })
         else:
+            # If the item is not already in the cart, create a new entry
             addons_string = "None"
             if addons:
                 addons_string = new_addons
 
             total_price = item_price + addons_price
 
+            # Create new cart item in Salesforce
             sf.Cart_Item__c.create({
                 "Name": item_name,
                 "Price__c": total_price,
@@ -842,10 +856,11 @@ def add_to_cart():
             })
 
         return jsonify({"success": True, "message": "Item added to cart successfully."})
+    
     except Exception as e:
+        # Log the error for debugging
         print(f"Error adding item to cart: {str(e)}")
-        return jsonify({"success": False, "error": str(e)})
-
+        return jsonify({"success": False, "error": "An error occurred while adding the item to the cart."}), 500
 
 
 @app.route("/cart/add_item", methods=["POST"])
@@ -901,8 +916,9 @@ def get_addons():
     item_name = request.args.get('item_name')  
     item_section = request.args.get('item_section')  
 
+    # Check if both item_name and item_section are provided
     if not item_name or not item_section:
-        return jsonify({"success": False, "error": "Item name and section are required."})
+        return jsonify({"success": False, "error": "Item name and section are required."}), 400
 
     try:
         # Fetch customization options from Salesforce based on the section
@@ -914,10 +930,20 @@ def get_addons():
         result = sf.query(query)
         addons = result.get('records', [])
 
+        # Check if we found any addons
+        if not addons:
+            return jsonify({"success": False, "error": "No customization options found for the given section."}), 404
+
         # Format data for frontend
         formatted_addons = []
         for addon in addons:
-            options = addon.get("Options__c", "").split(", ")  # Convert comma-separated options into a list
+            # Ensure 'Options__c' exists and is not None
+            options = addon.get("Options__c", "")
+            if options:  # If options are available, split them
+                options = options.split(", ")  # Convert comma-separated options into a list
+            else:
+                options = []  # If no options, default to an empty list
+
             formatted_addons.append({
                 "name": addon["Name"],
                 "type": addon["Customization_Type__c"],
@@ -930,8 +956,10 @@ def get_addons():
         return jsonify({"success": True, "addons": formatted_addons})
 
     except Exception as e:
-        print(f"Error fetching add-ons: {e}")
-        return jsonify({"success": False, "error": "Unable to fetch add-ons. Please try again later."})
+        # Log the exception for debugging
+        app.logger.error(f"Error fetching addons: {str(e)}")
+        return jsonify({"success": False, "error": "An error occurred while fetching customization options."}), 500
+
 
 @app.route("/cart/update_quantity", methods=["POST"])
 def update_quantity():