Update app.py

app.py

@@ -12,15 +12,27 @@ import joblib
 import numpy as np
 from flask import (Flask, render_template, request, redirect,
                    url_for, session, flash, g)
+from werkzeug.middleware.proxy_fix import ProxyFix
 
 # ── App setup ─────────────────────────────────────────────────────────────────
 app = Flask(__name__)
-app.secret_key = os.environ.get("SECRET_KEY", "iris-ml-secret-2024")
+
+# CRITIQUE pour HuggingFace : l'app est derrière un reverse proxy
+app.wsgi_app = ProxyFix(app.wsgi_app, x_for=1, x_proto=1, x_host=1, x_prefix=1)
+
+app.secret_key = os.environ.get("SECRET_KEY", "iris-ml-secret-2024-xK9pL")
+
+# CRITIQUE : config cookies pour fonctionner derrière proxy HF
+app.config.update(
+    SESSION_COOKIE_SAMESITE="None",
+    SESSION_COOKIE_SECURE=True,
+    SESSION_COOKIE_HTTPONLY=True,
+    SESSION_COOKIE_NAME="irisai_session",
+)
 
 BASE_DIR  = os.path.dirname(os.path.abspath(__file__))
 
-# ── Chemin DB persistant sur HuggingFace ──────────────────────────────────────
-# Sur HF Spaces, /data/ est persistant si activé, sinon fallback local
+# Chemin DB persistant sur HuggingFace
 _data_dir = "/data" if os.path.isdir("/data") else BASE_DIR
 DB_PATH   = os.path.join(_data_dir, "users.db")
 
@@ -54,7 +66,6 @@ def close_db(_):
 def init_db():
     with app.app_context():
         db = get_db()
-
         db.execute("""
             CREATE TABLE IF NOT EXISTS users (
                 id       INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -64,7 +75,6 @@ def init_db():
                 created  TEXT NOT NULL
             )
         """)
-
         db.execute("""
             CREATE TABLE IF NOT EXISTS sessions (
                 id        INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -74,7 +84,6 @@ def init_db():
                 FOREIGN KEY (user_id) REFERENCES users(id)
             )
         """)
-
         db.execute("""
             CREATE TABLE IF NOT EXISTS predictions (
                 id           INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -88,7 +97,6 @@ def init_db():
                 timestamp    TEXT NOT NULL
             )
         """)
-
         try:
             db.execute("ALTER TABLE users ADD COLUMN email TEXT NOT NULL DEFAULT ''")
         except sqlite3.OperationalError:
@@ -137,8 +145,11 @@ def login():
             (username, hash_pw(password))
         ).fetchone()
         if user:
+            session.clear()
             session["username"]      = username
             session["user_id"]       = user["id"]
+            session.permanent        = True
+
             db.execute(
                 "INSERT INTO sessions (user_id, login_at) VALUES (?, ?)",
                 (user["id"], datetime.now().isoformat())
@@ -146,6 +157,7 @@ def login():
             cursor = db.execute("SELECT last_insert_rowid()")
             session["session_db_id"] = cursor.fetchone()[0]
             db.commit()
+
             flash(f"Bienvenue, {username} ! 👋", "success")
             return redirect(url_for("predict"))
         flash("Nom d'utilisateur ou mot de passe incorrect.", "danger")
@@ -294,7 +306,7 @@ def admin():
     return render_template("admin.html", users=users, sessions_log=sessions_log, stats=stats)
 
 
-# ── Init DB au démarrage (fonctionne avec Gunicorn ET python app.py) ──────────
+# ── Init DB au démarrage ───────────────────────────────────────────────────────
 init_db()
 
 if __name__ == "__main__":