from pathlib import Path
from typing import Any, Dict

from docling.document_converter import DocumentConverter


class DocumentProcessor:

    SUPPORTED_EXTENSIONS = {
        ".pdf",
        ".docx",
        ".pptx",
        ".html",
        ".htm",
        ".md",
        ".txt",
    }

    def __init__(self):

        self.converter = DocumentConverter()

    def process(
        self,
        file_path: str,
    ) -> Dict[str, Any]:

        path = Path(file_path)

        extension = path.suffix.lower()

        if extension not in self.SUPPORTED_EXTENSIONS:

            raise ValueError(
                f"Unsupported file type: {extension}"
            )

        result = self.converter.convert(
            str(path)
        )

        document = result.document

        raw_json = document.export_to_dict()

        markdown = document.export_to_markdown()

        return {
            "document": document,
            "raw_json": raw_json,
            "markdown": markdown,
        }