docling-core 2.12.1__py3-none-any.whl → 2.13.1__py3-none-any.whl

docling_core/types/doc/document.py

@@ -14,7 +14,7 @@ import warnings
 from io import BytesIO
 from pathlib import Path
 from typing import Any, Dict, Final, List, Literal, Optional, Tuple, Union
-from urllib.parse import unquote
+from urllib.parse import quote, unquote
 
 import pandas as pd
 import yaml
@@ -830,7 +830,7 @@ class PictureItem(FloatingItem):
             ):
                 return default_response
 
-            text = f"\n![Image]({str(self.image.uri)})\n"
+            text = f"\n![Image]({quote(str(self.image.uri))})\n"
             return text
 
         else:
@@ -884,7 +884,7 @@ class PictureItem(FloatingItem):
             ):
                 return default_response
 
-            img_text = f'<img src="{str(self.image.uri)}">'
+            img_text = f'<img src="{quote(str(self.image.uri))}">'
             return f"<figure>{caption_text}{img_text}</figure>"
 
         else:

docling_core/types/doc/labels.py

@@ -1,12 +1,12 @@
 """Models for the labels types."""
 
 from enum import Enum
+from typing import Tuple
 
 
 class DocItemLabel(str, Enum):
     """DocItemLabel."""
 
-    # DocLayNet v2
     CAPTION = "caption"
     FOOTNOTE = "footnote"
     FORMULA = "formula"
@@ -26,13 +26,39 @@ class DocItemLabel(str, Enum):
     KEY_VALUE_REGION = "key_value_region"
 
     # Additional labels for markup-based formats (e.g. HTML, Word)
-    PARAGRAPH = "paragraph"  # explicitly a paragraph and not arbitrary text
+    PARAGRAPH = "paragraph"
     REFERENCE = "reference"
 
     def __str__(self):
         """Get string value."""
         return str(self.value)
 
+    @staticmethod
+    def get_color(label: "DocItemLabel") -> Tuple[int, int, int]:
+        """Return the RGB color associated with a given label."""
+        color_map = {
+            DocItemLabel.CAPTION: (255, 204, 153),
+            DocItemLabel.FOOTNOTE: (200, 200, 255),
+            DocItemLabel.FORMULA: (192, 192, 192),
+            DocItemLabel.LIST_ITEM: (153, 153, 255),
+            DocItemLabel.PAGE_FOOTER: (204, 255, 204),
+            DocItemLabel.PAGE_HEADER: (204, 255, 204),
+            DocItemLabel.PICTURE: (255, 204, 164),
+            DocItemLabel.SECTION_HEADER: (255, 153, 153),
+            DocItemLabel.TABLE: (255, 204, 204),
+            DocItemLabel.TEXT: (255, 255, 153),
+            DocItemLabel.TITLE: (255, 153, 153),
+            DocItemLabel.DOCUMENT_INDEX: (220, 220, 220),
+            DocItemLabel.CODE: (125, 125, 125),
+            DocItemLabel.CHECKBOX_SELECTED: (255, 182, 193),
+            DocItemLabel.CHECKBOX_UNSELECTED: (255, 182, 193),
+            DocItemLabel.FORM: (200, 255, 255),
+            DocItemLabel.KEY_VALUE_REGION: (183, 65, 14),
+            DocItemLabel.PARAGRAPH: (255, 255, 153),
+            DocItemLabel.REFERENCE: (176, 224, 230),
+        }
+        return color_map[label]
+
 
 class GroupLabel(str, Enum):
     """GroupLabel."""

{docling_core-2.12.1.dist-info → docling_core-2.13.1.dist-info}/METADATA

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

{docling_core-2.12.1.dist-info → docling_core-2.13.1.dist-info}/RECORD

@@ -24,8 +24,8 @@ docling_core/types/__init__.py,sha256=MVRSgsk5focwGyAplh_TRR3dEecIXpd98g_u3zZ5HX
 docling_core/types/base.py,sha256=PusJskRVL19y-hq0BgXr5e8--QEqSqLnFNJ8UbOqW88,8318
 docling_core/types/doc/__init__.py,sha256=bEL4zKVOG7Wxm6xQrgF58mu-Teds9aSavuEAKVNhrTU,639
 docling_core/types/doc/base.py,sha256=_ttU8QI8wXDTQRUnN5n7L6D9wYFVLSAibxlFoMbgAsk,4557
-docling_core/types/doc/document.py,sha256=2W4wZunI0K_EOxNtY5jbKeyw7bYWKKNLiljxfN8anHc,91844
-docling_core/types/doc/labels.py,sha256=Pc5avKtGM2fv-w7mXinoxs9BkhktmFaJ6ACsgFiAAm4,1702
+docling_core/types/doc/document.py,sha256=ZHQBozH4-85p0YNBjzpTFURmTOenUrOwRyNr67shVcs,91865
+docling_core/types/doc/labels.py,sha256=7fzopzvUhnSUalKumQvxSUCutbvEh9aUIKvJ1WKA_lU,2888
 docling_core/types/doc/tokens.py,sha256=uU_MYW_p7ypf7eYICFBvxdnVaPZ7CQnvZmbJ6oPrtEA,6134
 docling_core/types/doc/utils.py,sha256=YDOh_ZD1Y7OmCEDdCLJ_MO5K3HA67nc_acfhOK6WztU,1439
 docling_core/types/gen/__init__.py,sha256=C6TuCfvpSnSL5XDOFMcYHUY2-i08vvfOGRcdu6Af0pI,124
@@ -56,8 +56,8 @@ docling_core/utils/generate_jsonschema.py,sha256=uNX1O5XnjyB5nA66XqZXTt3YbGuR2ty
 docling_core/utils/legacy.py,sha256=xfp7U0JqjI60K3loWiNTk8w08_KfCUzTb2MNULBOIz4,24396
 docling_core/utils/validate.py,sha256=aQ11UbFyl8iD_N7yTTZmm_VVeXz8KcCyn3GLXgkfYRM,2049
 docling_core/utils/validators.py,sha256=azcrndLzhNkTWnbFSu9shJ5D3j_znnLrIFA5R8hzmGU,2798
-docling_core-2.12.1.dist-info/LICENSE,sha256=2M9-6EoQ1sxFztTOkXGAtwUDJvnWaAHdB9BYWVwGkIw,1087
-docling_core-2.12.1.dist-info/METADATA,sha256=7GQuAsiJL6EGBIEjeiX-XQs7gELqhCGWsNeSrsGm5Kk,5744
-docling_core-2.12.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-docling_core-2.12.1.dist-info/entry_points.txt,sha256=oClcdb2L2RKx4jdqUykY16Kum_f0_whwWhGzIodyidc,216
-docling_core-2.12.1.dist-info/RECORD,,
+docling_core-2.13.1.dist-info/LICENSE,sha256=2M9-6EoQ1sxFztTOkXGAtwUDJvnWaAHdB9BYWVwGkIw,1087
+docling_core-2.13.1.dist-info/METADATA,sha256=xKJXvKHpi2TIOxxghWq6wxlg-U7QcAoF9WPZ1U0c0EU,5744
+docling_core-2.13.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+docling_core-2.13.1.dist-info/entry_points.txt,sha256=oClcdb2L2RKx4jdqUykY16Kum_f0_whwWhGzIodyidc,216
+docling_core-2.13.1.dist-info/RECORD,,