Skip to content

Commit

Permalink
added the detection of h1 and the skip_furniture parameter
Browse files Browse the repository at this point in the history
Signed-off-by: Peter Staar <[email protected]>
  • Loading branch information
PeterStaar-IBM committed Oct 31, 2024
1 parent c52e68c commit ebe0b20
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions docling/backend/html_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@


class HTMLDocumentBackend(DeclarativeDocumentBackend):
def __init__(self, in_doc: "InputDocument", path_or_stream: Union[BytesIO, Path]):
def __init__(self, in_doc: "InputDocument", path_or_stream: Union[BytesIO, Path], skip_furniture:bool=False):
super().__init__(in_doc, path_or_stream)
_log.debug("About to init HTML backend...")
self.soup = None
Expand All @@ -35,6 +35,8 @@ def __init__(self, in_doc: "InputDocument", path_or_stream: Union[BytesIO, Path]
self.parents[i] = None
self.labels = {} # type: ignore

self.skip_furniture = skip_furniture

try:
if isinstance(self.path_or_stream, BytesIO):
text_stream = self.path_or_stream.getvalue().decode("utf-8")
Expand Down Expand Up @@ -82,7 +84,7 @@ def convert(self) -> DoclingDocument:
for br in self.soup.body.find_all("br"):
br.replace_with("\n")

self.contains_h1 = True
self.contains_h1 = bool(soup.find('h1')) and self.skip_furniture
self.detected_h1 = False

doc = self.walk(self.soup.body, doc)
Expand Down

0 comments on commit ebe0b20

Please sign in to comment.