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.INFO) 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('chef-bot.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 try: soql = "SELECT Name, Category__c FROM Sector_Detail__c WHERE Category__c = 'Both'" soql += " LIMIT 200" logger.info(f"Executing SOQL query for Sector_Detail__c: {soql}") result = sf.query(soql) ingredients = [ { "name": record['Name'], "category": record.get('Category__c', '') } for record in result['records'] if 'Name' in record ] logger.info(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: logger.error("Salesforce connection failed after retry") return jsonify({"error": "Unable to connect to Salesforce"}), 500 data = request.json dietary_preference = data.get('dietary_preference', 'both').lower() search_term = data.get('search_term', '').strip() source = data.get('source', 'Menu_Item__c') try: items = [] if source == 'Menu_Item__c': soql_menu = "SELECT Name, Description__c, Image1__c, Ingredientsinfo__c, NutritionalInfo__c, Price__c, Sector__c, Spice_Levels__c, Veg_NonVeg__c, Category__c, Dynamic_Dish__c FROM Menu_Item__c" if search_term: soql_menu += f" WHERE Name LIKE '%{search_term}%'" elif dietary_preference == 'vegetarian': soql_menu += " WHERE Veg_NonVeg__c = 'Vegetarian'" elif dietary_preference == 'non-vegetarian': soql_menu += " WHERE Veg_NonVeg__c = 'Non-Vegetarian'" soql_menu += " LIMIT 200" logger.info(f"Executing SOQL query for Menu_Item__c: {soql_menu}") result_menu = sf.query(soql_menu) menu_items = [ { "name": record['Name'], "description": record.get('Description__c', 'No description available'), "image_url": record.get('Image1__c', ''), "ingredients": record.get('Ingredientsinfo__c', ''), "nutritional_info": record.get('NutritionalInfo__c', ''), "price": record.get('Price__c', 0.0), "sector": record.get('Sector__c', ''), "spice_levels": record.get('Spice_Levels__c', ''), "veg_nonveg": record.get('Veg_NonVeg__c', ''), "category": record.get('Category__c', ''), "dynamic_dish": record.get('Dynamic_Dish__c', False), "source": "Menu_Item__c" } for record in result_menu['records'] if 'Name' in record ] items.extend(menu_items) logger.info(f"Fetched {len(items)} items from {source} for {dietary_preference or search_term}") return jsonify({"menu_items": items}) except Exception as e: logger.error(f"Failed to fetch items: {str(e)}") return jsonify({"error": f"Failed to fetch items from Salesforce: {str(e)}"}), 500 @app.route('/get_related_menu_items', methods=['POST']) def get_related_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_name = data.get('ingredient_name', '').strip() dietary_preference = data.get('dietary_preference', 'both').lower() if not ingredient_name: return jsonify({"error": "Ingredient name is required"}), 400 try: soql = "SELECT Name, Description__c, Image1__c, Ingredientsinfo__c, NutritionalInfo__c, Price__c, Sector__c, Spice_Levels__c, Veg_NonVeg__c, Category__c, Dynamic_Dish__c FROM Menu_Item__c WHERE Ingredientsinfo__c LIKE '%" + ingredient_name + "%'" if dietary_preference == 'vegetarian': soql += " AND Veg_NonVeg__c = 'Vegetarian'" elif dietary_preference == 'non-vegetarian': soql += " AND Veg_NonVeg__c = 'Non-Vegetarian'" soql += " LIMIT 200" logger.info(f"Executing SOQL query for related 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', ''), "ingredients": record.get('Ingredientsinfo__c', ''), "nutritional_info": record.get('NutritionalInfo__c', ''), "price": record.get('Price__c', 0.0), "sector": record.get('Sector__c', ''), "spice_levels": record.get('Spice_Levels__c', ''), "veg_nonveg": record.get('Veg_NonVeg__c', ''), "category": record.get('Category__c', ''), "dynamic_dish": record.get('Dynamic_Dish__c', False), "source": "Menu_Item__c" } for record in result['records'] if 'Name' in record ] logger.info(f"Fetched {len(menu_items)} related menu items for {ingredient_name}") return jsonify({"menu_items": menu_items}) except Exception as e: logger.error(f"Failed to fetch related menu items: {str(e)}") return jsonify({"error": f"Failed to fetch related menu items from Salesforce: {str(e)}"}), 500 @app.route('/submit_ingredients', methods=['POST']) def submit_ingredients(): global sf if not sf: sf = get_salesforce_connection() if not sf: return jsonify({"error": "Unable to connect to Salesforce"}), 500 items = request.json.get('items', []) custom_order_name = request.json.get('custom_order_name', '') ingredients_info = request.json.get('ingredients_info', '') if not items: return jsonify({"error": "No items to submit"}), 400 try: ingredient_name = custom_order_name or f"Order_{datetime.now().strftime('%Y%m%d')}_{uuid.uuid4().hex[:8]}" item_names = ', '.join(item['name'] for item in items) description = f"Contains: {item_names}" for item in items: item_description = item['description'] if item.get('ingredientsInfo'): item_description += f", additional info: {item['ingredientsInfo']}" sf.Ingredient_Object__c.create({ 'Ingredient_Name__c': ingredient_name, 'Category__c': item.get('category', ''), 'Description__c': item_description, 'Image_URL__c': item.get('image_url', ''), 'Quantity__c': item.get('quantity', 1), 'Ingredientsinfo__c': ingredients_info or item.get('ingredientsInfo', '') }) logger.info(f"Submitted {len(items)} items under {ingredient_name}") return jsonify({"success": f"Submitted {len(items)} items under {ingredient_name}", "ingredient_name": ingredient_name}) except Exception as e: logger.error(f"Failed to submit items: {str(e)}") return jsonify({"error": f"Failed to submit items: {str(e)}"}), 500 if __name__ == '__main__': app.run(debug=True, host='0.0.0.0', port=7860)