update

app.py

@@ -236,27 +236,31 @@ def experiment(username):
             return redirect(url_for('completed', username=username))
 
         sample = selected_samples[current_index]
+        logger.info(f"sample:\n{sample}")
+
         method_a, method_b = methods
 
         # Find matching files for both methods
-        file_a = None
-        file_b = None
 
         method_a_dir = get_method_dir(method_a)
         method_b_dir = get_method_dir(method_b)
 
-        for category in ['TP', 'TN', 'FP', 'FN']:
-            dir_a = f'htmls_{method_a_dir.upper()}/{category}'
-            dir_b = f'htmls_{method_b_dir.upper()}/{category}'
-
-            files_a = os.listdir(dir_a)
-            files_b = os.listdir(dir_b)
-
-            matching_files = set(files_a) & set(files_b)
-            if matching_files:
-                file_a = os.path.join(dir_a, next(iter(matching_files)))
-                file_b = os.path.join(dir_b, next(iter(matching_files)))
-                break
+        # for category in ['TP', 'TN', 'FP', 'FN']:
+        category = sample['category']
+        dir_a = f'htmls_{method_a_dir.upper()}/{category}'
+        dir_b = f'htmls_{method_b_dir.upper()}/{category}'
+
+        file_a = os.path.join(dir_a, sample['file'])
+        file_b = os.path.join(dir_b, sample['file'])
+
+            # files_a = os.listdir(dir_a)
+            # files_b = os.listdir(dir_b)
+            #
+            # matching_files = set(files_a) & set(files_b)
+            # if matching_files:
+            #     file_a = os.path.join(dir_a, next(iter(matching_files)))
+            #     file_b = os.path.join(dir_b, next(iter(matching_files)))
+            #     break
 
         if not file_a or not file_b:
             logger.error(f"Missing files for comparison at index {current_index}")