from flask import Flask, render_template, request, jsonify from simple_salesforce import Salesforce from dotenv import load_dotenv import os import logging # 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: {e}") return None # Initialize Salesforce connection sf = get_salesforce_connection() @app.route('/') def index(): return render_template('index.html') # Fetch items from Sector_Detail__c for dietary preferences @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 dietary_preference = request.json.get('dietary_preference', 'both').lower() try: soql = "SELECT Name, Image_URL__c, Category__c FROM Sector_Detail__c" if dietary_preference == 'vegetarian': soql += " WHERE Category__c = 'Veg'" elif dietary_preference == 'non-vegetarian': soql += " WHERE Category__c = 'Non-Veg'" soql += " LIMIT 200" logger.info(f"Executing SOQL query: {soql}") result = sf.query(soql) items = [ { "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.info(f"Fetched {len(items)} items from Sector_Detail__c for {dietary_preference}") return jsonify({"menu_items": items}) except Exception as e: logger.error(f"Failed to fetch items from Sector_Detail__c: {str(e)}") return jsonify({"error": f"Failed to fetch items from Salesforce: {str(e)}"}), 500 # Fetch suggestions from Ingredient_Object__c @app.route('/suggest_items', methods=['POST']) def suggest_items(): global sf if not sf: sf = get_salesforce_connection() if not sf: return jsonify({"error": "Unable to connect to Salesforce"}), 500 search_term = request.json.get('search_term', '').strip() if not search_term: return jsonify({"error": "Search term is required"}), 400 try: soql = f"SELECT Name, Image_URL__c FROM Ingredient_Object__c WHERE Name LIKE '%{search_term}%' OR Ingredients__c LIKE '%{search_term}%' LIMIT 10" logger.info(f"Executing SOQL query: {soql}") result = sf.query(soql) suggestions = [ {"name": record['Name'], "image_url": record.get('Image_URL__c', '')} for record in result['records'] if 'Name' in record ] logger.info(f"Fetched {len(suggestions)} suggestions for '{search_term}'") return jsonify({"suggestions": suggestions}) except Exception as e: logger.error(f"Failed to fetch suggestions: {str(e)}") return jsonify({"error": "Failed to fetch suggestions from Salesforce"}), 500 # Fetch item details from Ingredient_Object__c @app.route('/get_item_details', methods=['POST']) def get_item_details(): global sf if not sf: sf = get_salesforce_connection() if not sf: return jsonify({"error": "Unable to connect to Salesforce"}), 500 item_name = request.json.get('item_name', '').strip() if not item_name: return jsonify({"error": "Item name is required"}), 400 try: soql = f"SELECT Name, Description__c, Ingredients__c, Image_URL__c FROM Ingredient_Object__c WHERE Name LIKE '%{item_name}%' LIMIT 1" logger.info(f"Executing SOQL query: {soql}") result = sf.query(soql) if result['totalSize'] == 0: soql_fallback = f"SELECT Name, Image_URL__c FROM Ingredient_Object__c WHERE Name LIKE '%{item_name}%' LIMIT 1" result_fallback = sf.query(soql_fallback) if result_fallback['totalSize'] > 0: record = result_fallback['records'][0] item_details = { "name": record.get('Name', ''), "description": "Description not available", "ingredients": "Ingredients not listed", "image_url": record.get('Image_URL__c', '') } logger.info(f"Fallback: Found '{item_name}' in Ingredient_Object__c") return jsonify({"item_details": item_details}) return jsonify({"error": f"No item found matching '{item_name}' in Ingredient_Object__c"}), 404 record = result['records'][0] item_details = { "name": record.get('Name', ''), "description": record.get('Description__c', 'No description available'), "ingredients": record.get('Ingredients__c', 'No ingredients listed'), "image_url": record.get('Image_URL__c', '') } logger.info(f"Fetched details for '{item_name}'") return jsonify({"item_details": item_details}) except Exception as e: logger.error(f"Failed to fetch item details: {str(e)}") return jsonify({"error": "Failed to fetch item details from Salesforce"}), 500 # Submit selected items to Ingredient_Object__c @app.route('/submit_items', methods=['POST']) def submit_items(): 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', []) if not items: return jsonify({"error": "No items to submit"}), 400 try: for item in items: sf.Ingredient_Object__c.create({ 'Name': item['name'], 'Description__c': item['description'], 'Ingredients__c': item['ingredients'], 'Image_URL__c': item['image_url'], 'Category__c': item.get('category', 'Both') }) logger.info(f"Submitted item to Ingredient_Object__c: {item['name']}") return jsonify({"success": f"Successfully submitted {len(items)} items to Salesforce"}) except Exception as e: logger.error(f"Failed to submit items: {str(e)}") return jsonify({"error": "Failed to submit items to Salesforce"}), 500 if __name__ == '__main__': app.run(debug=True, host='0.0.0.0', port=7860)