import os
from flask import Flask, render_template, request
from simple_salesforce import Salesforce

app = Flask(__name__)
app.secret_key = os.environ.get('FLASK_SECRET_KEY', 'your_secret_key')

# Secure Salesforce Connection
sf = Salesforce(
    username=os.environ.get('SF_USERNAME'),
    password=os.environ.get('SF_PASSWORD'),
    security_token=os.environ.get('SF_SECURITY_TOKEN')
)

@app.route("/menu")
def menu():
    selected_category = request.args.get("category", "All")
    try:
        query = "SELECT Name, Price__c, Image1__c, Category__c, Description__c FROM Menu_Item__c"
        result = sf.query(query)
        food_items = result['records'] if 'records' in result else []
        categories = {item['Category__c'] for item in food_items if 'Category__c' in item}
        if selected_category != "All":
            food_items = [item for item in food_items if item.get("Category__c") == selected_category]
    except Exception as e:
        food_items = []
        categories = []
        print(f"Error fetching data: {e}")
    return render_template("menu.html", food_items=food_items, categories=categories, selected_category=selected_category)

if __name__ == "__main__":
    app.run(debug=True)