Update app.py

app.py

@@ -43,17 +43,20 @@ def get_ingredients():
 
     dietary_preference = request.json.get('dietary_preference', '').lower()
     
-    # SOQL query based on dietary preference with corrected field name 'Name'
+    # SOQL query based on dietary preference
     if dietary_preference == 'veg':
-        soql = "SELECT Name FROM Sector_Detail__c WHERE Category__c IN ('Veg', 'Both') LIMIT 200"
+        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Veg' LIMIT 200"
     elif dietary_preference == 'non-vegetarian':
-        soql = "SELECT Name FROM Sector_Detail__c WHERE Category__c IN ('Non-Veg', 'Both') LIMIT 200"
+        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Non-Veg' LIMIT 200"
     else:
-        soql = "SELECT Name FROM Sector_Detail__c LIMIT 200"
+        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c LIMIT 200"
 
     try:
         result = sf.query(soql)
-        ingredients = [record['Name'] for record in result['records'] if 'Name' in record]
+        ingredients = [
+            {"name": record['Name'], "image_url": record.get('Image_URL__c', '')}
+            for record in result['records'] if 'Name' in record
+        ]
         return jsonify({"ingredients": ingredients})
     except Exception as e:
         return jsonify({"error": f"Failed to fetch ingredients: {str(e)}"}), 500