Update app.py

app.py

@@ -120,6 +120,8 @@ def login():
             return render_template("login.html", error=f"Error: {str(e)}")
 
     return render_template("login.html")
+
+    
 @app.route("/logout")
 def logout():
     # Retrieve table number before clearing session
@@ -134,5 +136,160 @@ def logout():
     # Pass table number to redirect page
     return render_template("redirect_page.html", table_number=table_number)
 
+
+
+@app.route('/get_ingredients', methods=['POST'])
+def get_ingredients():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Unable to connect to Salesforce"}), 500
+
+    data = request.json
+    dietary_preference = data.get('dietary_preference', 'both').lower()
+
+    try:
+        category_map = {
+            'vegetarian': 'Veg',
+            'non-vegetarian': 'Non-Veg',
+            'chicken': 'Non-Veg',
+            'beef': 'Non-Veg',
+            'lamb': 'Non-Veg',
+            'both': 'both'
+        }
+        category = category_map.get(dietary_preference, 'both')
+        soql = f"SELECT Name, Image_URL__c, Category__c FROM Sector_Detail__c WHERE Category__c = '{category}'"
+        soql += " LIMIT 200"
+        logger.debug(f"Executing SOQL query for Sector_Detail__c: {soql}")
+        result = sf.query(soql)
+        ingredients = [
+            {
+                "name": record['Name'],
+                "image_url": record.get('Image_URL__c', ''),
+                "category": record.get('Category__c', '')
+            }
+            for record in result['records'] if 'Name' in record
+        ]
+        logger.debug(f"Fetched {len(ingredients)} ingredients from Sector_Detail__c")
+        return jsonify({"ingredients": ingredients})
+    except Exception as e:
+        logger.error(f"Failed to fetch ingredients: {str(e)}")
+        return jsonify({"error": f"Failed to fetch ingredients from Salesforce: {str(e)}"}), 500
+
+@app.route('/get_menu_items', methods=['POST'])
+def get_menu_items():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Unable to connect to Salesforce"}), 500
+
+    data = request.json
+    ingredient_names = data.get('ingredient_names', '')
+    category = data.get('category', '')
+
+    try:
+        soql = "SELECT Name, Description__c, Image1__c, Image2__c, Price__c, Section__c, Veg_NonVeg__c, Total_Ordered__c FROM Menu_Item__c"
+        conditions = []
+        if ingredient_names:
+            words = ingredient_names.split()
+            name_conditions = [f"Name LIKE '%{word}%'" for word in words]
+            conditions.append(f"({' OR '.join(name_conditions)})")
+        if category:
+            if category.lower() == 'vegetarian':
+                conditions.append("Veg_NonVeg__c = 'Vegetarian'")
+            elif category.lower() == 'non-vegetarian':
+                conditions.append("Veg_NonVeg__c = 'Non-Vegetarian'")
+        if conditions:
+            soql += " WHERE " + " AND ".join(conditions)
+        soql += " LIMIT 200"
+        logger.debug(f"Executing SOQL query for Menu_Item__c: {soql}")
+        result = sf.query(soql)
+        menu_items = [
+            {
+                "name": record['Name'],
+                "description": record.get('Description__c', 'No description available'),
+                "image_url": record.get('Image1__c', '') or record.get('Image2__c', ''),
+                "price": record.get('Price__c', 0.0),
+                "section": record.get('Section__c', ''),
+                "veg_nonveg": record.get('Veg_NonVeg__c', ''),
+                "total_ordered": record.get('Total_Ordered__c', 0)
+            }
+            for record in result['records'] if 'Name' in record
+        ]
+        logger.debug(f"Fetched {len(menu_items)} menu items")
+        return jsonify({"menu_items": menu_items})
+    except Exception as e:
+        logger.error(f"Failed to fetch menu items: {str(e)}")
+        return jsonify({"error": f"Failed to fetch menu items from Salesforce: {str(e)}"}), 500
+
+@app.route('/submit_customization_ingredients', methods=['POST'])
+def submit_customization_ingredients():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Unable to connect to Salesforce"}), 500
+
+    data = request.json
+    items = data.get('items', [])
+    menu_item = data.get('menu_item', {})
+    ingredients = data.get('ingredients', [])
+    instructions = data.get('instructions', '')
+
+    try:
+        if items:  # Cart submission
+            for item in items:
+                ingredient_names = ', '.join(i['name'] for i in item.get('ingredients', [])) if item.get('ingredients') else ''
+                base_price = item.get('price', 0.0)
+                quantity = 1
+                addons_price = 0
+                total_price = (base_price * quantity) + addons_price
+
+                sf.Cart_Item__c.create({
+                    'Name': item['name'],
+                    'Base_Price__c': base_price,
+                    'Quantity__c': quantity,
+                    'Add_Ons__c': ingredient_names,
+                    'Add_Ons_Price__c': addons_price,
+                    'Price__c': total_price,
+                    'Image1__c': item.get('image_url', ''),
+                    'Instructions__c': item.get('instructions', ''),
+                    'Category__c': item.get('veg_nonveg', ''),
+                    'Section__c': item.get('section', '')
+                })
+            logger.debug(f"Submitted {len(items)} items to Cart_Item__c")
+            return jsonify({"success": True, "message": f"Submitted {len(items)} items"})
+
+        elif menu_item:  # Single item customization
+            ingredient_names = ', '.join(i['name'] for i in ingredients) if ingredients else ''
+            base_price = menu_item.get('price', 0.0)
+            quantity = 1
+            addons_price = 0
+            total_price = (base_price * quantity) + addons_price
+
+            sf.Cart_Item__c.create({
+                'Name': menu_item['name'],
+                'Base_Price__c': base_price,
+                'Quantity__c': quantity,
+                'Add_Ons__c': ingredient_names,
+                'Add_Ons_Price__c': addons_price,
+                'Price__c': total_price,
+                'Image1__c': menu_item.get('image_url', ''),
+                'Instructions__c': instructions,
+                'Category__c': menu_item.get('veg_nonveg', ''),
+                'Section__c': menu_item.get('section', '')
+            })
+            logger.debug(f"Submitted customization for {menu_item['name']} to Cart_Item__c")
+            return jsonify({"success": True, "message": "Customization submitted"})
+
+        else:
+            return jsonify({"error": "No items or menu item provided"}), 400
+
+    except Exception as e:
+        logger.error(f"Failed to submit: {str(e)}")
+        return jsonify({"error": f"Failed to submit: {str(e)}"}), 500
+
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=7860)