Update app.py

app.py

@@ -1,34 +1,48 @@
+import os
 from flask import Flask, render_template, request
-from salesforce import get_salesforce_connection
+from simple_salesforce import Salesforce
 
-# Initialize Flask app and Salesforce connection
-print("Starting app...")
+# Initialize Flask app
 app = Flask(__name__)
-print("Flask app initialized.")
 
-# Salesforce Connection
-sf = get_salesforce_connection()
-print("Salesforce connection established.")
+# Securely store the secret key
+app.secret_key = os.environ.get('FLASK_SECRET_KEY', 'your_secret_key')
+
+# Secure Salesforce Connection (Uses environment variables)
+sf = Salesforce(
+    username=os.environ.get('SF_USERNAME'),
+    password=os.environ.get('SF_PASSWORD'),
+    security_token=os.environ.get('SF_SECURITY_TOKEN')
+)
 
 @app.route("/")
 def home():
-    return render_template("menu.html")  # Directly render the menu page
+    return redirect("/menu")  # Redirect home page to menu page
 
-@app.route("/menu", methods=["GET"])
+@app.route("/menu")
 def menu():
     selected_category = request.args.get("category", "All")
     try:
+        # Salesforce SOQL Query
         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 []
+
+        # Extract menu items
+        food_items = result.get('records', [])
+
+        # Extract unique categories
         categories = {item['Category__c'] for item in food_items if 'Category__c' in item}
+
+        # Filter by category if selected
         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(host='0.0.0.0', port=7860, debug=False)
+    app.run(host="0.0.0.0", port=7860, debug=True)