lokesh341's picture
Update app.py
40bbbf4 verified
raw
history blame
6.84 kB
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 menu items with dietary filter
@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
dietary_preference = request.json.get('dietary_preference', 'both').lower()
try:
soql = "SELECT Name, Image_URL__c, Category__c FROM Ingredient_Object__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)
menu_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(menu_items)} menu items for {dietary_preference}")
return jsonify({"menu_items": menu_items})
except Exception as e:
logger.error(f"Failed to fetch menu items: {str(e)}")
return jsonify({"error": "Failed to fetch menu items from Salesforce"}), 500
# Fetch suggestions based on search term
@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
@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 menu items")
return jsonify({"item_details": item_details})
return jsonify({"error": f"No item found matching '{item_name}'"}), 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 Salesforce
@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
ingredients = request.json.get('ingredients', [])
if not ingredients:
return jsonify({"error": "No ingredients provided"}), 400
try:
# Create a new record in Ingredient_Object__c with the selected ingredients
ingredient_names = ", ".join(ingredients)
sf.Ingredient_Object__c.create({
'Name': f"User Selection {ingredient_names[:50]}", # Truncate if too long
'Ingredients__c': ingredient_names
})
logger.info(f"Submitted ingredients to Salesforce: {ingredient_names}")
return jsonify({"success": "Ingredients submitted successfully"})
except Exception as e:
logger.error(f"Failed to submit ingredients: {str(e)}")
return jsonify({"error": "Failed to submit ingredients to Salesforce"}), 500
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=7860)