update

app.py

@@ -109,7 +109,6 @@ def index():
             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:
@@ -134,15 +133,15 @@ def experiment(username, sample_index, seed, filename):
         else:
             visualization_dirs = VISUALIZATION_DIRS_PLAN_OF_SQLS
 
-        logger.info(f"visualization_dirs: {visualization_path}")
+        logger.info(f"Checking directories for method: {method}")
 
         # Find the correct visualization path
         for category, dir_path in visualization_dirs.items():
             absolute_dir_path = os.path.abspath(dir_path)
-            for file in os.listdir(absolute_dir_path):
-                if file.strip() == visualization_file.strip():
-                    visualization_path = f"{visualization_dirs}/{category}/{visualization_file}"
-                    break
+            logger.info(f"Checking directory: {absolute_dir_path} for file: {visualization_file}")
+            if visualization_file.strip() in os.listdir(absolute_dir_path):
+                visualization_path = os.path.join(category, visualization_file)
+                break
 
         if not visualization_path:
             logger.error(