Update app.py

app.py

@@ -10,6 +10,7 @@ from cart import cart_blueprint  # Same for other blueprints
 from order import order_blueprint  # Same for user blueprint
 from orderhistory import orderhistory_blueprint
 from user_details import user_details_blueprint
+from customdish import customdish_blueprint
 from datetime import datetime
 from datetime import datetime
 import pytz  # Library to handle timezone conversions
@@ -40,6 +41,7 @@ app.register_blueprint(user_details_blueprint, url_prefix='/user')
 app.register_blueprint(menu_blueprint)
 app.register_blueprint(order_blueprint)
 app.register_blueprint(orderhistory_blueprint, url_prefix='/orderhistory')
+app.register_blueprint(customdish_blueprint, url_prefix='/customdish')
 
 
 
@@ -62,105 +64,6 @@ def home():
 
     return render_template("index.html")
 
-@app.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
-
-
-
-
 app.permanent_session_lifetime = timedelta(minutes=5)
 @app.before_request
 def check_session_timeout():
@@ -193,12 +96,5 @@ def logout():
     # Pass table number to redirect page
     return render_template("redirect_page.html", table_number=table_number)
 
-
-
-
-
-
-
-
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=7860)