Update app.py

app.py

@@ -95,21 +95,22 @@ def merge_pdfs(reordered_pages, uploaded_file):
     merger = PdfMerger()
     uploaded_file.seek(0)
     doc = fitz.open(stream=uploaded_file.read(), filetype="pdf")
+    temp_files = []
 
     for page_data in reordered_pages:
-      page = doc.load_page(page_data['page_index'])
-      page.set_rotation(page_data['rotation'])
-      pix = page.get_pixmap(alpha=False)
-      pil_image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-      temp_pdf_page = BytesIO()
-      pil_image.save(temp_pdf_page, "png")
-      temp_pdf_page.seek(0)
-      merger.append(temp_pdf_page)
-
+        page = doc.load_page(page_data['page_index'])
+        page.set_rotation(page_data['rotation'])
+        temp_pdf_page = tempfile.NamedTemporaryFile(suffix=".pdf", delete=False)
+        page.get_pixmap().save(temp_pdf_page.name)
+        temp_files.append(temp_pdf_page.name)
+        merger.append(temp_pdf_page.name)
 
     output_filename = "combined_document.pdf"
     with open(output_filename, "wb") as output_file:
-        merger.write(output_file)
+      merger.write(output_file)
+
+    for file in temp_files:
+      os.remove(file)
     return output_filename
 
 def download_file(output_file, file_name, mime_type):