Update app.py

app.py

@@ -1,30 +1,64 @@
 import streamlit as st
 from PyPDF2 import PdfMerger
+import fitz  # PyMuPDF
+from PIL import Image
+from io import BytesIO
 import os
+from streamlit_sortables import sort_items
 
-# Streamlit app layout
-st.title("PDF Combiner App")
-st.write("Upload individual PDF pages to combine them into one PDF document.")
+# Function to generate thumbnail from PDF
+def get_pdf_thumbnail(uploaded_file):
+    doc = fitz.open(stream=uploaded_file.read(), filetype="pdf")
+    page = doc.load_page(0)
+    pix = page.get_pixmap()
+    img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+    img.thumbnail((100, 140))
+    return img
+
+# App layout
+st.title("PDF Combiner with Preview & Reordering")
+st.write("Upload individual PDF pages, visualize them, reorder, and merge into a single PDF.")
 
 uploaded_files = st.file_uploader("Upload PDF pages", type="pdf", accept_multiple_files=True)
 
 if uploaded_files:
-    merger = PdfMerger()
+    # Generate thumbnails for each PDF
+    thumbnails = []
+    filenames = []
+
+    for file in uploaded_files:
+        thumbnails.append(get_pdf_thumbnail(file))
+        filenames.append(file.name)
+
+    # Display thumbnails with filenames for reordering
+    st.write("**Drag and drop to reorder the PDFs:**")
+    reordered_filenames = sort_items(filenames, labels=[file.name for file in uploaded_files])
+
+    # Map the filenames back to the corresponding files
+    reordered_files = [uploaded_files[filenames.index(name)] for name in reordered_filenames]
+
+    # Display the thumbnails in the new order
+    st.write("**Preview of selected order:**")
+    cols = st.columns(len(reordered_files))
+    for idx, file in enumerate(reordered_files):
+        with cols[idx]:
+            st.image(get_pdf_thumbnail(file), caption=file.name, use_column_width=True)
 
-    for uploaded_file in uploaded_files:
-        st.write(f"Adding: {uploaded_file.name}")
-        merger.append(uploaded_file)
+    # Merge PDFs in the specified order
+    if st.button("Merge PDFs"):
+        merger = PdfMerger()
+        for file in reordered_files:
+            merger.append(file)
 
-    # Save the combined PDF to a temporary file
-    output_filename = "combined_document.pdf"
-    with open(output_filename, "wb") as output_file:
-        merger.write(output_file)
+        output_filename = "combined_document.pdf"
+        with open(output_filename, "wb") as output_file:
+            merger.write(output_file)
 
-    st.success("PDF pages combined successfully!")
+        st.success("PDF pages combined successfully!")
 
-    # Provide download link
-    with open(output_filename, "rb") as f:
-        st.download_button("Download Combined PDF", f, file_name=output_filename, mime="application/pdf")
+        # Provide download link
+        with open(output_filename, "rb") as f:
+            st.download_button("Download Combined PDF", f, file_name=output_filename, mime="application/pdf")
 
-    # Clean up
-    os.remove(output_filename)
+        # Clean up
+        os.remove(output_filename)