Upload app.py with huggingface_hub

app.py

@@ -206,7 +206,7 @@ def get_confident_alleles(bead_results, bead_hla_map):
         return ag
 
     x6x8 = set()
-    not_x6x8 = set()  # Rxn=1, 2, 4 全部排除
+    not_x6x8 = set()  # Rxn<6 全部
     for bid, r in bead_results.items():
         hla = bead_hla_map.get(bid, {})
         alleles = {_clean(a) for a in _parse_allele_list(hla.get('allele', ''))}
@@ -217,6 +217,54 @@ def get_confident_alleles(bead_results, bead_hla_map):
     return x6x8 - not_x6x8
 
 
+def get_confident_alleles_pra2(bead_results, bead_hla_map, threshold=0.8):
+    """
+    PRA2 專用：80% Rule + Gray Zone
+    Step 1: 有 X2/X1 → 硬性排除
+    Step 2: 只有 X4 → 80% rule（X6X8/total ≥ 80% 才列入）
+    """
+    def _clean(ag):
+        ag = ag.split('/')[0]
+        ag = ag.split('=')[0]
+        return ag
+
+    # 收集每個 allele 在不同 Rxn 層級的 bead
+    allele_beads = {}  # allele -> {'x6x8': set, 'x4': set, 'x2x1': set}
+    for bid, r in bead_results.items():
+        hla = bead_hla_map.get(bid, {})
+        alleles = {_clean(a) for a in _parse_allele_list(hla.get('allele', ''))}
+        for ag in alleles:
+            if ag not in allele_beads:
+                allele_beads[ag] = {'x6x8': set(), 'x4': set(), 'x2x1': set()}
+            if r['rxn'] >= 6:
+                allele_beads[ag]['x6x8'].add(bid)
+            elif r['rxn'] >= 4:
+                allele_beads[ag]['x4'].add(bid)
+            else:
+                allele_beads[ag]['x2x1'].add(bid)
+
+    confident = set()
+    for ag, levels in allele_beads.items():
+        # 必須有 X6/X8
+        if not levels['x6x8']:
+            continue
+        # Step 1: 有 X2/X1 → 硬性排除
+        if levels['x2x1']:
+            continue
+        # Step 2: 只有 X4 → 80% rule
+        if levels['x4']:
+            total = len(levels['x6x8']) + len(levels['x4'])
+            ratio = len(levels['x6x8']) / total
+            if ratio >= threshold:
+                confident.add(ag)  # X4 調整為 Positive
+            # else: < 80% → 排除
+        else:
+            # 全部都是 X6/X8，直接列入
+            confident.add(ag)
+
+    return confident
+
+
 def _build_allele_to_sero(bead_hla_map):
     """從 bead map 建立 allele→sero 對照"""
     mapping = {}
@@ -613,7 +661,10 @@ def parse_xls_file(raw_bytes):
     br_dict = OrderedDict()
     for b in beads_detail:
         br_dict[b['bead']] = {'rxn': b['rxn']}
-    confident = get_confident_alleles(br_dict, bead_map)
+    if pra_class == 'PRA2':
+        confident = get_confident_alleles_pra2(br_dict, bead_map)
+    else:
+        confident = get_confident_alleles(br_dict, bead_map)
 
     sero_mfi = build_sero_mfi_stats(beads_detail, confident, bead_map) if overall == 'Positive' else []
 
@@ -727,7 +778,10 @@ def full_analyze(csv_content, filename='upload.csv'):
         br = analyze_sample(median_data[sname], nc_medians, bead_ids, bead_map)
         pra6, pos6, tot = calculate_pra(br, 'X6')
         overall = 'Positive' if pra6 > 0 else 'Negative'
-        confident = get_confident_alleles(br, bead_map)
+        if pra_class == 'PRA2':
+            confident = get_confident_alleles_pra2(br, bead_map)
+        else:
+            confident = get_confident_alleles(br, bead_map)
 
         # bead 明細
         beads_detail = []