update

app.py

@@ -136,10 +136,17 @@ def experiment(username, sample_index, seed, filename):
         visualization_path = None
 
         # Determine the correct visualization directory based on the method
+        # if method == "Chain-of-Table":
+        #     visualization_dirs = VISUALIZATION_DIRS_CHAIN_OF_TABLE
+        # else:
+        #     visualization_dirs = VISUALIZATION_DIRS_PLAN_OF_SQLS
+
         if method == "Chain-of-Table":
             visualization_dirs = VISUALIZATION_DIRS_CHAIN_OF_TABLE
+            vis_dir = 'htmls_COT'
         else:
             visualization_dirs = VISUALIZATION_DIRS_PLAN_OF_SQLS
+            vis_dir = 'visualizations'
 
         logger.info(f"Checking directories for method: {method}")
 
@@ -148,7 +155,7 @@ def experiment(username, sample_index, seed, filename):
             absolute_dir_path = os.path.abspath(dir_path)
             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)
+                visualization_path = os.path.join(vis_dir, category, visualization_file)
                 break
 
         if not visualization_path: