Update app.py

app.py

@@ -1,19 +1,15 @@
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, send_from_directory, request, jsonify
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
 import logging
 
-# Load environment variables from .env file
-load_dotenv()
+logging.basicConfig(level=logging.DEBUG)
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+load_dotenv()
 
 app = Flask(__name__, template_folder='templates', static_folder='static')
 
-# Function to get Salesforce connection
 def get_salesforce_connection():
     try:
         sf = Salesforce(
@@ -22,55 +18,82 @@ 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:
-        logger.error(f"Error connecting to Salesforce: {e}")
+        print(f"Error connecting to Salesforce: {e}")
         return None
 
-# Initialize Salesforce connection
 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
-    if not sf:
-        sf = get_salesforce_connection()
-        if not sf:
-            return jsonify({"error": "Failed to connect to Salesforce"}), 500
+    dietary_preference = request.json.get('dietary_preference', '').strip().lower()
+    logging.debug(f"Received dietary preference: {dietary_preference}")
 
-    dietary_preference = request.json.get('dietary_preference', '').lower()
-    
-    # Map dietary preference to SOQL condition
-    preference_map = {
-        'vegetarian': "Category__c = 'Veg'",
-        'non-vegetarian': "Category__c = 'Non-Veg'",
-        'both': None  # No condition to fetch all records
-    }
-    condition = preference_map.get(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:
+        logging.debug("Invalid dietary preference received.")
+        return jsonify({"error": "Invalid dietary preference."}), 400
 
     try:
-        # Construct the base query
-        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c"
-        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', '')}
             for record in result['records'] if 'Name' in record
         ]
-        logger.info(f"Fetched {len(ingredients)} ingredients: {ingredients}")
+        logging.debug(f"Fetched {len(ingredients)} ingredients.")
         return jsonify({"ingredients": ingredients})
     except Exception as e:
-        logger.error(f"Failed to fetch ingredients: {str(e)}")
+        logging.error(f"Error while fetching ingredients: {str(e)}")
         return jsonify({"error": f"Failed to fetch ingredients: {str(e)}"}), 500
 
+@app.route('/get_menu_items', methods=['POST'])
+def get_menu_items():
+    category = request.json.get('category', '').strip().lower()
+    logging.debug(f"Received category: {category}")
+
+    if category == 'fish':
+        logging.debug("Fetching fish-based menu items...")
+        soql = "SELECT Item_Name__c, Image_URL__c FROM Menu_Item__c WHERE Category__c = 'Fish' LIMIT 200"
+    else:
+        logging.debug("Invalid category received.")
+        return jsonify({"error": "Invalid category."}), 400
+
+    try:
+        result = sf.query(soql)
+        menu_items = [
+            {"name": record['Item_Name__c'], "image_url": record.get('Image_URL__c', '')}
+            for record in result['records'] if 'Item_Name__c' in record
+        ]
+        logging.debug(f"Fetched {len(menu_items)} menu items.")
+        return jsonify({"menu_items": menu_items})
+    except Exception as e:
+        logging.error(f"Error while fetching menu items: {str(e)}")
+        return jsonify({"error": f"Failed to fetch menu items: {str(e)}"}), 500
+
+@app.route('/submit_ingredients', methods=['POST'])
+def submit_ingredients():
+    data = request.json
+    ingredients = data.get('ingredients', [])
+    
+    if not ingredients:
+        return jsonify({'error': 'No ingredients selected'}), 400
+    
+    logging.debug(f"Ingredients submitted: {ingredients}")
+    return jsonify({'success': True})
+
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)