update

app.py

@@ -125,18 +125,29 @@ def load_samples():
 
 def select_balanced_samples(samples):
     try:
-        if len(samples) < 10:
-            logger.warning(f"Not enough common samples. Only {len(samples)} available.")
-            return samples
+        # Separate samples into two groups
+        tp_fp_samples = [s for s in samples if s['category'] in ['TP', 'FP']]
+        tn_fn_samples = [s for s in samples if s['category'] in ['TN', 'FN']]
 
-        selected_samples = random.sample(samples, 10)
-        logger.info(f"Selected 10 unique samples")
+        # Check if we have enough samples in each group
+        if len(tp_fp_samples) < 5 or len(tn_fn_samples) < 5:
+            logger.warning(f"Not enough samples in each category. TP+FP: {len(tp_fp_samples)}, TN+FN: {len(tn_fn_samples)}")
+            return samples if len(samples) <= 10 else random.sample(samples, 10)
+
+        # Select 5 samples from each group
+        selected_tp_fp = random.sample(tp_fp_samples, 5)
+        selected_tn_fn = random.sample(tn_fn_samples, 5)
+
+        # Combine and shuffle the selected samples
+        selected_samples = selected_tp_fp + selected_tn_fn
+        random.shuffle(selected_samples)
+
+        logger.info(f"Selected 10 balanced samples: 5 from TP+FP, 5 from TN+FN")
         return selected_samples
     except Exception as e:
         logger.exception("Error selecting balanced samples")
         return []
 
-
 @app.route('/')
 def introduction():
     return render_template('introduction.html')