update

app.py

@@ -2,25 +2,24 @@ import gradio as gr
 import os
 from pdfitdown.pdfconversion import Converter
 
-converter = Converter(reader="docling")
+converter = Converter()
 
-def convert_readme_to_pdf(markdown_file) -> str:
+def convert_file_to_pdf(filename:str) -> str:
     """
     Converts a markdown file to PDF format.
 
     Args:
-        markdown_file: A file object representing the markdown file to be converted.
-                      Expected to have a 'name' attribute containing the file path.
+        filename: str
+            The path to the markdown file to be converted.
 
     Returns:
-        str: Path to the generated PDF file. The output filename is created by 
-             replacing the '.md' extension of the input file with '.pdf'.
+        str: The file path of the generated PDF file.
     """
-    output_path = markdown_file.name.replace('.md', '.pdf')
-    converter.convert(markdown_file.name, output_path)
+    output_path = filename.name.rsplit('.', 1)[0] + '.pdf'
+    converter.convert(filename.name, output_path)
     return output_path
 
-def convert_image_to_pdf(image_file) -> str:
+def convert_file_to_img(image_file) -> str:
     """
     Convert an image file to PDF format.
 
@@ -40,16 +39,16 @@ def convert_image_to_pdf(image_file) -> str:
 
 
 # Create individual interfaces
-readme_to_pdf = gr.Interface(
-    fn=convert_readme_to_pdf,
-    inputs=gr.File(label="Upload README/Markdown file", file_types=[".md"]),
+file_to_pdf = gr.Interface(
+    fn=convert_file_to_pdf,
+    inputs=gr.File(label="Upload README/Markdown file"),
     outputs=gr.File(label="Converted PDF"),
-    title="README to PDF Converter",
-    description="Convert your markdown files to PDF format"
+    title="File to PDF Converter",
+    description="Convert your files to PDF format"
 )
 
 image_to_pdf = gr.Interface(
-    fn=convert_image_to_pdf,
+    fn=convert_file_to_img,
     inputs=gr.File(label="Upload Image", file_types=["image"]),
     outputs=gr.File(label="Converted PDF"),
     title="Image to PDF Converter",
@@ -58,8 +57,8 @@ image_to_pdf = gr.Interface(
 
 # Create tabbed interface
 demo = gr.TabbedInterface(
-    [readme_to_pdf, image_to_pdf],
-    ["README to PDF", "Image to PDF"],
+    [file_to_pdf, image_to_pdf],
+    ["File to PDF", "File to Image"],
 )
 
 if __name__ == "__main__":

requirements.txt

@@ -6,4 +6,5 @@ pdfplumber
 python-docx
 gradio
 python-pptx
-pdfitdown
+pdfitdown
+loadimg