update

app.py

@@ -110,10 +110,6 @@ def index():
             return "An error occurred", 500
     return render_template('index.html')
 
-@app.route('/visualization/<path:filename>')
-def send_visualization(filename):
-    return send_from_directory('visualizations', filename)
-
 @app.route('/experiment/<username>/<sample_index>/<seed>/<filename>', methods=['GET'])
 def experiment(username, sample_index, seed, filename):
     try:
@@ -146,7 +142,7 @@ def experiment(username, sample_index, seed, filename):
         # Determine the correct visualization directory based on the category
         for category, dir_path in (VISUALIZATION_DIRS_CHAIN_OF_TABLE if method == "Chain-of-Table" else VISUALIZATION_DIRS_PLAN_OF_SQLS).items():
             if visualization_file in os.listdir(dir_path):
-                visualization_path = f'{category}/{visualization_file}'
+                visualization_path = os.path.join(vis_dir, category, visualization_file)
                 break
         else:
             logger.error(f"Visualization file {visualization_file} not found.")
@@ -156,20 +152,12 @@ def experiment(username, sample_index, seed, filename):
 
         statement = "Please make a decision to Accept/Reject the AI prediction based on the explanation."
 
-        # Use the custom route to serve the visualization
-        return render_template('experiment.html',
-                               sample_id=sample_index,
-                               statement=statement,
-                               visualization=url_for('send_visualization', filename=visualization_path),
-                               username=username,
-                               seed=seed,
-                               sample_index=sample_index,
-                               filename=filename)
+        # Serve the file directly from the directory
+        return send_from_directory(directory=os.path.join(os.getcwd(), vis_dir, category), path=visualization_file)
     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: