Update app.py

app.py

@@ -238,15 +238,18 @@ def run_ocr(uploaded_file):
         if is_pdf:
             doc = fitz.open(path)
             page_images = []
+            page_heights = []
             for i in range(len(doc)):
                 pix = doc[i].get_pixmap(matrix=fitz.Matrix(1.5, 1.5), alpha=False)
                 img_path = os.path.join(tempfile.gettempdir(), f"maas_page_{i}.png")
                 pix.save(img_path)
                 page_images.append(img_path)
+                page_heights.append(pix.height)
             doc.close()
             results = parser.parse(page_images)
         else:
             page_images = [path]
+            page_heights = []
             results = parser.parse(path)
 
         if not isinstance(results, list):
@@ -255,7 +258,8 @@ def run_ocr(uploaded_file):
         all_pages = []
         for page_num, page_result in enumerate(results):
             page_md, regions = get_page_md_and_regions(page_result)
-            header_end_frac, footer_start_frac = get_header_footer_zones(regions, 1000)
+            img_h = page_heights[page_num] if page_num < len(page_heights) else 1000
+            header_end_frac, footer_start_frac = get_header_footer_zones(regions, img_h)
 
             # MaaS does not return header/footer regions; always use at least default bands
             he = header_end_frac if header_end_frac is not None else DEFAULT_ZONE_FRAC