Update app.py

app.py

@@ -1,7 +1,7 @@
 from flask import Flask, render_template, request, jsonify, redirect, url_for, session
 import os
-from flask_session import Session
 from salesforce import get_salesforce_connection
+from flask.sessions import SecureCookieSessionInterface
 
 # Initialize Flask app and Salesforce connection
 print("Starting app...")
@@ -14,11 +14,10 @@ print("Salesforce connection established.")
 
 # Set the secret key to handle sessions securely
 app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q")  # Replace with a secure key
-app.config["SESSION_TYPE"] = "filesystem"
 
-# Initialize session
-Session(app)
-print("Session configured.")
+# Ensure secure session handling for environments like Hugging Face
+app.session_interface = SecureCookieSessionInterface()
+print("Session interface configured.")
 
 @app.route("/")
 def home():
@@ -58,6 +57,7 @@ def login():
                 session['user_id'] = result["records"][0]['Id']
                 session['user_email'] = email
                 print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
+                 print(f"Cookies after login: {request.cookies}"
                 return redirect(url_for("menu"))
             else:
                 print("Invalid credentials!")