update

app.py

@@ -3,7 +3,6 @@ import json
 import random
 import os
 import string
-from flask_session import Session
 import logging
 
 # Set up logging
@@ -17,8 +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'] = 'redis'
-Session(app)
 
 # Directories for visualizations
 VISUALIZATION_DIRS_PLAN_OF_SQLS = {
@@ -99,7 +96,10 @@ def index():
 
             logger.info(f"Generated filename: {filename}")
 
-            session['selected_samples'] = selected_samples
+            # Save selected samples to a JSON file
+            with open(f'session_data/{filename}', 'w') as f:
+                json.dump(selected_samples, f)
+
             session['responses'] = []  # Initialize responses list
             session['method'] = method  # Store the selected method
 
@@ -108,11 +108,17 @@ def index():
             logger.exception(f"Error in index route: {e}")
             return "An error occurred", 500
     return render_template('index.html')
+
+
 @app.route('/experiment/<username>/<sample_index>/<seed>/<filename>', methods=['GET'])
 def experiment(username, sample_index, seed, filename):
     try:
         sample_index = int(sample_index)
-        selected_samples = session.get('selected_samples', [])
+
+        # Load selected samples from the JSON file
+        with open(f'session_data/{filename}', 'r') as f:
+            selected_samples = json.load(f)
+
         method = session.get('method')  # Retrieve the selected method
 
         if sample_index >= len(selected_samples):
@@ -154,21 +160,6 @@ def experiment(username, sample_index, seed, filename):
         return "An error occurred", 500
 
 
-@app.route('/visualizations/<path:path>')
-def send_visualization(path):
-    try:
-        method = session.get('method')
-        if method == "Chain-of-Table":
-            visualization_dir = 'htmls_COT'
-        else:  # Default to Plan-of-SQLs
-            visualization_dir = 'visualizations'
-
-        return send_from_directory(visualization_dir, path)
-    except Exception as e:
-        logger.exception(f"Error sending visualization: {e}")
-        return "An error occurred", 500
-
-
 @app.route('/feedback', methods=['POST'])
 def feedback():
     try:
@@ -179,7 +170,10 @@ def feedback():
         sample_index = int(request.form['sample_index'])
         filename = request.form['filename']
 
-        selected_samples = session.get('selected_samples', [])
+        # Load selected samples from the JSON file
+        with open(f'session_data/{filename}', 'r') as f:
+            selected_samples = json.load(f)
+
         responses = session.get('responses', [])
 
         responses.append({
@@ -227,6 +221,7 @@ def completed(filename):
     try:
         responses = session.get('responses', [])
         method = session.get('method')
+
         if method == "Chain-of-Table":
             json_file = 'Tabular_LLMs_human_study_vis_6_COT.json'
         else:  # Default to Plan-of-SQLs
@@ -254,8 +249,7 @@ def completed(filename):
             else:
                 ground_truth_key = f"POS_test-{index}.html"
 
-            if ground_truth_key in ground_truth and ground_truth[ground_truth_key][
-                'answer'].upper() == feedback.upper():
+            if ground_truth_key in ground_truth and ground_truth[ground_truth_key]['answer'].upper() == feedback.upper():
                 correct_responses += 1
             else:
                 logger.warning(f"Missing or mismatched key: {ground_truth_key}")
@@ -278,24 +272,6 @@ def completed(filename):
         return "An error occurred", 500
 
 
-# if __name__ == '__main__':
-#     try:
-#         app.run(debug=False, port=7860)
-#     except Exception as e:
-#         logger.exception(f"Failed to start app: {e}")
-
-
-# from flask import Flask
-#
-# app = Flask(__name__)
-#
-# @app.route('/')
-# def index():
-#     return "Hello, world!"
-
-# if __name__ == '__main__':
-#     app.run(debug=False, port=7860)
-
 if __name__ == "__main__":
+    os.makedirs('session_data', exist_ok=True)  # Ensure the directory for session files exists
     app.run(host="0.0.0.0", port=7860)
-