Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|