Update app.py

app.py

@@ -752,7 +752,8 @@ def mask_non_monetary_long_numbers(md: str) -> str:
         # Keep YYYYMMDD-like date compact tokens.
         if len(tok) == 8 and tok.startswith(("19", "20")):
             return tok
-        return f"ID#{tok}"
+        # Replace digits entirely so downstream cannot reinterpret ID tokens as money.
+        return "IDNUM"
 
     return _LONG_NUM_TOKEN.sub(repl, md)
 
@@ -1161,6 +1162,83 @@ def normalize_ambiguous_amount_balance_tables(md: str) -> str:
     )
 
 
+def normalize_amount_only_transaction_tables(md: str) -> str:
+    """
+    Convert transaction-like tables with Date/Description/Amount (no Credit/Debit)
+    into explicit Credit/Debit columns using generic lexical heuristics.
+    """
+    if not md or "<table" not in md.lower():
+        return md
+
+    credit_hint_re = re.compile(
+        r"\b(deposit|deposits|credit|credits|incoming|received|return|refund|interest|pmt cr)\b",
+        re.IGNORECASE,
+    )
+    debit_hint_re = re.compile(
+        r"\b(debit|debits|withdrawal|withdrawals|check|checks|fee|fees|charge|charges|payment|purchase|ach|bill)\b",
+        re.IGNORECASE,
+    )
+
+    def repl_table(m: re.Match) -> str:
+        full = m.group(0)
+        rows = _extract_rows_plain_from_table(full)
+        if len(rows) < 3:
+            return full
+        hdr = [c.strip().lower() for c in rows[0]]
+        if any("credit" in c for c in hdr) or any("debit" in c for c in hdr):
+            return full
+        amount_idxs = [i for i, c in enumerate(hdr) if "amount" in c]
+        date_idxs = [i for i, c in enumerate(hdr) if "date" in c]
+        desc_idxs = [i for i, c in enumerate(hdr) if "description" in c or "memo" in c or "details" in c]
+        if not amount_idxs or not date_idxs:
+            return full
+        amount_idx = amount_idxs[0]
+        date_idx = date_idxs[0]
+        desc_idx = desc_idxs[0] if desc_idxs else min(1, len(rows[0]) - 1)
+        table_ctx = _cell_plain_text(full)
+
+        new_rows = [["Date", "Description", "Credit", "Debit"]]
+        converted = 0
+        for r in rows[1:]:
+            if not r:
+                continue
+            padded = r + [""] * (max(amount_idx, date_idx, desc_idx) + 1 - len(r))
+            date_txt = (padded[date_idx] or "").strip()
+            if not _is_date_like(date_txt):
+                continue
+            desc_txt = (padded[desc_idx] or "").strip()
+            amt = _parse_amount_or_none((padded[amount_idx] or "").strip())
+            if amt is None:
+                continue
+            ctx = f"{table_ctx} {desc_txt}"
+            credit = ""
+            debit = ""
+            if credit_hint_re.search(ctx) and not debit_hint_re.search(ctx):
+                credit = _fmt_money(abs(amt))
+            elif debit_hint_re.search(ctx) and not credit_hint_re.search(ctx):
+                debit = _fmt_money(abs(amt))
+            else:
+                # Conservative default for unknown direction.
+                debit = _fmt_money(abs(amt))
+            new_rows.append([date_txt, desc_txt, credit, debit])
+            converted += 1
+
+        if converted < 5:
+            return full
+
+        html_rows = ["<tr>" + "".join(f"<th>{html.escape(c)}</th>" for c in new_rows[0]) + "</tr>"]
+        for rr in new_rows[1:]:
+            html_rows.append("<tr>" + "".join(f"<td>{html.escape(c)}</td>" for c in rr) + "</tr>")
+        return "<table>\n" + "\n".join(html_rows) + "\n</table>"
+
+    return re.sub(
+        r"<table\b[^>]*>.*?</table>",
+        repl_table,
+        md,
+        flags=re.IGNORECASE | re.DOTALL,
+    )
+
+
 def _extract_daily_balance_by_date(md: str):
     """
     Build a date->balance map from daily-balance style tables.
@@ -1672,6 +1750,7 @@ def run_ocr(uploaded_file):
             merged = mask_non_monetary_long_numbers(merged)
             merged = directionalize_amount_headers(merged)
             merged = normalize_ambiguous_amount_balance_tables(merged)
+            merged = normalize_amount_only_transaction_tables(merged)
             merged = stabilize_table_markup(merged, rounds=4)
             merged = enrich_transaction_tables_with_daily_balances(merged)
             # Keep transaction amounts untouched; add explicit snapshot for metadata extraction.