Update app.py

app.py

@@ -41,15 +41,14 @@ def get_ingredients():
         if not sf:
             return jsonify({"error": "Failed to connect to Salesforce"}), 500
 
-    dietary_preference = request.json.get('dietary_preference', '').lower()
-    
-    # SOQL query based on dietary preference
+    dietary_preference = request.json.get('dietary_preference', '').strip().lower()
+
     if dietary_preference == 'veg':
         soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Veg' LIMIT 200"
-    elif dietary_preference == 'non-vegetarian':
+    elif dietary_preference in ['non-vegetarian', 'non veg', 'nonveg']:
         soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Non-Veg' LIMIT 200"
     else:
-        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c LIMIT 200"
+        return jsonify({"error": "Invalid dietary preference."}), 400
 
     try:
         result = sf.query(soql)