errantanomie commited on
Commit
30bda3d
·
verified ·
1 Parent(s): 13d07d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -8,6 +8,7 @@ from streamlit_sortables import sort_items
8
 
9
  # Function to generate thumbnail from PDF
10
  def get_pdf_thumbnail(uploaded_file):
 
11
  doc = fitz.open(stream=uploaded_file.read(), filetype="pdf")
12
  page = doc.load_page(0)
13
  pix = page.get_pixmap()
@@ -48,17 +49,18 @@ if uploaded_files:
48
  if st.button("Merge PDFs"):
49
  merger = PdfMerger()
50
  for file in reordered_files:
 
51
  merger.append(file)
52
-
53
  output_filename = "combined_document.pdf"
54
  with open(output_filename, "wb") as output_file:
55
  merger.write(output_file)
56
-
57
  st.success("PDF pages combined successfully!")
58
-
59
  # Provide download link
60
  with open(output_filename, "rb") as f:
61
  st.download_button("Download Combined PDF", f, file_name=output_filename, mime="application/pdf")
62
-
63
  # Clean up
64
  os.remove(output_filename)
 
8
 
9
  # Function to generate thumbnail from PDF
10
  def get_pdf_thumbnail(uploaded_file):
11
+ uploaded_file.seek(0) # Reset the file pointer to the beginning
12
  doc = fitz.open(stream=uploaded_file.read(), filetype="pdf")
13
  page = doc.load_page(0)
14
  pix = page.get_pixmap()
 
49
  if st.button("Merge PDFs"):
50
  merger = PdfMerger()
51
  for file in reordered_files:
52
+ file.seek(0) # Reset the file pointer before merging
53
  merger.append(file)
54
+
55
  output_filename = "combined_document.pdf"
56
  with open(output_filename, "wb") as output_file:
57
  merger.write(output_file)
58
+
59
  st.success("PDF pages combined successfully!")
60
+
61
  # Provide download link
62
  with open(output_filename, "rb") as f:
63
  st.download_button("Download Combined PDF", f, file_name=output_filename, mime="application/pdf")
64
+
65
  # Clean up
66
  os.remove(output_filename)