feat(329): Store routing in LayoutElement (#493)

Author: vladimir-kivi-ds

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 1.5.3
+
+- Store routing in LayoutElement
+
 ## 1.5.2
 
 ### Fix

unstructured_inference/__version__.py

@@ -1 +1 @@
-__version__ = "1.5.2"  # pragma: no cover
+__version__ = "1.5.3"  # pragma: no cover

unstructured_inference/inference/layout.py

@@ -203,9 +203,13 @@ def get_elements_with_detection_model(
         # remote call in the future.
         assert self.image is not None
         inferred_layout: LayoutElements = self.detection_model(self.image)
+        routing = inferred_layout.routing
+        routing_score = inferred_layout.routing_score
         inferred_layout = self.detection_model.deduplicate_detected_elements(
             inferred_layout,
         )
+        inferred_layout.routing = routing
+        inferred_layout.routing_score = routing_score
 
         if inplace:
             self.elements_array = inferred_layout

unstructured_inference/inference/layoutelement.py

@@ -26,6 +26,8 @@ class LayoutElements(TextRegions):
     element_class_id_map: dict[int, str] = field(default_factory=dict)
     text_as_html: np.ndarray = field(default_factory=lambda: np.array([]))
     table_as_cells: np.ndarray = field(default_factory=lambda: np.array([]))
+    routing: str | None = None
+    routing_score: float | None = None
     _optional_array_attributes: list[str] = field(
         init=False,
         default_factory=lambda: [