Update app.py

app.py

@@ -831,6 +831,46 @@ def directionalize_amount_headers(md: str) -> str:
     )
 
 
+def directionalize_amount_headers_by_nearby_context(md: str) -> str:
+    """
+    Use nearby narrative labels around each table (outside HTML cells) to set
+    Amount->Credit/Debit when the table itself is ambiguous.
+    """
+    if not md or "<table" not in md.lower():
+        return md
+
+    credit_hint_re = re.compile(
+        r"\b(all\s+credit\s+activity|credit\s+activity|deposits?|credits?|money\s+in)\b",
+        re.IGNORECASE,
+    )
+    debit_hint_re = re.compile(
+        r"\b(all\s+debit\s+activity|debit\s+activity|withdrawals?|debits?|checks?|money\s+out)\b",
+        re.IGNORECASE,
+    )
+
+    table_re = re.compile(r"<table\b[^>]*>.*?</table>", re.IGNORECASE | re.DOTALL)
+    out_parts: List[str] = []
+    last_end = 0
+    for tm in table_re.finditer(md):
+        out_parts.append(md[last_end:tm.start()])
+        tbl = tm.group(0)
+
+        left = re.sub(r"<[^>]+>", " ", md[max(0, tm.start() - 700): tm.start()])
+        right = re.sub(r"<[^>]+>", " ", md[tm.end(): min(len(md), tm.end() + 250)])
+        ctx = f"{left} {right}"
+        is_credit = bool(credit_hint_re.search(ctx))
+        is_debit = bool(debit_hint_re.search(ctx))
+        if is_credit ^ is_debit:
+            w = "Credit" if is_credit else "Debit"
+            tbl = re.sub(r"(<th\b[^>]*>\s*)amount(\s*</th>)", rf"\1{w}\2", tbl, flags=re.IGNORECASE)
+            tbl = re.sub(r"(<td\b[^>]*>\s*)amount(\s*</td>)", rf"\1{w}\2", tbl, flags=re.IGNORECASE)
+
+        out_parts.append(tbl)
+        last_end = tm.end()
+    out_parts.append(md[last_end:])
+    return "".join(out_parts)
+
+
 def _strip_trailing_money_tokens(rest: str, max_take: int = 2) -> Tuple[str, List[str]]:
     """Strip 1–2 currency tokens from the right of *rest*; return (description, tokens oldest-first)."""
     cur = rest.rstrip()
@@ -1004,6 +1044,7 @@ def light_stabilize_markdown(page_md: str) -> str:
     merged = close_unclosed_html("\n\n".join(out_blocks))
     merged = mask_non_monetary_long_numbers(merged)
     merged = directionalize_amount_headers(merged)
+    merged = directionalize_amount_headers_by_nearby_context(merged)
     merged = repair_thead_cell_semantics(merged)
     merged = stabilize_table_markup(merged, rounds=4)
     merged = strip_degenerate_html_tables(merged)
@@ -1750,6 +1791,7 @@ def run_ocr(uploaded_file):
         if merged and merged != "(No content)" and not merged.lstrip().startswith("Error:"):
             merged = mask_non_monetary_long_numbers(merged)
             merged = directionalize_amount_headers(merged)
+            merged = directionalize_amount_headers_by_nearby_context(merged)
             merged = normalize_ambiguous_amount_balance_tables(merged)
             merged = normalize_amount_only_transaction_tables(merged)
             merged = stabilize_table_markup(merged, rounds=4)