update to v1.0.1

app.py

@@ -1,69 +1,34 @@
-from pdfitdown.pdfconversion import convert_to_pdf, convert_markdown_to_pdf
+from pdfconversion import Converter
 import warnings
 from typing import List
 import gradio as gr
+import os
 
 class FileNotConvertedWarning(Warning):
     """The file was not in one of the specified formats for conversion to PDF,thus it was not converted"""
 
 def to_pdf(files: List[str]) -> List[str]:
-    """
-    Converts various file formats to PDF.
-    
-    Args:
-        files: List of file paths to convert. Supports .docx, .pdf, .html, .pptx, 
-              .csv, .xml, and .md files.
-    
-    Returns:
-        List of paths to converted PDF files. For files already in PDF format, 
-        returns original path.
-    
-    Raises:
-        FileNotConvertedWarning: When file format is not supported.
-    """
     pdfs = []
-    for f in files:
-        if f.endswith(".docx"):
-            newfile = f.replace(".docx", ".pdf")
-            file_to_add = convert_to_pdf(f, newfile, newfile.split(".")[0])
-            pdfs.append(file_to_add)
-        elif f.endswith(".pdf"):
-            pdfs.append(f)
-        elif f.endswith(".html"):
-            newfile = f.replace(".html", ".pdf")
-            file_to_add = convert_to_pdf(f, newfile, newfile.split(".")[0])
-            pdfs.append(file_to_add)
-        elif f.endswith(".pptx"):
-            newfile = f.replace(".pptx", ".pdf")
-            file_to_add = convert_to_pdf(f, newfile, newfile.split(".")[0])
-            pdfs.append(file_to_add)
-        elif f.endswith(".csv"):
-            newfile = f.replace(".csv", ".pdf")
-            file_to_add = convert_to_pdf(f, newfile, newfile.split(".")[0])
-            pdfs.append(file_to_add)
-        elif f.endswith(".xml"):
-            newfile = f.replace(".xml", ".pdf")
-            file_to_add = convert_to_pdf(f, newfile, newfile.split(".")[0])
-            pdfs.append(file_to_add)
-        elif f.endswith(".md"):
-            newfile = f.replace(".md", ".pdf")
-            file_to_add = convert_markdown_to_pdf(f, newfile, newfile.split(".")[0])
-            pdfs.append(file_to_add)
+    converter = Converter()
+    for fl in files:
+        try:
+            outf = converter.convert(fl, fl.replace(os.path.splitext(fl)[1], ".pdf"))
+        except Exception as e:
+            warnings.warn(f"File {fl} not converted because of an error during the conversion: {e}", FileNotConvertedWarning)
         else:
-            warnings.warn(f"File {f} was not converted to PDF because its file format is not included in those that can be converted", FileNotConvertedWarning)
-            continue
+            pdfs.append(outf)
     return pdfs
 
-def convert(file: str) -> str:
-    files = [file]
+def convert_files(files: List[str]) -> List[str]:
     pdfs = to_pdf(files)
-    return pdfs[0]
+    return pdfs
 
 iface = gr.Interface(
-    fn=convert,
-    inputs=gr.File(label="Upload your file"),
-    outputs=gr.File(label="Converted PDF"),
+    fn=convert_files,
+    inputs=gr.File(label="Upload your file", file_count="multiple"),
+    outputs=gr.File(label="Converted PDF", file_count="multiple"),
     title="File to PDF Converter",
-    description="Upload a file in .docx, .pdf, .html, .pptx, .csv, .xml, or .md format, and get it converted to PDF."
+    description="Upload a file in .docx, .xlsx, .html, .pptx, .json, .csv, .xml, .md, .jpg/.jpeg, .png format, and get it converted to PDF."
 )
+
 iface.launch()