Update app.py

app.py

@@ -1,57 +1,228 @@
-from flask import Flask, request, jsonify
-from transformers import pipeline, set_seed
+from flask import Flask, render_template, request, jsonify
+from simple_salesforce import Salesforce
+from dotenv import load_dotenv
+import os
+import logging
 
-app = Flask(__name__)
+# Load environment variables
+load_dotenv()
 
-# Initialize the Hugging Face text-generation pipeline (DialoGPT model)
-chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
-set_seed(42)
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-# Example restaurant data
-menu = {
-    'vegan': ['Vegan Burger', 'Tofu Stir Fry', 'Veggie Pizza'],
-    'vegetarian': ['Cheese Pizza', 'Pasta Alfredo', 'Caprese Salad'],
-    'non_vegetarian': ['Chicken Curry', 'Grilled Salmon', 'Steak'],
-    'spicy': ['Spicy Tacos', 'Chili Chicken', 'Spicy Wings'],
-}
+app = Flask(__name__, template_folder='templates', static_folder='static')
+
+# Salesforce connection function
+def get_salesforce_connection():
+    try:
+        sf = Salesforce(
+            username=os.getenv('SFDC_USERNAME'),
+            password=os.getenv('SFDC_PASSWORD'),
+            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}")
+        return None
+
+# Initialize Salesforce connection
+sf = get_salesforce_connection()
 
 @app.route('/')
-def home():
-    return "Welcome to the Restaurant Chatbot API!"
-
-@app.route('/chat', methods=['POST'])
-def chat():
-    # Get user input from the frontend
-    user_input = request.json.get('user_input', '')
-
-    # Add user input to the conversation history (we'll use a simple list to store history)
-    conversation_history = user_input
-
-    # Generate a response from the chatbot model using text generation
-    response = chatbot(conversation_history, max_length=100, num_return_sequences=1, pad_token_id=50256)
-
-    # Get the chatbot's response text
-    chatbot_reply = response[0]['generated_text']
-
-    # Match user preference to the menu (based on keywords in the input)
-    user_preference = None
-    if 'vegan' in user_input.lower():
-        user_preference = 'vegan'
-    elif 'vegetarian' in user_input.lower():
-        user_preference = 'vegetarian'
-    elif 'non-vegetarian' in user_input.lower():
-        user_preference = 'non_vegetarian'
-    elif 'spicy' in user_input.lower():
-        user_preference = 'spicy'
-
-    # Get menu suggestions based on user preference
-    menu_suggestion = menu.get(user_preference, [])
-
-    # Return the chatbot response and menu suggestion
-    return jsonify({
-        'bot_response': chatbot_reply,
-        'menu_suggestion': menu_suggestion
-    })
+def index():
+    return render_template('index.html')
+
+# Fetch items from Sector_Detail__c for dietary preferences
+@app.route('/get_menu_items', methods=['POST'])
+def get_menu_items():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            logger.error("Salesforce connection failed after retry")
+            return jsonify({"error": "Unable to connect to Salesforce"}), 500
+
+    dietary_preference = request.json.get('dietary_preference', 'both').lower()
+    try:
+        soql = "SELECT Name, Image_URL__c, Category__c, Description__c FROM Sector_Detail__c"
+        if dietary_preference == 'vegetarian':
+            soql += " WHERE Category__c = 'Veg'"
+        elif dietary_preference == 'non-vegetarian':
+            soql += " WHERE Category__c = 'Non-Veg'"
+        soql += " LIMIT 200"
+        
+        logger.info(f"Executing SOQL query: {soql}")
+        result = sf.query(soql)
+        items = [
+            {
+                "name": record['Name'],
+                "image_url": record.get('Image_URL__c', ''),
+                "category": record.get('Category__c', ''),
+                "description": record.get('Description__c', 'No description available')
+            }
+            for record in result['records'] if 'Name' in record
+        ]
+        logger.info(f"Fetched {len(items)} items from Sector_Detail__c for {dietary_preference}")
+        return jsonify({"menu_items": items})
+    except Exception as e:
+        logger.error(f"Failed to fetch items from Sector_Detail__c: {str(e)}")
+        return jsonify({"error": f"Failed to fetch items from Salesforce: {str(e)}"}), 500
+
+# Fetch suggestions from Ingredient_Object__c
+@app.route('/suggest_items', methods=['POST'])
+def suggest_items():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Unable to connect to Salesforce"}), 500
+
+    search_term = request.json.get('search_term', '').strip()
+    if not search_term:
+        return jsonify({"error": "Search term is required"}), 400
+
+    try:
+        soql = f"SELECT Ingredient_Name__c, Image_URL__c FROM Ingredient_Object__c WHERE Ingredient_Name__c LIKE '%{search_term}%' LIMIT 10"
+        logger.info(f"Executing SOQL query: {soql}")
+        result = sf.query(soql)
+        suggestions = [
+            {"name": record['Ingredient_Name__c'], "image_url": record.get('Image_URL__c', '')}
+            for record in result['records'] if 'Ingredient_Name__c' in record
+        ]
+        logger.info(f"Fetched {len(suggestions)} suggestions for '{search_term}'")
+        return jsonify({"suggestions": suggestions})
+    except Exception as e:
+        logger.error(f"Failed to fetch suggestions: {str(e)}")
+        return jsonify({"error": "Failed to fetch suggestions from Salesforce"}), 500
+
+# Fetch item details from Ingredient_Object__c
+@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": "Unable 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:
+        soql = f"SELECT Ingredient_Name__c, Image_URL__c FROM Ingredient_Object__c WHERE Ingredient_Name__c LIKE '%{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 matching '{item_name}' in Ingredient_Object__c"}), 404
+
+        record = result['records'][0]
+        item_details = {
+            "name": record.get('Ingredient_Name__c', ''),
+            "image_url": record.get('Image_URL__c', '')
+        }
+        logger.info(f"Fetched details for '{item_name}'")
+        return jsonify({"item_details": item_details})
+    except Exception as e:
+        logger.error(f"Failed to fetch item details: {str(e)}")
+        return jsonify({"error": "Failed to fetch item details from Salesforce"}), 500
+
+# Fetch item details from Sector_Detail__c by name
+@app.route('/get_item_details_by_name', methods=['POST'])
+def get_item_details_by_name():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Unable 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:
+        soql = f"SELECT Name, Category__c, Description__c FROM Sector_Detail__c WHERE Name LIKE '%{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 matching '{item_name}' in Sector_Detail__c"}), 404
+
+        record = result['records'][0]
+        item_details = {
+            "name": record.get('Name', ''),
+            "category": record.get('Category__c', 'Unknown'),
+            "description": record.get('Description__c', 'No description available')
+        }
+        logger.info(f"Fetched details for '{item_name}' from Sector_Detail__c")
+        return jsonify({"item_details": item_details})
+    except Exception as e:
+        logger.error(f"Failed to fetch item details from Sector_Detail__c: {str(e)}")
+        return jsonify({"error": f"Failed to fetch item details from Salesforce: {str(e)}"}), 500
+
+# Fetch item details (Name, Image_URL, Category, Description) from Sector_Detail__c by name
+@app.route('/get_sector_item_details', methods=['POST'])
+def get_sector_item_details():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Unable 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:
+        soql = f"SELECT Name, Image_URL__c, Category__c, Description__c FROM Sector_Detail__c WHERE Name LIKE '%{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 matching '{item_name}' in Sector_Detail__c"}), 404
+
+        record = result['records'][0]
+        item_details = {
+            "name": record.get('Name', ''),
+            "image_url": record.get('Image_URL__c', 'https://via.placeholder.com/30'),
+            "category": record.get('Category__c', 'Unknown'),
+            "description": record.get('Description__c', 'No description available')
+        }
+        logger.info(f"Fetched details for '{item_name}' from Sector_Detail__c")
+        return jsonify({"item_details": item_details})
+    except Exception as e:
+        logger.error(f"Failed to fetch item details from Sector_Detail__c: {str(e)}")
+        return jsonify({"error": f"Failed to fetch item details from Salesforce: {str(e)}"}), 500
+
+# Submit selected items to Ingredient_Object__c
+@app.route('/submit_items', methods=['POST'])
+def submit_items():
+    global sf
+    if not sf:
+        sf = get_salesforce_connection()
+        if not sf:
+            return jsonify({"error": "Unable to connect to Salesforce"}), 500
+
+    items = request.json.get('items', [])
+    if not items:
+        return jsonify({"error": "No items to submit"}), 400
+
+    try:
+        for item in items:
+            logger.info(f"Submitting item: {item}")
+            sf.Ingredient_Object__c.create({
+                'Ingredient_Name__c': item['name'],
+                'Category__c': item.get('category', 'Unknown'),  # Add Category__c
+                'Description__c': item.get('description', 'No description available')  # Ensure description is included
+            })
+            logger.info(f"Successfully submitted item to Ingredient_Object__c: {item['name']}")
+        return jsonify({"success": f"Successfully submitted {len(items)} items to Salesforce"})
+    except Exception as e:
+        logger.error(f"Failed to submit items: {str(e)}")
+        return jsonify({"error": f"Failed to submit items to Salesforce: {str(e)}"}), 500
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(debug=True, host='0.0.0.0', port=7860)