Update app.py

app.py

@@ -207,18 +207,17 @@ def ocr_zone(image_path, y_start_frac, y_end_frac):
     return ""
 
 
-def fix_account_number(page_md, hdr):
-    """If page_md has 'Account Number:' with no value after it,
-    find the account number from header text and inject it."""
+def fix_account_number(hdr):
+    """If hdr contains Account Number: with no value, inject the number from hdr."""
     import re
-    if not page_md or not hdr:
-        return page_md
-    if not re.search(r'Account Number:\s*\S', page_md):
-        match = re.search(r'(\d{10,})', hdr)
+    if not hdr:
+        return hdr
+    if "Account Number:" in hdr and not re.search("Account Number: [0-9]", hdr):
+        match = re.search("[0-9]{5,}", hdr)
         if match:
-            acct = match.group(1)
-            page_md = page_md.replace('Account Number:', 'Account Number: ' + acct)
-    return page_md
+            acct = match.group(0)
+            hdr = hdr.replace("Account Number:", "Account Number: " + acct)
+    return hdr
 
 
 def close_unclosed_html(md):