changes

app.py

@@ -7,6 +7,8 @@ import logging
 import sys
 from utils.ai_helpers import generate_notebook, stream_notebook_generation, stream_notebook_edit, edit_notebook
 from utils.notebook_helpers import format_notebook, extract_notebook_info
+# Added for server-side sessions
+from flask_session import Session
 
 # Configure logging
 logging.basicConfig(
@@ -31,6 +33,10 @@ if IS_HUGGINGFACE:
     app.config["SESSION_FILE_THRESHOLD"] = 10  # Low threshold to ensure writes
     # Important: Don't use large session lifetime on Hugging Face
     app.config["PERMANENT_SESSION_LIFETIME"] = 60 * 60 * 24  # 1 day only
+    # Added: Use server-side sessions to persist API keys across workers
+    app.config["SESSION_TYPE"] = "filesystem"
+    app.config["SESSION_FILE_DIR"] = os.path.join(os.getcwd(), "flask_session")
+    Session(app)
 
 # Map front-end model names to API model names
 MODEL_MAPPING = {