mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-13 07:08:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Lots of import refactoring

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer
2024-09-23 12:22:49 +02:00
parent ac51a09065
commit 12477c8cac
7 changed files with 26 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/batch_convert.py
View File

@@ -32,8 +32,16 @@ def export_documents(
with (output_dir / f"{doc_filename}.json").open("w") as fp:
fp.write(json.dumps(conv_res.render_as_dict()))
# Export Docling document format to JSON (experimental):
with (output_dir / f"{doc_filename}.experimental.json").open("w") as fp:
fp.write(
json.dumps(
conv_res.experimental.model_dump(mode="json", by_alias=True)
)
)
# Export Docling document format to YAML (experimental):
with (output_dir / f"{doc_filename}.yaml").open("w") as fp:
with (output_dir / f"{doc_filename}.experimental.yaml").open("w") as fp:
fp.write(
yaml.safe_dump(
conv_res.experimental.model_dump(mode="json", by_alias=True)

4
examples/export_figures.py
View File

@@ -8,7 +8,7 @@ from docling.datamodel.base_models import (
ConversionStatus,
FigureElement,
PageElement,
TableElement,
Table,
)
from docling.datamodel.document import DocumentConversionInput
from docling.document_converter import DocumentConverter
@@ -61,7 +61,7 @@ def main():
# Export figures and tables
for element, image in conv_res.render_element_images(
element_types=(FigureElement, TableElement)
element_types=(FigureElement, Table)
):
element_image_filename = (
output_dir / f"{doc_filename}-element-{element.id}.png"