Yaswanth56 commited on
Commit
b9667b8
·
verified ·
1 Parent(s): 7eddb98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -38,10 +38,10 @@ def get_ingredients():
38
  dietary_preference = request.json.get('dietary_preference', '').strip().lower()
39
  logging.debug(f"Received dietary preference: {dietary_preference}")
40
 
41
- # Map dietary preference to SOQL condition
42
  preference_map = {
43
- 'vegetarian': "Category__c = 'Veg'", # Adjust field name based on Sector_Detail__c schema
44
- 'non-vegetarian': "Category__c = 'Non-Veg'"
45
  }
46
  condition = preference_map.get(dietary_preference)
47
 
@@ -50,10 +50,10 @@ def get_ingredients():
50
  return jsonify({"error": "Invalid dietary preference."}), 400
51
 
52
  try:
53
- soql = f"SELECT Name FROM Sector_Detail__c WHERE {condition} LIMIT 200"
54
  result = sf.query(soql)
55
  ingredients = [
56
- {"name": record['Name'], "image_url": "https://picsum.photos/80"} # Default image URL
57
  for record in result['records'] if 'Name' in record
58
  ]
59
  logging.debug(f"Fetched ingredients: {ingredients}")
 
38
  dietary_preference = request.json.get('dietary_preference', '').strip().lower()
39
  logging.debug(f"Received dietary preference: {dietary_preference}")
40
 
41
+ # Map dietary preference to SOQL condition for Menu_Item__c
42
  preference_map = {
43
+ 'vegetarian': "Category__c = 'Veg'",
44
+ 'non-vegetarian': "Category__c IN ('Chicken', 'Mutton', 'Fish')"
45
  }
46
  condition = preference_map.get(dietary_preference)
47
 
 
50
  return jsonify({"error": "Invalid dietary preference."}), 400
51
 
52
  try:
53
+ soql = f"SELECT Name, Image1__c FROM Menu_Item__c WHERE {condition} LIMIT 200"
54
  result = sf.query(soql)
55
  ingredients = [
56
+ {"name": record['Name'], "image_url": record.get('Image1__c', '')}
57
  for record in result['records'] if 'Name' in record
58
  ]
59
  logging.debug(f"Fetched ingredients: {ingredients}")