app.py

from flask import Flask, render_template, request, jsonify, send_from_directory
from simple_salesforce import Salesforce
from dotenv import load_dotenv
import os
import logging
import uuid
from datetime import datetime

# Load environment variables
load_dotenv()

# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

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: {str(e)}")
        return None

# Initialize Salesforce connection
sf = get_salesforce_connection()

@app.route('/')
def index():
    return render_template('chef-bot.html')

@app.route('/static/<path:filename>')
def static_files(filename):
    return send_from_directory(app.static_folder, 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": "Unable to connect to Salesforce"}), 500

    try:
        soql = "SELECT Name, Category__c FROM Sector_Detail__c WHERE Category__c = 'Both'"
        soql += " LIMIT 200"
        logger.info(f"Executing SOQL query for Sector_Detail__c: {soql}")
        result = sf.query(soql)
        ingredients = [
            {
                "name": record['Name'],
                "category": record.get('Category__c', '')
            }
            for record in result['records'] if 'Name' in record
        ]
        logger.info(f"Fetched {len(ingredients)} ingredients from Sector_Detail__c")
        return jsonify({"ingredients": ingredients})
    except Exception as e:
        logger.error(f"Failed to fetch ingredients: {str(e)}")
        return jsonify({"error": f"Failed to fetch ingredients from Salesforce: {str(e)}"}), 500

@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

    data = request.json
    dietary_preference = data.get('dietary_preference', 'both').lower()
    search_term = data.get('search_term', '').strip()
    source = data.get('source', 'Menu_Item__c')

    try:
        items = []
        if source == '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 search_term:
                soql_menu += f" WHERE Name LIKE '%{search_term}%'"
            elif 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
            ]
            items.extend(menu_items)

        logger.info(f"Fetched {len(items)} items from {source} for {dietary_preference or search_term}")
        return jsonify({"menu_items": items})
    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

@app.route('/get_related_menu_items', methods=['POST'])
def get_related_menu_items():
    global sf
    if not sf:
        sf = get_salesforce_connection()
        if not sf:
            return jsonify({"error": "Unable to connect to Salesforce"}), 500

    data = request.json
    ingredient_name = data.get('ingredient_name', '').strip()
    dietary_preference = data.get('dietary_preference', 'both').lower()

    if not ingredient_name:
        return jsonify({"error": "Ingredient name is required"}), 400

    try:
        soql = "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 Ingredientsinfo__c LIKE '%" + ingredient_name + "%'"
        if dietary_preference == 'vegetarian':
            soql += " AND Veg_NonVeg__c = 'Vegetarian'"
        elif dietary_preference == 'non-vegetarian':
            soql += " AND Veg_NonVeg__c = 'Non-Vegetarian'"
        soql += " LIMIT 200"
        logger.info(f"Executing SOQL query for related Menu_Item__c: {soql}")
        result = sf.query(soql)
        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['records'] if 'Name' in record
        ]
        logger.info(f"Fetched {len(menu_items)} related menu items for {ingredient_name}")
        return jsonify({"menu_items": menu_items})
    except Exception as e:
        logger.error(f"Failed to fetch related menu items: {str(e)}")
        return jsonify({"error": f"Failed to fetch related menu items from Salesforce: {str(e)}"}), 500

@app.route('/submit_ingredients', methods=['POST'])
def submit_ingredients():
    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', [])
    custom_order_name = request.json.get('custom_order_name', '')
    ingredients_info = request.json.get('ingredients_info', '')
    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:
            item_description = item['description']
            if item.get('ingredientsInfo'):
                item_description += f", additional info: {item['ingredientsInfo']}"
            sf.Ingredient_Object__c.create({
                'Ingredient_Name__c': ingredient_name,
                'Category__c': item.get('category', ''),
                'Description__c': item_description,
                'Image_URL__c': item.get('image_url', ''),
                'Quantity__c': item.get('quantity', 1),
                'Ingredientsinfo__c': ingredients_info or item.get('ingredientsInfo', '')
            })

        logger.info(f"Submitted {len(items)} items under {ingredient_name}")
        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: {str(e)}")
        return jsonify({"error": f"Failed to submit items: {str(e)}"}), 500

if __name__ == '__main__':
    app.run(debug=True, host='0.0.0.0', port=7860)