33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from docling.document_converter import (
|
|
DocumentConverter,
|
|
PdfFormatOption,
|
|
PowerpointFormatOption,
|
|
WordFormatOption,
|
|
)
|
|
from docling.datamodel.pipeline_options import PdfPipelineOptions
|
|
from docling.datamodel.base_models import InputFormat
|
|
|
|
|
|
class DoclingService:
|
|
def __init__(self):
|
|
self.pipeline_options = PdfPipelineOptions()
|
|
self.pipeline_options.do_ocr = False
|
|
|
|
self.converter = DocumentConverter(
|
|
allowed_formats=[InputFormat.PPTX, InputFormat.PDF, InputFormat.DOCX],
|
|
format_options={
|
|
InputFormat.DOCX: WordFormatOption(
|
|
pipeline_options=self.pipeline_options,
|
|
),
|
|
InputFormat.PPTX: PowerpointFormatOption(
|
|
pipeline_options=self.pipeline_options,
|
|
),
|
|
InputFormat.PDF: PdfFormatOption(
|
|
pipeline_options=self.pipeline_options,
|
|
),
|
|
},
|
|
)
|
|
|
|
def parse_to_markdown(self, file_path: str) -> str:
|
|
result = self.converter.convert(file_path)
|
|
return result.document.export_to_markdown()
|