Update app.py

app.py

@@ -39,11 +39,14 @@ def get_ingredients():
     logging.debug(f"Received dietary preference: {dietary_preference}")
 
     # Map dietary preference to SOQL condition
-    preference_map = {
-        'vegetarian': "Category__c = 'Veg'",
-        'non-vegetarian': "Category__c = 'Non-Veg'"
-    }
-    condition = preference_map.get(dietary_preference)
+    if dietary_preference == 'both':
+        condition = "Category__c = 'both'"  # This will fetch both vegetarian and non-vegetarian
+    else:
+        preference_map = {
+            'vegetarian': "Category__c = 'Veg'",
+            'non-vegetarian': "Category__c = 'Non-Veg'"
+        }
+        condition = preference_map.get(dietary_preference)
 
     if not condition:
         logging.debug("Invalid dietary preference received.")
@@ -62,6 +65,7 @@ def get_ingredients():
         logging.error(f"Error while fetching ingredients: {str(e)}")
         return jsonify({"error": f"Failed to fetch ingredients: {str(e)}"}), 500
 
+
 @app.route('/get_menu_items', methods=['POST'])
 def get_menu_items():
     ingredient_names = request.json.get('ingredient_names', '').strip().lower()