Update app.py

app.py

@@ -78,36 +78,37 @@ def generate_custom_dish():
         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
+        # Check if the custom dish already exists (case-sensitive check)
+        query = f"SELECT Id, Name, Total_Ordered__c FROM Custom_Dish__c WHERE Name = '{dish_name}'"
+        result = sf.query(query)
 
-        # 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"
-
-        # Create a new Custom_Dish__c record in Salesforce
-        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,
-            'Total_Ordered__c': 0
-        }
+        # If the dish exists, use the existing record ID and increment Total_Ordered__c
+        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)
+
+            # Increment the Total_Ordered__c by 1
+            new_total_ordered = total_ordered + 1
+
+            # Update the existing custom dish with the new Total_Ordered__c value
+            update_data = {
+                'Total_Ordered__c': new_total_ordered
+            }
 
-        # Insert the custom dish into Salesforce
-        result = sf.Custom_Dish__c.create(custom_dish)
+            # Update the existing custom dish in Salesforce
+            update_result = sf.Custom_Dish__c.update(custom_dish_id, update_data)
 
-        if result.get('success'):
-            # After creating the custom dish, add it to the Cart_Item__c
+            # Check if update was successful
+            if not update_result.get('success'):
+                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
             email = session.get('user_email')  # Assuming you have the user's email in session
-            
-            # Create the cart item with custom dish details and set other fields to default values
+
+            # Create the cart item with custom dish details
             cart_item = {
                 'Name': dish_name,
                 'Price__c': price,
@@ -116,11 +117,12 @@ 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
+                'Instructions__c': description,  # Use the custom dish description as the instructions
+                'Customer_Email__c': email,  # Associate the custom dish with the logged-in user
+                'Custom_Dish__c': custom_dish_id  # Link the cart item to the existing custom dish
             }
 
-            # Insert the custom dish as a Cart_Item__c record in Salesforce
+            # 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'):
@@ -129,13 +131,65 @@ def generate_custom_dish():
                 return jsonify({"success": False, "error": "Failed to add custom dish to the cart"}), 500
 
         else:
-            return jsonify({"success": False, "error": "Failed to create custom dish in Salesforce"}), 500
+            # Custom dish does not exist, create a new one
+            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"
+
+            # Create a new Custom_Dish__c record in Salesforce
+            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,
+                'Total_Ordered__c': 1  # Set Total_Ordered__c to 1 for new custom dishes
+            }
+
+            # Insert the custom dish into Salesforce
+            result = sf.Custom_Dish__c.create(custom_dish)
+
+            if result.get('success'):
+                # 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
+
+                # Create the cart item with custom dish details and set other fields to default values
+                cart_item = {
+                    'Name': dish_name,
+                    'Price__c': price,
+                    'Base_Price__c': price,
+                    'Image1__c': item_image_url,
+                    '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': description,  # 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'):
+                    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
+
+            else:
+                return jsonify({"success": False, "error": "Failed to create custom dish in Salesforce"}), 500
 
     except Exception as e:
         return jsonify({"success": False, "error": str(e)}), 500
 
 
 
+
 import re
 @app.route("/edit_profile", methods=["GET", "POST"])
 def edit_profile():