From d0d1ac09573c8e7c6f13588f64a7eb32e901bca4 Mon Sep 17 00:00:00 2001
From: Michele Dolfi <dol@zurich.ibm.com>
Date: Mon, 23 Sep 2024 20:13:03 +0200
Subject: [PATCH] feat: add figures in markdown

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
---
 docling/datamodel/document.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/docling/datamodel/document.py b/docling/datamodel/document.py
index c6bdfcb2..4a80ca98 100644
--- a/docling/datamodel/document.py
+++ b/docling/datamodel/document.py
@@ -324,8 +324,10 @@ class ConvertedDocument(BaseModel):
             "paragraph",
             "caption",
             "table",
+            "figure",
         ],
         strict_text: bool = False,
+        image_placeholder: str = "<!-- image -->",
     ):
         return self.output.export_to_markdown(
             delim=delim,
@@ -333,6 +335,7 @@ class ConvertedDocument(BaseModel):
             main_text_stop=main_text_stop,
             main_text_labels=main_text_labels,
             strict_text=strict_text,
+            image_placeholder=image_placeholder,
         )
 
     def render_as_text(