from flask import Flask, render_template, request, jsonify, send_from_directory from simple_salesforce import Salesforce from dotenv import load_dotenv import os import logging import uuid from datetime import datetime # Load environment variables load_dotenv() # Set up logging logging.basicConfig(level=logging.DEBUG) logger = logging.getLogger(__name__) app = Flask(__name__, template_folder='templates', static_folder='static') # Salesforce connection function def get_salesforce_connection(): try: sf = Salesforce( username=os.getenv('SFDC_USERNAME'), password=os.getenv('SFDC_PASSWORD'), security_token=os.getenv('SFDC_SECURITY_TOKEN'), domain=os.getenv('SFDC_DOMAIN', 'login') ) logger.info("Successfully connected to Salesforce") return sf except Exception as e: logger.error(f"Error connecting to Salesforce: {str(e)}") return None # Initialize Salesforce connection sf = get_salesforce_connection() @app.route('/') def index(): return render_template('index.html') @app.route('/static/') def static_files(filename): return send_from_directory(app.static_folder, filename) @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() print(f"Fetching ingredients for dietary preference: {dietary_preference}") # Log the request 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)