Yaswanth56 commited on
Commit
124b94c
·
verified ·
1 Parent(s): a634135

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -50,10 +50,10 @@ def get_ingredients():
50
  return jsonify({"error": "Invalid dietary preference."}), 400
51
 
52
  try:
53
- soql = f"SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE {condition} LIMIT 200"
54
  result = sf.query(soql)
55
  ingredients = [
56
- {"name": record['Name'], "image_url": record.get('Image_URL__c', '')}
57
  for record in result['records'] if 'Name' in record
58
  ]
59
  logging.debug(f"Fetched {len(ingredients)} ingredients.")
@@ -69,11 +69,11 @@ def get_menu_items():
69
 
70
  soql = ""
71
  if category == 'chicken':
72
- soql = "SELECT Item_Name__c, Image_URL__c FROM Menu_Item__c WHERE Category__c = 'Chicken' LIMIT 200"
73
  elif category == 'mutton':
74
- soql = "SELECT Item_Name__c, Image_URL__c FROM Menu_Item__c WHERE Category__c = 'Mutton' LIMIT 200"
75
  elif category == 'non-vegetarian':
76
- soql = "SELECT Item_Name__c, Image_URL__c FROM Menu_Item__c WHERE Category__c IN ('Chicken', 'Mutton', 'Fish') LIMIT 200"
77
  else:
78
  logging.debug("Invalid category received.")
79
  return jsonify({"error": "Invalid category."}), 400
@@ -81,8 +81,8 @@ def get_menu_items():
81
  try:
82
  result = sf.query(soql)
83
  menu_items = [
84
- {"name": record['Item_Name__c'], "image_url": record.get('Image_URL__c', '')}
85
- for record in result['records'] if 'Item_Name__c' in record
86
  ]
87
  logging.debug(f"Fetched {len(menu_items)} menu items.")
88
  return jsonify({"menu_items": menu_items})
 
50
  return jsonify({"error": "Invalid dietary preference."}), 400
51
 
52
  try:
53
+ soql = f"SELECT Name, Image1__c FROM Sector_Detail__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 {len(ingredients)} ingredients.")
 
69
 
70
  soql = ""
71
  if category == 'chicken':
72
+ soql = "SELECT Name, Image1__c FROM Menu_Item__c WHERE Category__c = 'Chicken' LIMIT 200"
73
  elif category == 'mutton':
74
+ soql = "SELECT Name, Image1__c FROM Menu_Item__c WHERE Category__c = 'Mutton' LIMIT 200"
75
  elif category == 'non-vegetarian':
76
+ soql = "SELECT Name, Image1__c FROM Menu_Item__c WHERE Category__c IN ('Chicken', 'Mutton', 'Fish') LIMIT 200"
77
  else:
78
  logging.debug("Invalid category received.")
79
  return jsonify({"error": "Invalid category."}), 400
 
81
  try:
82
  result = sf.query(soql)
83
  menu_items = [
84
+ {"name": record['Name'], "image_url": record.get('Image1__c', '')}
85
+ for record in result['records'] if 'Name' in record
86
  ]
87
  logging.debug(f"Fetched {len(menu_items)} menu items.")
88
  return jsonify({"menu_items": menu_items})