luciano/markitdown
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update: add parameter "method" for PdfConverter

Browse source
This commit is contained in:
tungsten106 2024-12-23 10:24:51 +08:00
parent df5f14ef23
commit 263e0b5bd4
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/markitdown/_markitdown.py
View file

@ -685,12 +685,13 @@ class PdfConverter(DocumentConverter):
extension = kwargs.get("file_extension", "")
if extension.lower() != ".pdf":
return None
# return DocumentConverterResult(
# title=None,
# text_content=pdfminer.high_level.extract_text(local_path),
# )
text_content = pymupdf4llm.to_markdown(local_path, show_progress=False)
method = kwargs.get("method", "pdfminer")
if method == "pdfminer":
text_content = pdfminer.high_level.extract_text(local_path)
elif method == "pymupdf4llm":
text_content = pymupdf4llm.to_markdown(local_path, show_progress=False)
else:
return None # unknown method
return DocumentConverterResult(title=None, text_content=text_content)