restore function that was needlessly edited

Author: sebastianMindee

mindee/extraction/__init__.py

@@ -1,5 +1,6 @@
 from mindee.extraction.common.extracted_image import ExtractedImage
 from mindee.extraction.common.image_extractor import (
+    attach_image_as_new_file,
     extract_multiple_images_from_source,
 )
 from mindee.extraction.multi_receipts_extractor import multi_receipts_extractor

mindee/extraction/common/__init__.py

@@ -1,4 +1,5 @@
 from mindee.extraction.common.extracted_image import ExtractedImage
 from mindee.extraction.common.image_extractor import (
+    attach_image_as_new_file,
     extract_multiple_images_from_source,
 )

mindee/extraction/common/image_extractor.py

@@ -129,35 +129,36 @@ def load_pdf_doc(input_file: LocalInputSource) -> pdfium.PdfDocument:  # type: i
         input_file.file_object.seek(0)
         return pdfium.PdfDocument(input_file.file_object.read())
 
-    return attach_images_as_new_file([input_file.file_object])
+    return attach_image_as_new_file(input_file.file_object)
 
 
-def attach_images_as_new_file(  # type: ignore
-    input_buffer_list: List[BinaryIO],
+def attach_image_as_new_file(  # type: ignore
+    input_buffer: BinaryIO,
 ) -> pdfium.PdfDocument:
     """
-    Attaches a list of images as new pages in a PdfDocument object.
+    Attaches an image as a new page in a PdfDocument object.
 
-    :param input_buffer_list: List of images, represented as buffers.
+    :param input_buffer: Input buffer.
     :return: A PdfDocument handle.
     """
+    # Create a new page in the PdfDocument
+    input_buffer.seek(0)
+    image = Image.open(input_buffer)
+    image.convert("RGB")
+    image_buffer = io.BytesIO()
+    image.save(image_buffer, format="JPEG")
+
     pdf = pdfium.PdfDocument.new()
-    for input_buffer in input_buffer_list:
-        input_buffer.seek(0)
-        image = Image.open(input_buffer)
-        image.convert("RGB")
-        image_buffer = io.BytesIO()
-        image.save(image_buffer, format="JPEG")
-
-        image_pdf = pdfium.PdfImage.new(pdf)
-        image_pdf.load_jpeg(image_buffer)
-        width, height = image_pdf.get_size()
-
-        matrix = pdfium.PdfMatrix().scale(width, height)
-        image_pdf.set_matrix(matrix)
-
-        page = pdf.new_page(width, height)
-        page.insert_obj(image_pdf)
-        page.gen_content()
-        image.close()
+
+    image_pdf = pdfium.PdfImage.new(pdf)
+    image_pdf.load_jpeg(image_buffer)
+    width, height = image_pdf.get_size()
+
+    matrix = pdfium.PdfMatrix().scale(width, height)
+    image_pdf.set_matrix(matrix)
+
+    page = pdf.new_page(width, height)
+    page.insert_obj(image_pdf)
+    page.gen_content()
+    image.close()
     return pdf