update

app.py

@@ -109,6 +109,11 @@ def index():
             logger.exception(f"Error in index route: {e}")
             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:
@@ -138,10 +143,10 @@ def experiment(username, sample_index, seed, filename):
         else:
             vis_dir = 'visualizations'
 
-        # Construct the relative path to the static directory
+        # 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'static/{vis_dir}/{category}/{visualization_file}'
+                visualization_path = f'{category}/{visualization_file}'
                 break
         else:
             logger.error(f"Visualization file {visualization_file} not found.")
@@ -151,11 +156,11 @@ 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
+        # Use the custom route to serve the visualization
         return render_template('experiment.html',
                                sample_id=sample_index,
                                statement=statement,
-                               visualization=url_for('static', filename=f'{vis_dir}/{category}/{visualization_file}'),
+                               visualization=url_for('send_visualization', filename=visualization_path),
                                username=username,
                                seed=seed,
                                sample_index=sample_index,