update

app.py

@@ -4,7 +4,6 @@ import random
 import os
 import string
 import logging
-from flask_session import Session  # Add this import
 
 # Set up logging
 logging.basicConfig(level=logging.INFO,
@@ -17,9 +16,6 @@ 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 = {
@@ -36,6 +32,20 @@ VISUALIZATION_DIRS_CHAIN_OF_TABLE = {
     "FN": "htmls_COT/FN"
 }
 
+# Function to save session data to a file
+def save_session_data(username, data):
+    os.makedirs('session_data', exist_ok=True)
+    with open(f'session_data/{username}_session.json', 'w') as f:
+        json.dump(data, f)
+
+# Function to load session data from a file
+def load_session_data(username):
+    try:
+        with open(f'session_data/{username}_session.json', 'r') as f:
+            return json.load(f)
+    except FileNotFoundError:
+        return None
+
 # Load all sample files from the directories based on the selected method
 def load_samples(method):
     logger.info(f"Loading samples for method: {method}")
@@ -82,60 +92,53 @@ def index():
             return "Missing username, seed, or method", 400
 
         try:
-            # Save the method to a file
-            with open(f'session_data/method_{username}.txt', 'w') as f:
-                f.write(method)
-
             seed = int(seed)
             random.seed(seed)
             all_samples = load_samples(method)
             selected_samples = select_balanced_samples(all_samples)
-            logger.info(f"Number of selected samples: {len(selected_samples)}")  # Log the number of samples
+            logger.info(f"Number of selected samples: {len(selected_samples)}")
+
             if len(selected_samples) == 0:
                 logger.error("No samples were selected.")
                 return "No samples were selected", 500
 
-            random_string = generate_random_string()
-            filename = f'{username}_{seed}_{method}_{random_string}.json'
-
+            filename = f'{username}_{seed}_{method}_{generate_random_string()}.json'
             logger.info(f"Generated filename: {filename}")
 
-            # Save selected samples to a JSON file
-            os.makedirs('session_data', exist_ok=True)
-            with open(f'session_data/{filename}', 'w') as f:
-                json.dump(selected_samples, f)
-
-            session['responses'] = []  # Initialize responses list
-            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))
+            # Save session data
+            session_data = {
+                'responses': [],
+                'username': username,
+                'selected_samples': selected_samples,
+                'method': method,
+                'filename': filename,
+                'current_index': 0
+            }
+            save_session_data(username, session_data)
+            logger.info(f"Session data saved for user: {username}")
+
+            return redirect(url_for('experiment', username=username))
         except Exception as e:
             logger.exception(f"Error in index route: {e}")
             return "An error occurred", 500
     return render_template('index.html')
 
-@app.route('/experiment/<username>/<int:sample_index>/<seed>/<filename>', methods=['GET', 'POST'])
-def experiment(username, sample_index, seed, filename):
+@app.route('/experiment/<username>', methods=['GET', 'POST'])
+def experiment(username):
     try:
-        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.")
+        session_data = load_session_data(username)
+        if not session_data:
+            logger.error(f"No session data found for user: {username}")
             return redirect(url_for('index'))
 
-        if sample_index >= len(selected_samples):
-            return redirect(url_for('completed', filename=filename))
+        selected_samples = session_data['selected_samples']
+        method = session_data['method']
+        current_index = session_data['current_index']
+
+        if current_index >= len(selected_samples):
+            return redirect(url_for('completed', username=username))
 
-        visualization_file = selected_samples[sample_index]
+        visualization_file = selected_samples[current_index]
 
         if method == "Chain-of-Table":
             vis_dir = 'htmls_COT'
@@ -155,80 +158,56 @@ def experiment(username, sample_index, seed, filename):
 
         statement = "Please make a decision to Accept/Reject the AI prediction based on the explanation."
 
-        # Render the template with the correct path for the iframe
         return render_template('experiment.html',
-                               sample_id=sample_index,
+                               sample_id=current_index,
                                statement=statement,
                                visualization=url_for('send_visualization', path=visualization_path),
-                               username=username,
-                               seed=seed,
-                               sample_index=sample_index,
-                               filename=filename)
+                               username=username)
     except Exception as e:
         logger.exception(f"An error occurred in the experiment route: {e}")
         return "An error occurred", 500
 
-
 @app.route('/feedback', methods=['POST'])
 def feedback():
     try:
-        sample_id = request.form['sample_id']
-        feedback = request.form['feedback']
         username = request.form['username']
-        seed = request.form['seed']
-        sample_index = int(request.form['sample_index'])
-        filename = request.form['filename']
+        feedback = request.form['feedback']
 
-        selected_samples = session.get('selected_samples', [])
-        responses = session.get('responses', [])
+        session_data = load_session_data(username)
+        if not session_data:
+            logger.error(f"No session data found for user: {username}")
+            return redirect(url_for('index'))
 
         # Store the feedback
-        responses.append({
-            'sample_id': sample_id,
+        session_data['responses'].append({
+            'sample_id': session_data['current_index'],
             'feedback': feedback
         })
-        session['responses'] = responses
 
-        # Create the result directory if it doesn't exist
-        result_dir = 'human_study'
-        os.makedirs(result_dir, exist_ok=True)
+        # Move to the next sample
+        session_data['current_index'] += 1
 
-        # Load existing data if the JSON file exists
-        filepath = os.path.join(result_dir, filename)
-        if os.path.exists(filepath):
-            with open(filepath, 'r') as f:
-                data = json.load(f)
-        else:
-            data = {}
-
-        # Update data with the current feedback
-        data[sample_index] = {
-            'Username': username,
-            'Seed': seed,
-            'Sample ID': sample_id,
-            'Task': "Please make a decision to Accept/Reject the AI prediction based on the explanation.",
-            'User Feedback': feedback
-        }
-
-        # Save updated data to the file
-        with open(filepath, 'w') as f:
-            json.dump(data, f, indent=4)
-
-        next_sample_index = sample_index + 1
-        if next_sample_index >= len(selected_samples):
-            return redirect(url_for('completed', filename=filename))
-
-        return redirect(
-            url_for('experiment', username=username, sample_index=next_sample_index, seed=seed, filename=filename))
+        # Save updated session data
+        save_session_data(username, session_data)
+
+        if session_data['current_index'] >= len(session_data['selected_samples']):
+            return redirect(url_for('completed', username=username))
+
+        return redirect(url_for('experiment', username=username))
     except Exception as e:
         logger.exception(f"Error in feedback route: {e}")
         return "An error occurred", 500
 
-@app.route('/completed/<filename>')
-def completed(filename):
+@app.route('/completed/<username>')
+def completed(username):
     try:
-        responses = session.get('responses', [])
-        method = session.get('method')
+        session_data = load_session_data(username)
+        if not session_data:
+            logger.error(f"No session data found for user: {username}")
+            return redirect(url_for('index'))
+
+        responses = session_data['responses']
+        method = session_data['method']
 
         if method == "Chain-of-Table":
             json_file = 'Tabular_LLMs_human_study_vis_6_COT.json'
@@ -245,7 +224,8 @@ def completed(filename):
         for response in responses:
             sample_id = response['sample_id']
             feedback = response['feedback']
-            index = sample_id.split('-')[1].split('.')[0]  # Extract index from filename
+            visualization_file = session_data['selected_samples'][sample_id]
+            index = visualization_file.split('-')[1].split('.')[0]  # Extract index from filename
 
             if feedback.upper() == "TRUE":
                 accept_count += 1