Update app.py

app.py

@@ -3,10 +3,13 @@ from PyPDF2 import PdfMerger
 from pdf2image import convert_from_path
 from google.cloud import vision
 from google.oauth2 import service_account
-import io
+import fitz  # PyMuPDF
+from PIL import Image
+from io import BytesIO
 import os
 import tempfile
 import json
+from streamlit_sortables import sort_items
 
 # Load Google Cloud Vision credentials from secret
 credentials_json = os.getenv("GOOGLE_CREDENTIALS_JSON")
@@ -50,33 +53,70 @@ def process_pdf(file):
 
     return text
 
+# Function to generate thumbnail from PDF
+def get_pdf_thumbnail(uploaded_file):
+    """Generates a thumbnail image of the first page of a PDF."""
+    uploaded_file.seek(0)
+    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
+
 # Function to merge PDFs
-def merge_pdfs(uploaded_files):
+def merge_pdfs(reordered_files):
     """Merges multiple PDFs into one."""
     merger = PdfMerger()
-    for uploaded_file in uploaded_files:
-        uploaded_file.seek(0)
-        merger.append(uploaded_file)
+    for file in reordered_files:
+        file.seek(0)
+        merger.append(file)
     output_filename = "combined_document.pdf"
     with open(output_filename, "wb") as output_file:
         merger.write(output_file)
     return output_filename
 
-# Streamlit Sidebar Navigation
+# Sidebar Navigation
 st.sidebar.title("Tool Selector")
 selection = st.sidebar.radio("Choose a tool:", ["PDF Combiner", "PDF Transcriber"])
 
-# PDF Combiner Tool
+# PDF Combiner with Preview and Reordering
 if selection == "PDF Combiner":
-    st.title("PDF Combiner Tool")
-    st.write("Upload multiple PDF files to combine them into a single document.")
-    uploaded_files = st.file_uploader("Upload PDF files", type="pdf", accept_multiple_files=True)
-    if uploaded_files and st.button("Merge PDFs"):
-        output_file = merge_pdfs(uploaded_files)
-        st.success("PDFs combined successfully!")
-        with open(output_file, "rb") as f:
-            st.download_button("Download Combined PDF", f, file_name=output_file, mime="application/pdf")
-        os.remove(output_file)
+    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:
+        # Generate thumbnails and filenames for each uploaded 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)
+
+        # 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_container_width=True)
+
+        # Merge PDFs in the specified order
+        if st.button("Merge PDFs"):
+            output_file = merge_pdfs(reordered_files)
+            st.success("PDF pages combined successfully!")
+            with open(output_file, "rb") as f:
+                st.download_button("Download Combined PDF", f, file_name=output_file, mime="application/pdf")
+            os.remove(output_file)
 
 # PDF Transcriber Tool
 elif selection == "PDF Transcriber":