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

app = Flask(__name__)

# Salesforce credentials
SF_USERNAME = os.getenv("SF_USERNAME", "your_username")
SF_PASSWORD = os.getenv("SF_PASSWORD", "your_password")
SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN", "your_security_token")

# Connect to Salesforce
sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN)

@app.route("/menu", methods=["GET"])
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)