Update files/functions.py

files/functions.py

@@ -11,8 +11,7 @@ import pypdf
 from pypdf import PdfReader
 from pypdf.errors import PdfReadError
 
-import pdf2image
-from pdf2image import convert_from_path
+import pypdfium2 as pdfium
 import langdetect
 from langdetect import detect_langs
 
@@ -363,7 +362,19 @@ def pdf_to_images(uploaded_pdf):
             images = [Image.open(image_blank)]
         else:
             try:
-                images = convert_from_path(path_to_file, last_page=max_imgboxes)
+                # images = convert_from_path(path_to_file, last_page=max_imgboxes)
+
+                pdf = pdfium.PdfDocument(str(filename))
+                version = pdf.get_version()  # get the PDF standard version
+                n_pages = len(pdf)  # get the number of pages in the document
+                last_page = max_imgboxes
+                page_indices = [i for i in range(last_page)]  # pages until last_page
+                images = list(pdf.render(
+                    pdfium.PdfBitmap.to_pil,
+                    page_indices = page_indices,
+                    scale = 300/72,  # 300dpi resolution
+                ))
+                
                 num_imgs = len(images)
                 msg = f'The PDF "{filename}" was converted into {num_imgs} images.'
             except: