Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, send_from_directory, request, jsonify
+from flask import Flask, render_template, request, jsonify
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
@@ -22,17 +22,13 @@ def get_salesforce_connection():
         print(f"Error connecting to Salesforce: {e}")
         return None
 
-# Initialize Salesforce connection
+# Initialize Salesforce connection (can be moved to request scope in production)
 sf = get_salesforce_connection()
 
 @app.route('/')
 def index():
     return render_template('index.html')
 
-@app.route('/static/<path:filename>')
-def serve_static(filename):
-    return send_from_directory('static', filename)
-
 @app.route('/get_ingredients', methods=['POST'])
 def get_ingredients():
     global sf
@@ -43,15 +39,15 @@ def get_ingredients():
 
     dietary_preference = request.json.get('dietary_preference', '').lower()
     
-    # SOQL query based on dietary preference
-    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':
-        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"
+    # Validate and map dietary preference to SOQL condition
+    preference_map = {
+        'vegetarian': "Category__c = 'Veg'",
+        'non-vegetarian': "Category__c = 'Non-Veg'"
+    }
+    condition = preference_map.get(dietary_preference, "1=1")  # Default to all if invalid
 
     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', '')}