feat: Read supported file extensions from docling (#214)

Signed-off-by: Eugene <fogaprod@gmail.com>
2025-11-29 00:23:36 +00:00 · 2025-06-05 11:38:28 +04:00
parent 9ccf8e3b5e
commit 524f6a8997
1 changed files with 6 additions and 13 deletions
--- a/docling_serve/gradio_ui.py
+++ b/docling_serve/gradio_ui.py
@@ -1,5 +1,6 @@
 import base64
 import importlib
+import itertools
 import json
 import logging
 import ssl
@@ -12,6 +13,7 @@ import certifi
 import gradio as gr
 import httpx

+from docling.datamodel.base_models import FormatToExtensions
 from docling.datamodel.pipeline_options import (
    PdfBackend,
    PdfPipeline,
@@ -545,19 +547,10 @@ with gr.Blocks(
                    elem_id="file_input_zone",
                    label="Upload File",
                    file_types=[
-                        ".pdf",
-                        ".docx",
-                        ".pptx",
-                        ".html",
-                        ".xlsx",
-                        ".json",
-                        ".asciidoc",
-                        ".txt",
-                        ".md",
-                        ".jpg",
-                        ".jpeg",
-                        ".png",
-                        ".gif",
+                        f".{v}"
+                        for v in itertools.chain.from_iterable(
+                            FormatToExtensions.values()
+                        )
                    ],
                    file_count="multiple",
                    scale=4,