Update app.py

app.py

@@ -2,16 +2,14 @@ from flask import Flask, render_template, send_from_directory, request, jsonify
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
-import openai
+import logging
 
-openai.api_key = "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"
+logging.basicConfig(level=logging.DEBUG)
 
-# Load environment variables from .env file
 load_dotenv()
 
 app = Flask(__name__, template_folder='templates', static_folder='static')
 
-# Function to get Salesforce connection
 def get_salesforce_connection():
     try:
         sf = Salesforce(
@@ -25,7 +23,6 @@ def get_salesforce_connection():
         print(f"Error connecting to Salesforce: {e}")
         return None
 
-# Initialize Salesforce connection
 sf = get_salesforce_connection()
 
 @app.route('/')
@@ -39,19 +36,16 @@ def serve_static(filename):
 @app.route('/get_ingredients', methods=['POST'])
 def get_ingredients():
     dietary_preference = request.json.get('dietary_preference', '').strip().lower()
-
-    # Debugging: print the received dietary preference
-    print(f"Received dietary preference: {dietary_preference}")
+    logging.debug(f"Received dietary preference: {dietary_preference}")
 
     if dietary_preference == 'veg':
-        print("Fetching ingredients for Vegetarian...")  # Debug when fetching vegetarian ingredients
-        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c IN ('Veg', 'Both') LIMIT 200"
+        logging.debug("Fetching ingredients for Vegetarian...")
+        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Veg' LIMIT 200"
     elif dietary_preference == 'non-vegetarian':
-        print("Fetching ingredients for Non-Vegetarian...")  # Debug when fetching non-vegetarian ingredients
-        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c IN ('Non-Veg', 'Both') LIMIT 200"
-
+        logging.debug("Fetching ingredients for Non-Vegetarian...")
+        soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c = 'Non-Veg' LIMIT 200"
     else:
-        print("Invalid dietary preference received.")  # Debug for invalid dietary preference
+        logging.debug("Invalid dietary preference received.")
         return jsonify({"error": "Invalid dietary preference."}), 400
 
     try:
@@ -60,11 +54,36 @@ def get_ingredients():
             {"name": record['Name'], "image_url": record.get('Image_URL__c', '')}
             for record in result['records'] if 'Name' in record
         ]
-        print(f"Fetched {len(ingredients)} ingredients.")  # Debug: print how many ingredients were fetched
+        logging.debug(f"Fetched {len(ingredients)} ingredients.")
         return jsonify({"ingredients": ingredients})
     except Exception as e:
-        print(f"Error while fetching ingredients: {str(e)}")  # Debug error
+        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
@@ -73,42 +92,8 @@ def submit_ingredients():
     if not ingredients:
         return jsonify({'error': 'No ingredients selected'}), 400
     
-    # You can now process the selected ingredients, store them in a database, etc.
-    print(f"Ingredients submitted: {ingredients}")
-
+    logging.debug(f"Ingredients submitted: {ingredients}")
     return jsonify({'success': True})
 
-import logging
-
-# Add this to the beginning of your Flask app file
-logging.basicConfig(level=logging.DEBUG)  # Enable debug-level logging
-
-
-
-@app.route('/get_dish_suggestions', methods=['POST'])
-def get_dish_suggestions():
-    selected_ingredients = request.json.get('ingredients', [])
-    if not selected_ingredients:
-        return jsonify({'error': 'No ingredients selected'}), 400
-
-    ingredients_text = ', '.join([ingredient['name'] for ingredient in selected_ingredients])
-    prompt = f"Suggest a recipe based on the following ingredients: {ingredients_text}"
-
-    try:
-        # Use the correct API structure for openai>=1.0.0
-        response = openai.Completion.create(
-            model="gpt-3.5-turbo",
-            prompt=prompt,
-            max_tokens=150,
-            temperature=0.7
-        )
-        dish_suggestions = response['choices'][0]['text'].strip()
-        return jsonify({"suggestions": dish_suggestions})
-
-    except Exception as e:
-        logging.error(f"Error in /get_dish_suggestions: {e}")
-        return jsonify({"error": f"Unexpected error: {str(e)}"}), 500
-
-
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)