Update app.py

app.py

@@ -38,17 +38,19 @@ def get_ingredients():
     dietary_preference = request.json.get('dietary_preference', '').strip().lower()
     logging.debug(f"Received dietary preference: {dietary_preference}")
 
-    if dietary_preference == 'vegetable':
-        logging.debug("Fetching ingredients for Vegetables...")
-        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Veg' AND Subcategory__c = 'Vegetable' LIMIT 200"
-    elif dietary_preference in ['chicken', 'beef', 'lamb']:
-        logging.debug(f"Fetching ingredients for {dietary_preference}...")
-        soql = f"SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Non-Veg' AND Subcategory__c = '{dietary_preference.capitalize()}' LIMIT 200"
-    else:
+    # 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 not condition:
         logging.debug("Invalid dietary preference received.")
         return jsonify({"error": "Invalid dietary preference."}), 400
 
     try:
+        soql = f"SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE {condition} LIMIT 200"
         result = sf.query(soql)
         ingredients = [
             {"name": record['Name'], "image_url": record.get('Image_URL__c', '')}