add markdown support

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
-import os
 from pdfitdown.pdfconversion import Converter
 import fitz
 from typing import List
 from PIL import Image
 from loadimg import load_img
 import io
-
+from docling.document_converter import DocumentConverter
 
 converter = Converter()
+docling_converter = DocumentConverter()
 
 def convert_file_to_pdf(filename:str) -> str:
     """
@@ -52,6 +52,17 @@ def convert_file_to_img(image_file:str=None,txt:str="") -> List[Image.Image] :
     return img_list
     
     
+def convert_file_to_markdown(filename:str) -> str:
+    """
+    Converts a file to markdown format.
+    Args:
+        filename: str
+            The path to the file to be converted.
+    Returns:
+        str: The markdown representation of the file.
+    """
+    result = converter.convert(filename)
+    return result.document.export_to_markdown()
 
 
 # Create individual interfaces
@@ -63,7 +74,7 @@ file_to_pdf = gr.Interface(
     description="Convert your files to PDF format"
 )
 
-image_to_pdf = gr.Interface(
+file_to_image = gr.Interface(
     fn=convert_file_to_img,
     inputs=[gr.File(label="Upload Image"),gr.Textbox(label="base64, url")],
     outputs=gr.Gallery(label="Converted Images"),
@@ -71,10 +82,18 @@ image_to_pdf = gr.Interface(
     description="Convert your images to an image format"
 )
 
+file_to_markdown = gr.Interface(
+    fn=convert_file_to_markdown,
+    inputs=gr.File(label="Upload File"),
+    outputs=gr.Textbox(label="Converted Markdown"),
+    title="File to Markdown Converter",
+    description="Convert your files to markdown format"
+)
+
 # Create tabbed interface
 demo = gr.TabbedInterface(
-    [file_to_pdf, image_to_pdf],
-    ["File to PDF", "File to Image"],
+    [file_to_pdf, file_to_image, file_to_markdown],
+    ["File to PDF", "File to Image", "File to Markdown"],
 )
 
 if __name__ == "__main__":

requirements.txt

@@ -7,4 +7,5 @@ python-docx
 gradio
 python-pptx
 pdfitdown
-loadimg
+loadimg
+docling