Bug fix in determine_heading_level

OneOffTech · Oct 2, 2024 · b2e6d98 · b2e6d98
1 parent 8443c20
commit b2e6d98
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/text_extractor/parser/pdfact_parser.py b/text_extractor/parser/pdfact_parser.py
@@ -276,6 +276,10 @@ def determine_heading_level(document: Document) -> Document:
                         })
                     else:
                         existing_style['occurrences'] += 1
+
+    if not heading_styles:
+        return document
+
     # Sort the styles by font size in descending order
     heading_styles = sorted(heading_styles, key=lambda x: x['font_size'], reverse=True)