Update app.py

app.py

@@ -35,42 +35,91 @@ sf = get_salesforce_connection()
 def index():
     return render_template('index.html')
 
-@app.route('/get_ingredients', methods=['POST'])
-def get_ingredients():
+# Fetch all menu items (names and images) from Salesforce
+@app.route('/get_menu_items', methods=['GET'])
+def get_menu_items():
     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', '').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)
+    try:
+        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c LIMIT 200"
+        logger.info(f"Executing SOQL query: {soql}")
+        result = sf.query(soql)
+        menu_items = [
+            {"name": record['Name'], "image_url": record.get('Image_URL__c', '')}
+            for record in result['records'] if 'Name' in record
+        ]
+        logger.info(f"Fetched {len(menu_items)} menu items: {menu_items}")
+        return jsonify({"menu_items": menu_items})
+    except Exception as e:
+        logger.error(f"Failed to fetch menu items: {str(e)}")
+        return jsonify({"error": f"Failed to fetch menu items: {str(e)}"}), 500
+
+# Fetch details (description and ingredients) for a specific menu item
+@app.route('/get_item_details', methods=['POST'])
+def get_item_details():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Failed to connect to Salesforce"}), 500
+
+    item_name = request.json.get('item_name', '').strip()
+    if not item_name:
+        return jsonify({"error": "Item name is required"}), 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"
+        # Query Salesforce for the specific item
+        soql = f"SELECT Name, Description__c, Ingredients__c, Image_URL__c FROM Sector_Detail__c WHERE Name = '{item_name}' LIMIT 1"
+        logger.info(f"Executing SOQL query: {soql}")
+        result = sf.query(soql)
 
+        if result['totalSize'] == 0:
+            return jsonify({"error": f"No item found with name '{item_name}'"}), 404
+
+        record = result['records'][0]
+        item_details = {
+            "name": record.get('Name', ''),
+            "description": record.get('Description__c', 'No description available'),
+            "ingredients": record.get('Ingredients__c', 'No ingredients listed'),
+            "image_url": record.get('Image_URL__c', '')
+        }
+        logger.info(f"Fetched details for {item_name}: {item_details}")
+        return jsonify({"item_details": item_details})
+    except Exception as e:
+        logger.error(f"Failed to fetch item details: {str(e)}")
+        return jsonify({"error": f"Failed to fetch item details: {str(e)}"}), 500
+
+# Suggest suitable items based on user input (e.g., "chicken")
+@app.route('/suggest_items', methods=['POST'])
+def suggest_items():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Failed to connect to Salesforce"}), 500
+
+    search_term = request.json.get('search_term', '').strip().lower()
+    if not search_term:
+        return jsonify({"error": "Search term is required"}), 400
+
+    try:
+        # Search for items where Name or Ingredients__c contains the search term
+        soql = f"SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Name LIKE '%{search_term}%' OR Ingredients__c LIKE '%{search_term}%' LIMIT 10"
         logger.info(f"Executing SOQL query: {soql}")
         result = sf.query(soql)
-        ingredients = [
+        suggestions = [
             {"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}")
-        return jsonify({"ingredients": ingredients})
+        logger.info(f"Fetched {len(suggestions)} suggestions for '{search_term}': {suggestions}")
+        return jsonify({"suggestions": suggestions})
     except Exception as e:
-        logger.error(f"Failed to fetch ingredients: {str(e)}")
-        return jsonify({"error": f"Failed to fetch ingredients: {str(e)}"}), 500
+        logger.error(f"Failed to fetch suggestions: {str(e)}")
+        return jsonify({"error": f"Failed to fetch suggestions: {str(e)}"}), 500
 
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)