update

app.py

@@ -1,5 +1,5 @@
 import uuid
-from flask import Flask, render_template, request, redirect, url_for, send_from_directory, session
+from flask import Flask, render_template, request, redirect, url_for, send_from_directory
 import json
 import random
 import os
@@ -27,6 +27,10 @@ else:
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'supersecretkey'  # Change this to a random secret key
 
+# File-based session storage
+SESSION_DIR = '/tmp/sessions'
+os.makedirs(SESSION_DIR, exist_ok=True)
+
 # Directories for visualizations
 VISUALIZATION_DIRS = {
     "No-XAI": "htmls_NO_XAI_mod",
@@ -53,6 +57,19 @@ def generate_session_id():
     return str(uuid.uuid4())
 
 def save_session_data(session_id, data):
+    file_path = os.path.join(SESSION_DIR, f'{session_id}.json')
+    with open(file_path, 'w') as f:
+        json.dump(data, f)
+    logger.info(f"Session data saved for session {session_id}")
+
+def load_session_data(session_id):
+    file_path = os.path.join(SESSION_DIR, f'{session_id}.json')
+    if os.path.exists(file_path):
+        with open(file_path, 'r') as f:
+            return json.load(f)
+    return None
+
+def save_session_data_to_hf(session_id, data):
     try:
         username = data.get('username', 'unknown')
         seed = data.get('seed', 'unknown')
@@ -128,7 +145,6 @@ def introduction():
 def attribution():
     return render_template('attribution.html')
 
-
 @app.route('/index', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
@@ -156,8 +172,8 @@ def index():
                 'start_time': start_time,
                 'session_id': session_id
             }
-            session['data'] = session_data  # Store data in session
-            logger.info(f"Session data stored for user {username}, method {method}")
+            save_session_data(session_id, session_data)
+            logger.info(f"Session data stored for user {username}, method {method}, session_id {session_id}")
 
             # Redirect based on the selected method
             if method == 'No-XAI':
@@ -171,7 +187,7 @@ def index():
 
 @app.route('/explanation/<session_id>')
 def explanation(session_id):
-    session_data = session.get('data')
+    session_data = load_session_data(session_id)
     if not session_data:
         logger.error(f"No session data found for session ID: {session_id}")
         return redirect(url_for('index'))
@@ -194,7 +210,7 @@ def explanation(session_id):
 @app.route('/experiment/<session_id>', methods=['GET', 'POST'])
 def experiment(session_id):
     try:
-        session_data = session.get('data')
+        session_data = load_session_data(session_id)
         if not session_data:
             return redirect(url_for('index'))
 
@@ -230,13 +246,13 @@ def subjective(session_id):
     if request.method == 'POST':
         understanding = request.form.get('understanding')
 
-        session_data = session.get('data')
+        session_data = load_session_data(session_id)
         if not session_data:
             logger.error(f"No session data found for session: {session_id}")
             return redirect(url_for('index'))
 
         session_data['subjective_feedback'] = understanding
-        session['data'] = session_data  # Update session data
+        save_session_data(session_id, session_data)
 
         return redirect(url_for('completed', session_id=session_id))
 
@@ -248,7 +264,7 @@ def feedback():
         session_id = request.form['session_id']
         prediction = request.form['prediction']
 
-        session_data = session.get('data')
+        session_data = load_session_data(session_id)
         if not session_data:
             logger.error(f"No session data found for session: {session_id}")
             return redirect(url_for('index'))
@@ -259,7 +275,7 @@ def feedback():
         })
 
         session_data['current_index'] += 1
-        session['data'] = session_data  # Update session data
+        save_session_data(session_id, session_data)
         logger.info(f"Prediction saved for session {session_id}, sample {session_data['current_index'] - 1}")
 
         if session_data['current_index'] >= len(session_data['selected_samples']):
@@ -273,7 +289,7 @@ def feedback():
 @app.route('/completed/<session_id>')
 def completed(session_id):
     try:
-        session_data = session.get('data')
+        session_data = load_session_data(session_id)
         if not session_data:
             logger.error(f"No session data found for session: {session_id}")
             return redirect(url_for('index'))
@@ -338,10 +354,10 @@ def completed(session_id):
         session_data['false_percentage'] = false_percentage
 
         # Save all the data to Hugging Face at the end
-        save_session_data(session_id, session_data)
+        save_session_data_to_hf(session_id, session_data)
 
-        # Clear the session data
-        session.pop('data', None)
+        # Remove the local session data file
+        os.remove(os.path.join(SESSION_DIR, f'{session_id}.json'))
 
         return render_template('completed.html',
                                accuracy=accuracy,
@@ -371,6 +387,5 @@ def send_visualization(filename):
 def send_examples(filename):
     return send_from_directory('', filename)
 
-
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)