Update menu.py

menu.py

@@ -72,6 +72,11 @@ def menu():
         referral_code = user_result['records'][0].get('Referral__c', 'N/A')
         reward_points = user_result['records'][0].get('Reward_Points__c', 0)
 
+        # Get cart item count
+        cart_query = f"SELECT COUNT() FROM Cart_Item__c WHERE Customer_Email__c = '{user_email}'"
+        cart_count_result = sf.query(cart_query)
+        cart_item_count = cart_count_result['totalSize']
+
         # Query to fetch Menu_Item__c records including Video1__c
         menu_query = """
             SELECT Name, Price__c, Description__c, Image1__c, Image2__c, 
@@ -158,6 +163,7 @@ def menu():
         categories = ["All", "Veg", "Non veg"]
         referral_code = 'N/A'
         reward_points = 0
+        cart_item_count = 0
 
     return render_template(
         "menu.html",
@@ -167,138 +173,8 @@ def menu():
         referral_code=referral_code,
         reward_points=reward_points,
         user_name=user_name,
-        first_letter=first_letter
+        first_letter=first_letter,
+        cart_item_count=cart_item_count
     )
 
-@menu_blueprint.route('/api/addons', methods=['GET'])
-def get_addons():
-    item_name = request.args.get('item_name')  
-    item_section = request.args.get('item_section')  
-
-    if not item_name or not item_section:
-        return jsonify({"success": False, "error": "Item name and section are required."}), 400
-
-    try:
-        query = f"""
-            SELECT Name, Customization_Type__c, Options__c, Max_Selections__c, Extra_Charge__c, Extra_Charge_Amount__c
-            FROM Customization_Options__c
-            WHERE Section__c = '{item_section}'
-        """
-        result = sf.query(query)
-        addons = result.get('records', [])
-
-        if not addons:
-            return jsonify({"success": False, "error": "No customization options found for the given section."}), 404
-
-        formatted_addons = []
-        for addon in addons:
-            options = addon.get("Options__c", "")
-            if options:
-                options = options.split(", ")
-            else:
-                options = []
-
-            formatted_addons.append({
-                "name": addon["Name"],
-                "type": addon["Customization_Type__c"],
-                "options": options,
-                "max_selections": addon.get("Max_Selections__c", 1),
-                "extra_charge": addon.get("Extra_Charge__c", False),
-                "extra_charge_amount": addon.get("Extra_Charge_Amount__c", 0)
-            })
-
-        return jsonify({"success": True, "addons": formatted_addons})
-
-    except Exception as e:
-        print(f"Error fetching addons: {str(e)}")
-        return jsonify({"success": False, "error": "An error occurred while fetching customization options."}), 500
-
-@menu_blueprint.route('/cart/add', methods=['POST'])
-def add_to_cart():
-    try:
-        data = request.json
-        item_name = data.get('itemName', '').strip()
-        item_price = data.get('itemPrice')
-        item_image = data.get('itemImage')
-        addons = data.get('addons', [])
-        instructions = data.get('instructions', '')
-        category = data.get('category')
-        section = data.get('section')
-        quantity = data.get('quantity', 1)
-        customer_email = session.get('user_email')
-
-        if not item_name or not item_price:
-            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
-
-        query = f"""
-            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", [])
-
-        addons_price = sum(addon['price'] for addon in addons)
-        new_addons = "; ".join([f"{addon['name']} (${addon['price']})" for addon in addons])
-
-        if cart_items:
-            cart_item_id = cart_items[0]['Id']
-            existing_quantity = cart_items[0]['Quantity__c']
-            existing_addons = cart_items[0].get('Add_Ons__c', "None")
-            existing_addons_price = cart_items[0].get('Add_Ons_Price__c', 0)
-            existing_instructions = cart_items[0].get('Instructions__c', "")
-
-            combined_addons = existing_addons if existing_addons != "None" else ""
-            if new_addons:
-                combined_addons = f"{combined_addons}; {new_addons}".strip("; ")
-
-            combined_instructions = existing_instructions
-            if instructions:
-                combined_instructions = f"{combined_instructions} | {instructions}".strip(" | ")
-
-            combined_addons_list = combined_addons.split("; ")
-            combined_addons_price = sum(
-                float(addon.split("($")[1][:-1]) for addon in combined_addons_list if "($" in addon
-            )
-
-            sf.Cart_Item__c.update(cart_item_id, {
-                "Quantity__c": existing_quantity + quantity,
-                "Add_Ons__c": combined_addons,
-                "Add_Ons_Price__c": combined_addons_price,
-                "Instructions__c": combined_instructions,
-                "Price__c": (existing_quantity + quantity) * item_price + combined_addons_price,
-                "Category__c": category,
-                "Section__c": section
-            })
-        else:
-            addons_string = "None"
-            if addons:
-                addons_string = new_addons
-
-            total_price = item_price * quantity + addons_price
-
-            sf.Cart_Item__c.create({
-                "Name": item_name,
-                "Price__c": total_price,
-                "Base_Price__c": item_price,
-                "Quantity__c": quantity,
-                "Add_Ons_Price__c": addons_price,
-                "Add_Ons__c": addons_string,
-                "Image1__c": item_image,
-                "Customer_Email__c": customer_email,
-                "Instructions__c": instructions,
-                "Category__c": category,
-                "Section__c": section
-            })
-
-        return jsonify({"success": True, "message": "Item added to cart successfully."})
-
-    except KeyError as e:
-        return jsonify({"success": False, "error": f"Missing required field: {str(e)}"}), 400
-
-    except Exception as e:
-        print(f"Error adding item to cart: {str(e)}")
-        return jsonify({"success": False, "error": "An error occurred while adding the item to the cart."}), 500
+# [Rest of your existing routes...]