proxy

create_app.py

@@ -5,8 +5,8 @@ from flask import render_template
 from transformers import LongformerTokenizer, AutoTokenizer, AutoModelForCausalLM, LongformerConfig
 from models.model import *
 from utils.util_func import *
-from safetensors.torch import load_file
 import os 
+from werkzeug.middleware.proxy_fix import ProxyFix
 
 db = SQLAlchemy()
 login_manager = LoginManager()
@@ -39,9 +39,12 @@ def load_models():
 def create_app():
     set_seed(42)
     app = Flask(__name__)
-    app.secret_key = os.environ.get("SECRET_KEY", "this-should-be-overridden-in-prod")
+    app.secret_key = "super-secret"
+    app.wsgi_app = ProxyFix(app.wsgi_app, x_for=1, x_proto=1)
     app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db'
     app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+    app.config['SESSION_COOKIE_SECURE'] = True
+    app.config['SESSION_COOKIE_SAMESITE'] = 'Lax'
     load_models()
     db.init_app(app)
     login_manager.login_view = 'auth.login'