fix: add the bbox to TableData from annotations (#123)

Signed-off-by: Peter Staar <[email protected]>

Author: PeterStaar-IBM

docling_eval/utils/utils.py

@@ -17,7 +17,7 @@
 from docling.backend.docling_parse_v4_backend import DoclingParseV4DocumentBackend
 from docling.datamodel.base_models import InputFormat, Page
 from docling.datamodel.document import InputDocument
-from docling_core.types.doc.base import BoundingBox, Size
+from docling_core.types.doc.base import BoundingBox, CoordOrigin, Size
 from docling_core.types.doc.document import (
     DoclingDocument,
     GraphData,
@@ -177,6 +177,20 @@ def yield_cells_from_html_table(
                 text_cell = text_cells[text_cell_id]
                 text = "".join(text_cell["tokens"])
 
+            bbox = None
+            if (
+                text_cells is not None
+                and text_cell_id < len(text_cells)
+                and "bbox" in text_cells[text_cell_id]
+            ):
+                bbox = BoundingBox(
+                    l=text_cells[text_cell_id]["bbox"][0],
+                    b=text_cells[text_cell_id]["bbox"][1],
+                    r=text_cells[text_cell_id]["bbox"][2],
+                    t=text_cells[text_cell_id]["bbox"][3],
+                    coord_origin=CoordOrigin.BOTTOMLEFT,
+                )
+
             rowspan = int(cell.get("rowspan", 1))
             colspan = int(cell.get("colspan", 1))
 
@@ -186,7 +200,7 @@ def yield_cells_from_html_table(
                     grid[row_idx + r][col_idx + c] = text
 
             # print(f"Row: {row_idx + 1}, Col: {col_idx + 1}, Text: {text}")
-            yield row_idx, col_idx, rowspan, colspan, text
+            yield row_idx, col_idx, rowspan, colspan, text, bbox
 
             col_idx += colspan  # Move to next column after colspan
 
@@ -202,9 +216,14 @@ def convert_html_table_into_docling_tabledata(
     cells = []
 
     try:
-        for row_idx, col_idx, rowspan, colspan, text in yield_cells_from_html_table(
-            table_html=table_html, text_cells=text_cells
-        ):
+        for (
+            row_idx,
+            col_idx,
+            rowspan,
+            colspan,
+            text,
+            bbox,
+        ) in yield_cells_from_html_table(table_html=table_html, text_cells=text_cells):
             cell = TableCell(
                 row_span=rowspan,
                 col_span=colspan,
@@ -213,6 +232,7 @@ def convert_html_table_into_docling_tabledata(
                 start_col_offset_idx=col_idx,
                 end_col_offset_idx=col_idx + colspan,
                 text=text,
+                bbox=bbox,
             )
             cells.append(cell)