diff --git a/docling/models/page_preprocessing_model.py b/docling/models/page_preprocessing_model.py index 9dc291a..d1b29e3 100644 --- a/docling/models/page_preprocessing_model.py +++ b/docling/models/page_preprocessing_model.py @@ -63,7 +63,13 @@ class PagePreprocessingModel(BasePageModel): def draw_text_boxes(image, cells, show: bool = False): draw = ImageDraw.Draw(image) for c in cells: - x0, y0, x1, y1 = c.bbox.as_tuple() + x0, y0, x1, y1 = ( + c.to_bounding_box().l, + c.to_bounding_box().t, + c.to_bounding_box().r, + c.to_bounding_box().b, + ) + draw.rectangle([(x0, y0), (x1, y1)], outline="red") if show: image.show()