Create customdish.py

customdish.py

@@ -0,0 +1,104 @@
+from flask import Blueprint, request, jsonify, redirect, url_for, session
+import random
+from your_salesforce_module import sf  # Import your Salesforce instance or utility
+
+# Create a Blueprint for custom dish-related routes
+customdish_blueprint = Blueprint('customdish', __name__)
+
+@customdish_blueprint.route("/generate_custom_dish", methods=["POST"])
+def generate_custom_dish():
+    try:
+        data = request.form
+        dish_name = data.get("name")
+        description = data.get("description")
+        item_image_url = "https://huggingface.co/spaces/nagasurendra/BiryaniHubflask30/resolve/main/static/customized.jpg"
+        item_image_url2 = "https://huggingface.co/spaces/nagasurendra/BiryaniHubflask30/resolve/main/static/customized1.jpg"
+
+        if not dish_name or not description:
+            return jsonify({"success": False, "error": "Both fields are required"}), 400
+
+        # Generate a random price for the custom dish
+        price = random.randint(10, 30)  # Example logic for price setting
+
+        # Determine Veg/Non-Veg
+        veg_keywords = ["paneer", "vegetable", "mushroom", "cheese"]
+        non_veg_keywords = ["chicken", "mutton", "fish", "egg"]
+        
+        category = "Veg" if any(word in description.lower() for word in veg_keywords) else \
+                   "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 (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)
+
+        if existing_dish_result['totalSize'] > 0:
+            # If the dish exists, use the existing details
+            existing_dish = existing_dish_result['records'][0]
+            price = existing_dish['Price__c']
+            item_image_url = existing_dish['Image1__c']
+            item_image_url2 = existing_dish['Image2__c']
+            category = existing_dish['Veg_NonVeg__c']
+        else:
+            # If the dish does not exist, create a new custom dish
+            custom_dish = {
+                'Name': dish_name,
+                'Price__c': price,
+                'Image1__c': item_image_url,
+                'Image2__c': item_image_url2,
+                'Description__c': description,
+                'Veg_NonVeg__c': category,
+                'Section__c': 'Customized dish',
+                'Total_Ordered__c': 0
+            }
+
+            # 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
+
+        # 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
+        
+        # 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)
+
+        else:
+            # 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,  # 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
+            }
+
+            # Insert the custom dish as a Cart_Item__c record in Salesforce
+            cart_result = sf.Cart_Item__c.create(cart_item)
+
+        # 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
+
+