Merge pull request #32 from Filimoa/29

#29 [minor tweak to mashihua's branch]

pyproject.toml

@@ -7,7 +7,7 @@ name = "openparse"
 description = "Streamlines the process of preparing documents for LLM's."
 readme = "README.md"
 requires-python = ">=3.8"
-version = "0.5.3"
+version = "0.5.4"
 authors = [{name = "Sergey Filimonov", email = "[email protected]"}]
 dependencies = [
     "PyMuPDF >= 1.23.2",

src/openparse/tables/pymupdf/parse.py

@@ -22,11 +22,19 @@ def output_to_html(headers: List[str], rows: List[List[str]]) -> str:
 
 
 def output_to_markdown(headers: List[str], rows: List[List[str]]) -> str:
-    markdown_output = "| " + " | ".join(headers) + " |\n"
+    markdown_output = ""
+    if headers is not None:
+        for header in headers:
+            safe_header = "" if header is None else header
+            markdown_output += "| " + safe_header + " "
+
+    markdown_output += "|\n"
     markdown_output += "|---" * len(headers) + "|\n"
 
     for row in rows:
-        processed_row = [" " if cell in [None, ""] else cell.replace("\n", " ") for cell in row]
+        processed_row = [
+            " " if cell in [None, ""] else cell.replace("\n", " ") for cell in row
+        ]
         markdown_output += "| " + " | ".join(processed_row) + " |\n"
 
     return markdown_output

src/openparse/version.py

@@ -1,4 +1,4 @@
-OPEN_PARSE_VERSION = "0.5.3"
+OPEN_PARSE_VERSION = "0.5.4"
 
 
 def version_info() -> str: