# pyright: ignore
from docling.datamodel.base_models import InputFormat
from docling.document_converter import DocumentConverter, PdfFormatOption
from docling.pipeline.vlm_pipeline import VlmPipeline

# Create converter with default VLM (Granite-Docling)
converter = DocumentConverter(
    format_options={
        InputFormat.PDF: PdfFormatOption(
            pipeline_cls=VlmPipeline,
        ),
    }
)

# Convert your PDF
pdf_path = "documents/TMUS_992_Q423.pdf"  # Change this to your PDF path
result = converter.convert(pdf_path)
document = result.document

# Save as markdown
output_path = "converted_document.md"
with open(output_path, "w") as f:
    f.write(document.export_to_markdown())

print(f"Converted {pdf_path} to {output_path}")
print(f"Found {len(document.tables)} tables")