Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
from flask import Flask, render_template, request, jsonify, redirect, url_for, session
|
| 2 |
from flask.sessions import SecureCookieSessionInterface
|
| 3 |
from salesforce import get_salesforce_connection
|
|
|
|
| 4 |
import os
|
| 5 |
|
| 6 |
# Initialize Flask app and Salesforce connection
|
|
@@ -15,12 +16,13 @@ print("Salesforce connection established.")
|
|
| 15 |
# Set the secret key to handle sessions securely
|
| 16 |
app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q") # Replace with a secure key
|
| 17 |
|
| 18 |
-
|
| 19 |
-
app.config["
|
| 20 |
-
app.config["
|
| 21 |
-
app.config["
|
| 22 |
-
app.config["
|
| 23 |
|
|
|
|
| 24 |
|
| 25 |
# Ensure secure session handling for environments like Hugging Face
|
| 26 |
app.session_interface = SecureCookieSessionInterface()
|
|
@@ -54,7 +56,6 @@ def login():
|
|
| 54 |
if request.method == "POST":
|
| 55 |
email = request.form.get("email")
|
| 56 |
password = request.form.get("password")
|
| 57 |
-
print(f"Login Succesful")
|
| 58 |
print(f"Login attempt with email: {email}") # Debug log
|
| 59 |
|
| 60 |
try:
|
|
@@ -65,7 +66,7 @@ def login():
|
|
| 65 |
session['user_id'] = result["records"][0]['Id']
|
| 66 |
session['user_email'] = email
|
| 67 |
print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
|
| 68 |
-
print(f"Session cookie: {
|
| 69 |
return redirect(url_for("menu"))
|
| 70 |
else:
|
| 71 |
print("Invalid credentials!")
|
|
@@ -78,9 +79,9 @@ def login():
|
|
| 78 |
|
| 79 |
@app.route("/menu", methods=["GET", "POST"])
|
| 80 |
def menu():
|
| 81 |
-
print(f"Cookies on /menu: {request.cookies}")
|
| 82 |
selected_category = request.args.get("category", "All")
|
| 83 |
user_id = session.get('user_id')
|
|
|
|
| 84 |
print(f"Session check in /menu: user_id={user_id}")
|
| 85 |
|
| 86 |
if not user_id:
|
|
|
|
| 1 |
from flask import Flask, render_template, request, jsonify, redirect, url_for, session
|
| 2 |
from flask.sessions import SecureCookieSessionInterface
|
| 3 |
from salesforce import get_salesforce_connection
|
| 4 |
+
from redis import Redis
|
| 5 |
import os
|
| 6 |
|
| 7 |
# Initialize Flask app and Salesforce connection
|
|
|
|
| 16 |
# Set the secret key to handle sessions securely
|
| 17 |
app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q") # Replace with a secure key
|
| 18 |
|
| 19 |
+
# Set session cookie policies
|
| 20 |
+
app.config["SESSION_TYPE"] = "redis"
|
| 21 |
+
app.config["SESSION_REDIS"] = Redis(host="localhost", port=6379) # Replace with Redis details
|
| 22 |
+
app.config["SESSION_COOKIE_SAMESITE"] = "None"
|
| 23 |
+
app.config["SESSION_COOKIE_SECURE"] = True
|
| 24 |
|
| 25 |
+
Session(app)
|
| 26 |
|
| 27 |
# Ensure secure session handling for environments like Hugging Face
|
| 28 |
app.session_interface = SecureCookieSessionInterface()
|
|
|
|
| 56 |
if request.method == "POST":
|
| 57 |
email = request.form.get("email")
|
| 58 |
password = request.form.get("password")
|
|
|
|
| 59 |
print(f"Login attempt with email: {email}") # Debug log
|
| 60 |
|
| 61 |
try:
|
|
|
|
| 66 |
session['user_id'] = result["records"][0]['Id']
|
| 67 |
session['user_email'] = email
|
| 68 |
print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
|
| 69 |
+
print(f"Session cookie: {request.cookies.get(app.session_cookie_name)}") # Check session cookie
|
| 70 |
return redirect(url_for("menu"))
|
| 71 |
else:
|
| 72 |
print("Invalid credentials!")
|
|
|
|
| 79 |
|
| 80 |
@app.route("/menu", methods=["GET", "POST"])
|
| 81 |
def menu():
|
|
|
|
| 82 |
selected_category = request.args.get("category", "All")
|
| 83 |
user_id = session.get('user_id')
|
| 84 |
+
print(f"Cookies on /menu: {request.cookies}") # Debug: Check cookies sent
|
| 85 |
print(f"Session check in /menu: user_id={user_id}")
|
| 86 |
|
| 87 |
if not user_id:
|