Sync main.py from blood-brain-omics repo

main.py

@@ -485,25 +485,37 @@ def h2h_blood_summary(
     metric: str = Query("r2"),
     covariates: str = Query("none", enum=["none", "included"]),
     model: str = Query("TabPFN"),
+    brain: Optional[str] = Query(None, description="Restrict win counts to a single brain target"),
 ):
-    """Win counts across all blood H2H pairs."""
+    """Win counts across blood H2H pairs.
+
+    If ``brain`` is provided, wins are counted only against that brain
+    target (e.g. ``neuropathology``). Otherwise wins are counted across
+    every brain target in the H2H phase.
+    """
     validate_metric(metric)
     phase = "h2h_blood_withcov" if covariates == "included" else "h2h_blood"
 
-    rows = q(f"""
+    sql = f"""
         SELECT h2h_pair, brain_target, target, blood_platform, {metric}
         FROM results
         WHERE phase = ? AND model = ? AND h2h_pair IS NOT NULL
-        ORDER BY h2h_pair, brain_target, target
-    """, [phase, model]).fetchall()
+    """
+    params = [phase, model]
+    if brain is not None:
+        sql += " AND brain_target = ?"
+        params.append(brain)
+    sql += " ORDER BY h2h_pair, brain_target, target"
+
+    rows = q(sql, params).fetchall()
 
     # Count wins per pair
     pair_wins = {}
     current = None
     buffer = {}
 
-    for pair, brain, target, platform, val in rows:
-        key = (pair, brain, target)
+    for pair, row_brain, target, platform, val in rows:
+        key = (pair, row_brain, target)
         if key != current:
             if current and len(buffer) == 2:
                 pair_key = current[0]
@@ -534,6 +546,7 @@ def h2h_blood_summary(
     return {
         "metric": metric,
         "covariates": covariates,
+        "brain": brain,
         "pairs": pair_wins,
     }