update

app.py

@@ -136,12 +136,15 @@ def experiment(username, sample_index, seed, filename):
 
         # Find the correct visualization path
         for category, dir_path in visualization_dirs.items():
-            if visualization_file in os.listdir(dir_path):
-                visualization_path = f"{category}/{visualization_file}"
-                break
+            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"{category}/{visualization_file}"
+                    break
 
         if not visualization_path:
-            logger.error(f"Visualization file {visualization_file} not found in any category.")
+            logger.error(
+                f"Visualization file {visualization_file} not found. Searched path: {absolute_dir_path}/{visualization_file}")
             return "Visualization file not found", 404
 
         logger.info(f"Rendering experiment page with visualization: {visualization_path}")