Update app.py

app.py

@@ -3,6 +3,9 @@ from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
 import logging
+import uuid
+from datetime import datetime
+from openai import OpenAI
 
 # Load environment variables
 load_dotenv()
@@ -31,79 +34,190 @@ def get_salesforce_connection():
 # Initialize Salesforce connection
 sf = get_salesforce_connection()
 
+# Initialize OpenAI client
+openai_api_key = os.getenv('OPENAI_API_KEY')
+if not openai_api_key:
+    logger.error("OPENAI_API_KEY not found in .env. ChatGPT functionality disabled.")
+    client = None
+else:
+    try:
+        client = OpenAI(api_key=openai_api_key)
+        test_response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": "Test API key"}]
+        )
+        logger.info("OpenAI API key validated successfully")
+    except Exception as e:
+        logger.error(f"Invalid OpenAI API key or connection issue: {e}")
+        client = None
+
+# Conversation storage
+conversation_sessions = {}
+
+def is_restaurant_related(query):
+    keywords = ['use', 'uses', 'dish', 'recipe', 'ingredient', 'food', 'menu', 'vegetarian', 'non-vegetarian',
+                'rice', 'chicken', 'pasta', 'veggie', 'mutton', 'lamb', 'beef', 'pork', 'fish', 'seafood',
+                'ingredients', 'nutrition', 'nutritional', 'info', 'message', 'messages']
+    return bool(query.strip() and any(keyword in query.lower() for keyword in keywords)) or any(item in query.lower() for item in ['mutton', 'rice', 'chicken'])
+
+def get_chatgpt_response(prompt, session_id):
+    if not client:
+        logger.warning("No OpenAI client available, using mock ChatGPT response")
+        if any(item in prompt.lower() for item in ['mutton', 'rice', 'chicken']):
+            return f"{prompt.capitalize()} is great for curries or roasts in a restaurant!"
+        return "I can help with food! Try 'uses of mutton' or 'ingredients for rice'."
+
+    if session_id not in conversation_sessions:
+        conversation_sessions[session_id] = [
+            {"role": "system", "content": "You are a restaurant culinary assistant. Respond only to queries about ingredients, dishes, or recipes. Provide concise, friendly answers. For unclear queries, suggest: 'Try 'uses of mutton' or 'ingredients for rice'.'"}
+        ]
+
+    prompt = prompt.strip().lower()
+    logger.info(f"Processing prompt: {prompt}")
+    if not is_restaurant_related(prompt):
+        logger.info(f"Off-topic or empty query: {prompt}")
+        return "I can help with food! Try 'uses of mutton' or 'ingredients for rice'."
+
+    intent = "uses"
+    item = prompt
+    if "what is the use of" in prompt or "uses of" in prompt:
+        item = prompt.replace("what is the use of ", "").replace("uses of ", "").strip() or prompt
+        prompt = f"Provide the culinary uses of '{item}' for a restaurant. Be concise and friendly. If unknown, say 'Great for many dishes!'"
+    elif any(word in prompt for word in ['suggest', 'recipe', 'dish']):
+        intent = "suggest"
+        prompt = f"Suggest a restaurant dish based on: {prompt}. Include ingredients and be concise."
+    elif any(word in prompt for word in ['ingredients', 'ingredient']):
+        intent = "ingredients"
+        item = prompt.replace("ingredients for ", "").replace("ingredient of ", "").strip() or prompt
+        prompt = f"List common ingredients for a restaurant dish using '{item}'. Be concise and realistic."
+    elif any(word in prompt for word in ['nutrition', 'nutritional', 'info']):
+        intent = "nutrition"
+        item = prompt.replace("nutrition of ", "").replace("nutritional info for ", "").strip() or prompt
+        prompt = f"Provide a concise, approximate nutritional overview (calories, protein, fat, carbs) for a restaurant portion of '{item}'. Use general culinary knowledge and keep it friendly."
+    elif any(item in prompt for item in ['mutton', 'rice', 'chicken', 'pasta', 'veggie', 'lamb', 'beef', 'pork', 'fish', 'seafood']):
+        prompt = f"Provide the culinary uses of '{item}' for a restaurant. Be concise and friendly. If unknown, say 'Great for many dishes!'"
+    else:
+        logger.info(f"Unclear intent, defaulting to uses for: {prompt}")
+        prompt = f"Provide the culinary uses of '{prompt}' for a restaurant. Be concise and friendly. If unknown, say 'Great for many dishes!'"
+
+    conversation_sessions[session_id].append({"role": "user", "content": prompt})
+
+    try:
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=conversation_sessions[session_id],
+            max_tokens=150,
+            timeout=10
+        )
+        reply = response.choices[0].message.content.strip()
+        conversation_sessions[session_id].append({"role": "assistant", "content": reply})
+        logger.info(f"ChatGPT response for '{prompt}': {reply}")
+        return reply
+    except Exception as e:
+        logger.error(f"OpenAI API error: {e}")
+        return f"Sorry, API failed. Mock response: {prompt.capitalize()} is great for curries or roasts!"
+
 @app.route('/')
 def index():
     return render_template('index.html')
 
-# Fetch items from Sector_Detail__c and Menu_Item__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()
+    data = request.json
+    dietary_preference = data.get('dietary_preference', 'both').lower()
+    search_term = data.get('search_term', '').strip()
     try:
-        # Query Sector_Detail__c
-        soql_sector = "SELECT Name, Image_URL__c, Category__c, Description__c FROM Sector_Detail__c"
-        if dietary_preference == 'vegetarian':
-            soql_sector += " WHERE Category__c = 'Veg'"
-        elif dietary_preference == 'non-vegetarian':
-            soql_sector += " WHERE Category__c = 'Non-Veg'"
-        soql_sector += " LIMIT 200"
-        
-        logger.info(f"Executing SOQL query for Sector_Detail__c: {soql_sector}")
-        result_sector = sf.query(soql_sector)
-        sector_items = [
-            {
-                "name": record['Name'],
-                "image_url": record.get('Image_URL__c', ''),
-                "category": record.get('Category__c', ''),
-                "description": record.get('Description__c', 'No description available'),
-                "source": "Sector_Detail__c"
-            }
-            for record in result_sector['records'] if 'Name' in record
-        ]
+        items = []
+        if not search_term:
+            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"
+            result = sf.query(soql)
+            items = [
+                {
+                    "name": r['Name'],
+                    "image_url": r.get('Image_URL__c', ''),
+                    "category": r.get('Category__c', ''),
+                    "description": r.get('Description__c', 'No description'),
+                    "source": "Sector_Detail__c"
+                }
+                for r in result['records'] if 'Name' in r
+            ]
+        else:
+            soql_menu = "SELECT Name, Description__c, Image1__c, Ingredientsinfo__c, NutritionalInfo__c, Price__c, Sector__c, Spice_Levels__c, Veg_NonVeg__c, Category__c, Dynamic_Dish__c FROM Menu_Item__c"
+            soql_menu += f" WHERE Name LIKE '%{search_term}%' OR Ingredientsinfo__c LIKE '%{search_term}%'"
+            if dietary_preference == 'vegetarian':
+                soql_menu += " AND Veg_NonVeg__c = 'Vegetarian'"
+            elif dietary_preference == 'non-vegetarian':
+                soql_menu += " AND Veg_NonVeg__c = 'Non-Vegetarian'"
+            soql_menu += " LIMIT 200"
+            logger.info(f"Executing SOQL query for Menu_Item__c: {soql_menu}")
+            result_menu = sf.query(soql_menu)
+            menu_items = [
+                {
+                    "name": record['Name'],
+                    "description": record.get('Description__c', 'No description available'),
+                    "image_url": record.get('Image1__c', ''),
+                    "ingredients": record.get('Ingredientsinfo__c', ''),
+                    "nutritional_info": record.get('NutritionalInfo__c', ''),
+                    "price": record.get('Price__c', 0.0),
+                    "sector": record.get('Sector__c', ''),
+                    "spice_levels": record.get('Spice_Levels__c', ''),
+                    "veg_nonveg": record.get('Veg_NonVeg__c', ''),
+                    "category": record.get('Category__c', ''),
+                    "dynamic_dish": record.get('Dynamic_Dish__c', False),
+                    "source": "Menu_Item__c"
+                }
+                for record in result_menu['records'] if 'Name' in record
+            ]
+            items.extend(menu_items)
+        return jsonify({"menu_items": items})
+    except Exception as e:
+        logger.error(f"Failed to fetch items: {e}")
+        return jsonify({"error": f"Failed to fetch items: {e}"}), 500
 
-        # Query Menu_Item__c
-        soql_menu = "SELECT Name, Description__c, Image1__c, Ingredientsinfo__c, NutritionalInfo__c, Price__c, Sector__c, Spice_Levels__c, Veg_NonVeg__c, Category__c, Dynamic_Dish__c FROM Menu_Item__c"
-        if dietary_preference == 'vegetarian':
-            soql_menu += " WHERE Veg_NonVeg__c = 'Vegetarian'"
-        elif dietary_preference == 'non-vegetarian':
-            soql_menu += " WHERE Veg_NonVeg__c = 'Non-Vegetarian'"
-        soql_menu += " LIMIT 200"
-        
-        logger.info(f"Executing SOQL query for Menu_Item__c: {soql_menu}")
-        result_menu = sf.query(soql_menu)
-        menu_items = [
-            {
-                "name": record['Name'],
-                "description": record.get('Description__c', 'No description available'),
-                "image_url": record.get('Image1__c', ''),
-                "ingredients": record.get('Ingredientsinfo__c', ''),
-                "nutritional_info": record.get('NutritionalInfo__c', ''),
-                "price": record.get('Price__c', 0.0),
-                "sector": record.get('Sector__c', ''),
-                "spice_levels": record.get('Spice_Levels__c', ''),
-                "veg_nonveg": record.get('Veg_NonVeg__c', ''),
-                "category": record.get('Category__c', ''),
-                "dynamic_dish": record.get('Dynamic_Dish__c', False),
-                "source": "Menu_Item__c"
-            }
-            for record in result_menu['records'] if 'Name' in record
-        ]
+@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
 
-        items = sector_items + menu_items
-        logger.info(f"Fetched {len(items)} items (Sector_Detail__c: {len(sector_items)}, Menu_Item__c: {len(menu_items)}) for {dietary_preference}")
-        return jsonify({"menu_items": items})
+    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', ''),
+            "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 items: {str(e)}")
-        return jsonify({"error": f"Failed to fetch items from Salesforce: {str(e)}"}), 500
+        logger.error(f"Failed to fetch item details from Sector_Detail__c: {e}")
+        return jsonify({"error": f"Failed to fetch item details: {e}"}), 500
 
-# Fetch suggestions from Ingredient_Object__c
 @app.route('/suggest_items', methods=['POST'])
 def suggest_items():
     global sf
@@ -127,12 +241,11 @@ def suggest_items():
         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
+        logger.error(f"Failed to fetch suggestions: {e}")
+        return jsonify({"error": f"Failed to fetch suggestions: {e}"}), 500
 
-# Fetch item details from Menu_Item__c by name
-@app.route('/get_menu_item_details', methods=['POST'])
-def get_menu_item_details():
+@app.route('/get_item_details', methods=['POST'])
+def get_item_details():
     global sf
     if not sf:
         sf = get_salesforce_connection()
@@ -144,70 +257,24 @@ def get_menu_item_details():
         return jsonify({"error": "Item name is required"}), 400
 
     try:
-        soql = f"SELECT Name, Description__c, Image1__c, Ingredientsinfo__c, NutritionalInfo__c, Price__c, Sector__c, Spice_Levels__c, Veg_NonVeg__c, Category__c, Dynamic_Dish__c FROM Menu_Item__c WHERE Name LIKE '%{item_name}%' LIMIT 1"
+        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 Menu_Item__c"}), 404
+            return jsonify({"error": f"No item found matching '{item_name}' in Ingredient_Object__c"}), 404
 
         record = result['records'][0]
         item_details = {
-            "name": record.get('Name', ''),
-            "description": record.get('Description__c', 'No description available'),
-            "image_url": record.get('Image1__c', 'https://via.placeholder.com/30'),
-            "ingredients": record.get('Ingredientsinfo__c', ''),
-            "nutritional_info": record.get('NutritionalInfo__c', ''),
-            "price": record.get('Price__c', 0.0),
-            "sector": record.get('Sector__c', ''),
-            "spice_levels": record.get('Spice_Levels__c', ''),
-            "veg_nonveg": record.get('Veg_NonVeg__c', ''),
-            "category": record.get('Category__c', ''),
-            "dynamic_dish": record.get('Dynamic_Dish__c', False),
-            "source": "Menu_Item__c"
-        }
-        logger.info(f"Fetched details for '{item_name}' from Menu_Item__c")
-        return jsonify({"item_details": item_details})
-    except Exception as e:
-        logger.error(f"Failed to fetch item details from Menu_Item__c: {str(e)}")
-        return jsonify({"error": f"Failed to fetch item details from Salesforce: {str(e)}"}), 500
-
-# Fetch item details 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'),
-            "source": "Sector_Detail__c"
+            "name": record.get('Ingredient_Name__c', ''),
+            "image_url": record.get('Image_URL__c', '')
         }
-        logger.info(f"Fetched details for '{item_name}' from Sector_Detail__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 from Sector_Detail__c: {str(e)}")
-        return jsonify({"error": f"Failed to fetch item details from Salesforce: {str(e)}"}), 500
+        logger.error(f"Failed to fetch item details: {e}")
+        return jsonify({"error": f"Failed to fetch item details: {e}"}), 500
 
-# Submit selected items to Ingredient_Object__c
 @app.route('/submit_items', methods=['POST'])
 def submit_items():
     global sf
@@ -217,22 +284,75 @@ def submit_items():
             return jsonify({"error": "Unable to connect to Salesforce"}), 500
 
     items = request.json.get('items', [])
+    custom_order_name = request.json.get('custom_order_name', '')
     if not items:
         return jsonify({"error": "No items to submit"}), 400
 
     try:
+        ingredient_name = custom_order_name or f"Order_{datetime.now().strftime('%Y%m%d')}_{uuid.uuid4().hex[:8]}"
+        item_names = ', '.join(item['name'] for item in items)
+        description = f"Contains: {item_names}"
+
         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'),
-                'Description__c': item.get('description', 'No description available')
+                'Ingredient_Name__c': ingredient_name,
+                'Category__c': item.get('category', ''),
+                'Description__c': f"{item.get('description', 'No description')} - {description}",
+                'Image_URL__c': item.get('image_url', ''),
+                'Quantity__c': item.get('quantity', 1)
             })
-            logger.info(f"Successfully submitted item to Ingredient_Object__c: {item['name']}")
-        return jsonify({"success": f"Successfully submitted {len(items)} items to Salesforce"})
+
+        return jsonify({"success": f"Submitted {len(items)} items under {ingredient_name}", "ingredient_name": ingredient_name})
+    except Exception as e:
+        logger.error(f"Failed to submit items: {e}")
+        return jsonify({"error": f"Failed to submit items: {e}"}), 500
+
+@app.route('/get_item_info', methods=['POST'])
+def get_item_info():
+    global sf, client
+    item_name = request.json.get('item_name', '').strip()
+    if not item_name or not client:
+        return jsonify({"error": "Item name required or OpenAI unavailable"}), 400
+
+    try:
+        soql = f"SELECT Name, Description__c FROM Sector_Detail__c WHERE Name LIKE '%{item_name}%' LIMIT 1"
+        result = sf.query(soql)
+        if result['totalSize'] == 0:
+            return jsonify({"error": f"No item found matching '{item_name}'"}), 404
+
+        record = result['records'][0]
+        base_info = record.get('Description__c', 'No description available')
+
+        prompt = f"Based on general culinary knowledge, provide a concise list of common ingredients and an approximate nutritional overview (calories, protein, fat, carbs) for a restaurant portion of '{item_name}'. Use realistic values and keep it friendly."
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=150,
+            timeout=10
+        )
+        ai_response = response.choices[0].message.content.strip()
+
+        item_info = {
+            "name": record['Name'],
+            "description": base_info,
+            "details": ai_response
+        }
+        logger.info(f"Generated info for '{item_name}'")
+        return jsonify({"item_info": item_info})
     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
+        logger.error(f"Failed to get item info: {e}")
+        return jsonify({"error": f"Failed to get item info: {e}"}), 500
+
+@app.route('/chat', methods=['POST'])
+def chat():
+    user_message = request.json.get('message', '').strip()
+    session_id = request.json.get('session_id', 'default')
+    if not user_message:
+        return jsonify({"error": "No message provided"}), 400
+
+    response = get_chatgpt_response(user_message, session_id)
+    logger.info(f"Chat response sent: {response}")
+    return jsonify({"response": response, "search_term": user_message})
 
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)