app.py

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

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

# Initialize Salesforce connection
sf = get_salesforce_connection()

@app.route('/')
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 and Menu_Item__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:
        # Query Ingredient_Object__c
        soql_ingredient = f"SELECT Ingredient_Name__c, Image_URL__c FROM Ingredient_Object__c WHERE Ingredient_Name__c LIKE '%{search_term}%' LIMIT 5"
        logger.info(f"Executing SOQL query for Ingredient_Object__c: {soql_ingredient}")
        result_ingredient = sf.query(soql_ingredient)
        suggestions = [
            {"name": record['Ingredient_Name__c'], "image_url": record.get('Image_URL__c', ''), "source": "Ingredient_Object__c"}
            for record in result_ingredient['records'] if 'Ingredient_Name__c' in record
        ]

        # Query Menu_Item__c
        soql_menu = f"SELECT Name, Image1__c, Veg_NonVeg__c FROM Menu_Item__c WHERE Name LIKE '%{search_term}%' LIMIT 5"
        logger.info(f"Executing SOQL query for Menu_Item__c: {soql_menu}")
        result_menu = sf.query(soql_menu)
        suggestions += [
            {"name": record['Name'], "image_url": record.get('Image1__c', ''), "source": "Menu_Item__c", "veg_nonveg": record.get('Veg_NonVeg__c', '')}
            for record in result_menu['records'] if 'Name' 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 Menu_Item__c by name
@app.route('/get_menu_item_details', methods=['POST'])
def get_menu_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, 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
        """
        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

        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/100'),
            "ingredients": record.get('Ingredientsinfo__c', 'No ingredients info'),
            "nutritional_info": record.get('NutritionalInfo__c', 'No nutritional info'),
            "price": record.get('Price__c', 0.0),
            "sector": record.get('Sector__c', 'Unknown'),
            "spice_levels": record.get('Spice_Levels__c', 'Not specified'),
            "veg_nonveg": record.get('Veg_NonVeg__c', 'Unknown'),
            "category": record.get('Category__c', 'Unknown'),
            "dynamic_dish": record.get('Dynamic_Dish__c', False)
        }
        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

# Submit selected items to Ingredient_Object__c or Menu_Item__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}")
            # Check if item is from Menu_Item__c or Ingredient_Object__c
            if item.get('source') == 'Menu_Item__c':
                sf.Menu_Item__c.create({
                    'Name': item['name'],
                    'Description__c': item.get('description', 'No description available'),
                    'Image1__c': item.get('image_url', ''),
                    'Ingredientsinfo__c': item.get('ingredients', 'No ingredients info'),
                    'NutritionalInfo__c': item.get('nutritional_info', 'No nutritional info'),
                    'Price__c': item.get('price', 0.0),
                    'Sector__c': item.get('sector', 'Unknown'),
                    'Spice_Levels__c': item.get('spice_levels', 'Not specified'),
                    'Veg_NonVeg__c': item.get('veg_nonveg', 'Unknown'),
                    'Category__c': item.get('category', 'Unknown'),
                    'Dynamic_Dish__c': item.get('dynamic_dish', False)
                })
            else:
                sf.Ingredient_Object__c.create({
                    'Ingredient_Name__c': item['name'],
                    'Category__c': item.get('category', 'Unknown'),
                    'Description__c': item.get('description', 'No description available')
                })
            logger.info(f"Successfully submitted item to {item.get('source', '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, host='0.0.0.0', port=7860)