Update app.py

app.py

@@ -2,10 +2,15 @@ from flask import Flask, render_template, request, jsonify
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
+import logging
 
 # Load environment variables from .env file
 load_dotenv()
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 app = Flask(__name__, template_folder='templates', static_folder='static')
 
 # Function to get Salesforce connection
@@ -17,12 +22,13 @@ def get_salesforce_connection():
             security_token=os.getenv('SFDC_SECURITY_TOKEN'),
             domain=os.getenv('SFDC_DOMAIN', 'login')
         )
+        logger.info("Successfully connected to Salesforce")
         return sf
     except Exception as e:
-        print(f"Error connecting to Salesforce: {e}")
+        logger.error(f"Error connecting to Salesforce: {e}")
         return None
 
-# Initialize Salesforce connection (can be moved to request scope in production)
+# Initialize Salesforce connection
 sf = get_salesforce_connection()
 
 @app.route('/')
@@ -43,19 +49,27 @@ def get_ingredients():
     preference_map = {
         'vegetarian': "Category__c = 'Veg'",
         'non-vegetarian': "Category__c = 'Non-Veg'",
-        'both': "1=1"  # No category filter for "Both"
+        'both': None  # No condition to fetch all records, including "Both" category
     }
-    condition = preference_map.get(dietary_preference, "1=1")  # Default to all if invalid
+    condition = preference_map.get(dietary_preference)
 
     try:
-        soql = f"SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE {condition} LIMIT 200"
+        # Construct the base query
+        soql = "SELECT Name, Image_URL__c, Category__c FROM Sector_Detail__c"  # Added Category__c for debugging
+        if condition:
+            soql += f" WHERE {condition}"
+        soql += " LIMIT 200"
+
+        logger.info(f"Executing SOQL query: {soql}")
         result = sf.query(soql)
         ingredients = [
-            {"name": record['Name'], "image_url": record.get('Image_URL__c', '')}
+            {"name": record['Name'], "image_url": record.get('Image_URL__c', ''), "category": record.get('Category__c', 'Unknown')}
             for record in result['records'] if 'Name' in record
         ]
+        logger.info(f"Fetched {len(ingredients)} ingredients: {ingredients}")
         return jsonify({"ingredients": ingredients})
     except Exception as e:
+        logger.error(f"Failed to fetch ingredients: {str(e)}")
         return jsonify({"error": f"Failed to fetch ingredients: {str(e)}"}), 500
 
 if __name__ == '__main__':