From 524f6a8997b86d2f869ca491ec8fb40585b42ca4 Mon Sep 17 00:00:00 2001
From: Eugene <fogaprod@gmail.com>
Date: Thu, 5 Jun 2025 11:38:28 +0400
Subject: [PATCH] feat: Read supported file extensions from docling (#214)

Signed-off-by: Eugene <fogaprod@gmail.com>
---
 docling_serve/gradio_ui.py | 19 ++++++-------------
 1 file changed, 6 insertions(+), 13 deletions(-)

diff --git a/docling_serve/gradio_ui.py b/docling_serve/gradio_ui.py
index eb98a1b..66ff4cb 100644
--- 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,