nagasurendra commited on
Commit
5885c76
·
verified ·
1 Parent(s): 17b014e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -10
app.py CHANGED
@@ -64,21 +64,29 @@ def get_ingredients():
64
 
65
  @app.route('/get_menu_items', methods=['POST'])
66
  def get_menu_items():
67
- category = request.json.get('category', '').strip().lower()
68
- logging.debug(f"Received category: {category}")
69
 
70
- if category == 'fish':
71
- logging.debug("Fetching fish-based menu items...")
72
- soql = "SELECT Item_Name__c, Image_URL__c FROM Menu_Item__c WHERE Category__c = 'Fish' LIMIT 200"
73
- else:
74
- logging.debug("Invalid category received.")
75
- return jsonify({"error": "Invalid category."}), 400
 
 
 
 
 
76
 
77
  try:
78
  result = sf.query(soql)
79
  menu_items = [
80
- {"name": record['Item_Name__c'], "image_url": record.get('Image_URL__c', '')}
81
- for record in result['records'] if 'Item_Name__c' in record
 
 
 
82
  ]
83
  logging.debug(f"Fetched {len(menu_items)} menu items.")
84
  return jsonify({"menu_items": menu_items})
@@ -86,6 +94,7 @@ def get_menu_items():
86
  logging.error(f"Error while fetching menu items: {str(e)}")
87
  return jsonify({"error": f"Failed to fetch menu items: {str(e)}"}), 500
88
 
 
89
  @app.route('/submit_ingredients', methods=['POST'])
90
  def submit_ingredients():
91
  data = request.json
 
64
 
65
  @app.route('/get_menu_items', methods=['POST'])
66
  def get_menu_items():
67
+ ingredient = request.json.get('ingredient', '').strip().lower()
68
+ logging.debug(f"Received ingredient: {ingredient}")
69
 
70
+ if not ingredient:
71
+ logging.debug("No ingredient provided.")
72
+ return jsonify({"error": "No ingredient provided."}), 400
73
+
74
+ # Query to fetch menu items where the name contains the selected ingredient
75
+ soql = f"""
76
+ SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c, Total_Ordered__c
77
+ FROM Menu_Item__c
78
+ WHERE LOWER(Name) LIKE '%{ingredient}%'
79
+ LIMIT 200
80
+ """
81
 
82
  try:
83
  result = sf.query(soql)
84
  menu_items = [
85
+ {"name": record['Name'], "price": record.get('Price__c', ''), "description": record.get('Description__c', ''),
86
+ "image1": record.get('Image1__c', ''), "image2": record.get('Image2__c', ''),
87
+ "veg_nonveg": record.get('Veg_NonVeg__c', ''), "section": record.get('Section__c', ''),
88
+ "total_ordered": record.get('Total_Ordered__c', '')}
89
+ for record in result['records'] if 'Name' in record
90
  ]
91
  logging.debug(f"Fetched {len(menu_items)} menu items.")
92
  return jsonify({"menu_items": menu_items})
 
94
  logging.error(f"Error while fetching menu items: {str(e)}")
95
  return jsonify({"error": f"Failed to fetch menu items: {str(e)}"}), 500
96
 
97
+
98
  @app.route('/submit_ingredients', methods=['POST'])
99
  def submit_ingredients():
100
  data = request.json