docling-core 2.3.1__tar.gz → 2.4.0__tar.gz

{docling_core-2.3.1 → docling_core-2.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: docling-core
-Version: 2.3.1
+Version: 2.4.0
 Summary: A python library to define and validate data types in Docling.
 Home-page: https://ds4sd.github.io/
 License: MIT

{docling_core-2.3.1 → docling_core-2.4.0}/docling_core/types/doc/document.py

@@ -551,6 +551,28 @@ class DocItem(
 
         return location
 
+    def get_image(self, doc: "DoclingDocument") -> Optional[PILImage.Image]:
+        """Returns the image of this DocItem.
+
+        The function returns None if this DocItem has no valid provenance or
+        if a valid image of the page containing this DocItem is not available
+        in doc.
+        """
+        if not len(self.prov):
+            return None
+
+        page = doc.pages.get(self.prov[0].page_no)
+        if page is None or page.size is None or page.image is None:
+            return None
+
+        page_image = page.image.pil_image
+        crop_bbox = (
+            self.prov[0]
+            .bbox.to_top_left_origin(page_height=page.size.height)
+            .scaled(scale=page_image.height / page.size.height)
+        )
+        return page_image.crop(crop_bbox.as_tuple())
+
 
 class TextItem(DocItem):
     """TextItem."""
@@ -633,6 +655,20 @@ class FloatingItem(DocItem):
             text += cap.resolve(doc).text
         return text
 
+    def get_image(self, doc: "DoclingDocument") -> Optional[PILImage.Image]:
+        """Returns the image corresponding to this FloatingItem.
+
+        This function returns the PIL image from self.image if one is available.
+        Otherwise, it uses DocItem.get_image to get an image of this FloatingItem.
+
+        In particular, when self.image is None, the function returns None if this
+        FloatingItem has no valid provenance or the doc does not contain a valid image
+        for the required page.
+        """
+        if self.image is not None:
+            return self.image.pil_image
+        return super().get_image(doc=doc)
+
 
 class PictureItem(FloatingItem):
     """PictureItem."""
@@ -1255,7 +1291,10 @@ class DoclingDocument(BaseModel):
                 # If the child is a NodeItem, recursively traverse it
                 if not isinstance(child, PictureItem) or traverse_pictures:
                     yield from self.iterate_items(
-                        child, _level=_level + 1, with_groups=with_groups
+                        child,
+                        _level=_level + 1,
+                        with_groups=with_groups,
+                        page_no=page_no,
                     )
 
     def print_element_tree(self):
@@ -1281,11 +1320,12 @@ class DoclingDocument(BaseModel):
         image_mode: ImageRefMode = ImageRefMode.PLACEHOLDER,
         indent: int = 4,
         text_width: int = -1,
+        page_no: Optional[int] = None,
     ) -> str:
         r"""Serialize to Markdown.
 
-        Operates on a slice of the document's main_text as defined through arguments
-        main_text_start and main_text_stop; defaulting to the whole main_text.
+        Operates on a slice of the document's body as defined through arguments
+        from_element and to_element; defaulting to the whole document.
 
         :param delim: Delimiter to use when concatenating the various
                 Markdown parts. Defaults to "\n\n".
@@ -1294,11 +1334,9 @@ class DoclingDocument(BaseModel):
                 Defaults to 0.
         :type from_element: int
         :param to_element: Body slicing stop index
-                (exclusive). Defaults to None.
-        :type to_element: Optional[int]
+                (exclusive). Defaults to 0maxint.
+        :type to_element: int
         :param delim: str:  (Default value = "\n\n")
-        :param from_element: int:  (Default value = 0)
-        :param to_element: Optional[int]:  (Default value = None)
         :param labels: set[DocItemLabel]
         :param "subtitle-level-1":
         :param "paragraph":
@@ -1306,7 +1344,6 @@ class DoclingDocument(BaseModel):
         :param "table":
         :param "Text":
         :param "text":
-        :param ]:
         :param strict_text: bool:  (Default value = False)
         :param image_placeholder str:  (Default value = "<!-- image -->")
             the placeholder to include to position images in the markdown.
@@ -1320,7 +1357,7 @@ class DoclingDocument(BaseModel):
         in_list = False  # Track if we're currently processing list items
 
         for ix, (item, level) in enumerate(
-            self.iterate_items(self.body, with_groups=True)
+            self.iterate_items(self.body, with_groups=True, page_no=page_no)
         ):
             # If we've moved to a lower level, we're exiting one or more groups
             if level < previous_level:
@@ -1331,7 +1368,7 @@ class DoclingDocument(BaseModel):
 
             previous_level = level  # Update previous_level for next iteration
 
-            if ix < from_element and to_element <= ix:
+            if ix < from_element or to_element <= ix:
                 continue  # skip as many items as you want
 
             # Handle newlines between different types of content

{docling_core-2.3.1 → docling_core-2.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docling-core"
-version = "2.3.1"
+version = "2.4.0"
 description = "A python library to define and validate data types in Docling."
 license = "MIT"
 authors = [