update

app.py

@@ -4,6 +4,7 @@ import random
 import os
 import string
 import logging
+from flask_session import Session  # Add this import
 
 # Set up logging
 logging.basicConfig(level=logging.INFO,
@@ -16,6 +17,9 @@ logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'supersecretkey'  # Change this to a random secret key
+app.config['SESSION_TYPE'] = 'filesystem'  # Use filesystem-based session
+Session(app)  # Initialize the session extension
+
 
 # Directories for visualizations
 VISUALIZATION_DIRS_PLAN_OF_SQLS = {
@@ -105,6 +109,9 @@ def index():
             session['username'] = username  # Store the username for later use
             session['selected_samples'] = selected_samples  # Store selected samples in session
             session['method'] = method  # Store the method in session
+            session['filename'] = filename  # Store the filename in session
+
+            logger.info(f"Session data set: {session}")  # Log session data
 
             return redirect(url_for('experiment', username=username, sample_index=0, seed=seed, filename=filename))
         except Exception as e:
@@ -118,6 +125,13 @@ def experiment(username, sample_index, seed, filename):
         selected_samples = session.get('selected_samples', [])
         method = session.get('method')
 
+        logger.info(f"Experiment route - Session data: {session}")  # Log session data
+        logger.info(f"Selected samples: {selected_samples}")  # Log selected samples
+
+        if not selected_samples:
+            logger.error("No samples in session. Redirecting to index.")
+            return redirect(url_for('index'))
+
         if sample_index >= len(selected_samples):
             return redirect(url_for('completed', filename=filename))
 
@@ -154,6 +168,7 @@ def experiment(username, sample_index, seed, filename):
         logger.exception(f"An error occurred in the experiment route: {e}")
         return "An error occurred", 500
 
+
 @app.route('/feedback', methods=['POST'])
 def feedback():
     try: