Update mdr_pdf_parser.py

mdr_pdf_parser.py

@@ -1542,9 +1542,13 @@ class MDRLayoutReader:
   def determine_reading_order(self, layouts: list[MDRLayoutElement], size: tuple[int, int]) -> list[MDRLayoutElement]:
     w, h = size
     if w <= 0 or h <= 0 or not layouts:
+        print("MDRLayoutReader: Invalid size or no layouts, returning early.")
         return layouts
-    model = self._get_model()
-    if model is None: # Fallback geometric sort
+    
+      model = self._get_model()
+    
+      if model is None: # Fallback geometric sort
+        print("MDRLayoutReader: Model is None, using fallback geometric sort.") 
         layouts.sort(key=lambda l: (l.rect.lt[1], l.rect.lt[0]))
         nfo = 0
         for l in layouts:
@@ -1552,9 +1556,15 @@ class MDRLayoutReader:
             [setattr(f, 'order', i + nfo) for i, f in enumerate(l.fragments)]
             nfo += len(l.fragments)
         return layouts
+    
+    print("MDRLayoutReader: Preparing bboxes...") # ADDED
     bbox_list = self._prepare_bboxes(layouts, w, h)
+    print(f"MDRLayoutReader: Prepared {len(bbox_list) if bbox_list else 'None or 0'} bboxes.")
+    
     if bbox_list is None or len(bbox_list) == 0:
+        print("MDRLayoutReader: No bboxes to process, returning layouts.")
         return layouts
+        
     l_size = 1000.0
     xs = l_size / float(w)
     ys = l_size / float(h)
@@ -1566,22 +1576,34 @@ class MDRLayoutReader:
         sx1 = max(0, min(l_size - 1, round(x1 * xs)))
         sy1 = max(0, min(l_size - 1, round(y1 * ys)))
         scaled_bboxes.append([min(sx0, sx1), min(sy0, sy1), max(sx0, sx1), max(sy0, sy1)])
+    print("MDRLayoutReader: Scaled bboxes prepared. Count: ", len(scaled_bboxes))
     orders = []
     try:
         with torch.no_grad():
+            print("MDRLayoutReader: Creating reader inputs...") # ADDED
             inputs = mdr_boxes_to_reader_inputs(scaled_bboxes)
+            print("MDRLayoutReader: Preparing inputs for model device...") # ADDED
             inputs = mdr_prepare_reader_inputs(inputs, model)
+            print("MDRLayoutReader: Running model inference...") # ADDED
             logits = model(**inputs).logits.cpu().squeeze(0)
+            print("MDRLayoutReader: Model inference complete. Parsing logits...") # ADDED
             orders = mdr_parse_reader_logits(logits, len(bbox_list))
+            print(f"MDRLayoutReader: Logits parsed. Orders count: {len(orders)}") # ADDED
     except Exception as e:
         print(f"MDR LayoutReader prediction error: {e}")
+        import traceback
+        traceback.print_exc() # ADDED for full traceback
         return layouts # Fallback
+        
     if len(orders) != len(bbox_list):
-        print("MDR LayoutReader order mismatch")
+        print(f"MDR LayoutReader order mismatch. Orders: {len(orders)}, BBoxes: {len(bbox_list)}")
         return layouts # Fallback
     for i, order_idx in enumerate(orders):
         bbox_list[i].order = order_idx
-    return self._apply_order(layouts, bbox_list)
+    print("MDRLayoutReader: Applying order...")
+    result_layouts = self._apply_order(layouts, bbox_list)
+    print("MDRLayoutReader: Order applied. Returning layouts.") # ADDED
+    return result_layouts
 
   def _prepare_bboxes(self, layouts: list[MDRLayoutElement], w: int, h: int) -> list[_MDR_ReaderBBox] | None:
     line_h = self._estimate_line_h(layouts)