Update app.py

app.py

@@ -216,7 +216,7 @@ def fix_account_number(hdr):
     if not hdr:
         return hdr
     # Step 1: inject account number if missing after label
-    if "Account Number:" in hdr and not re.search("Account Number: [0-9]", hdr):
+    if "Account Number:" in hdr and "Account Number: " not in hdr:
         match = re.search("[0-9]{5,}", hdr)
         if match:
             acct = match.group(0)
@@ -224,11 +224,12 @@ def fix_account_number(hdr):
     # Step 2: remove leading number that is duplicated after Account Number:
     # e.g. "000000731823008 JPMorgan ... Account Number: 000000731823008"
     # -> "JPMorgan ... Account Number: 000000731823008"
-    acct_match = re.search("Account Number:\s*([0-9]{5,})", hdr)
+    acct_match = re.search(r"Account Number: ([0-9]{5,})", hdr)
     if acct_match:
         acct = acct_match.group(1)
         # If hdr starts with that same number, remove it from the start
-        hdr = re.sub("^" + re.escape(acct) + "\s*", "", hdr)
+        if hdr.startswith(acct):
+            hdr = hdr[len(acct):].lstrip()
     return hdr