Update app.py

app.py

@@ -18,7 +18,7 @@ class MultiDocMerger:
             f.write(file.read())
         self.documents.append(file_path)
         self._extract_pages(file_path)
-        return self.documents
+        return self._generate_preview()
 
     def _extract_pages(self, file_path):
         reader = PdfReader(file_path)
@@ -32,16 +32,16 @@ class MultiDocMerger:
 
     def rearrange_documents(self, order):
         self.documents = [self.documents[i] for i in order]
-        return self.documents
+        return self._generate_preview()
 
     def rearrange_pages(self, order):
         self.pages = [self.pages[i] for i in order]
-        return self.pages
+        return self._generate_preview()
 
     def delete_page(self, index):
         if 0 <= index < len(self.pages):
             del self.pages[index]
-        return self.pages
+        return self._generate_preview()
 
     def add_blank_page(self):
         blank_path = os.path.join(self.temp_dir, f"blank_{len(self.pages)}.pdf")
@@ -49,7 +49,7 @@ class MultiDocMerger:
         can.showPage()
         can.save()
         self.pages.append(blank_path)
-        return self.pages
+        return self._generate_preview()
 
     def generate_index(self):
         index_content = "Index Page\n\n"
@@ -72,43 +72,46 @@ class MultiDocMerger:
             output.write(f)
         return output_path
 
+    def _generate_preview(self):
+        return [os.path.basename(page) for page in self.pages]
+
     def cleanup(self):
         shutil.rmtree(self.temp_dir)
 
 merger = MultiDocMerger()
 
-def upload_files(files):
-    paths = []
-    for file in files:
-        paths = merger.add_document(file)
-    return paths
-
-def merge_and_export():
-    return merger.export_merged_pdf()
-
-def delete_page(index):
-    return merger.delete_page(index)
-
-def add_blank_page():
-    return merger.add_blank_page()
-
-def rearrange_pages(order):
-    return merger.rearrange_pages(order)
-
 with gr.Blocks() as demo:
     gr.Markdown("# Multi-Document Merger and Organizer")
     with gr.Column():
         file_upload = gr.File(label="Upload Files", file_count="multiple")
-        rearrange_order = gr.Textbox(label="Rearrange Pages (comma-separated indices)")
+        preview_gallery = gr.Gallery(label="Document Preview and Reordering", interactive=True)
         delete_index = gr.Number(label="Delete Page Index")
         blank_button = gr.Button("Add Blank Page")
         merge_button = gr.Button("Merge and Export PDF")
         output_file = gr.File()
 
-    file_upload.upload(upload_files, inputs=file_upload, outputs=[])
-    delete_index.submit(delete_page, inputs=delete_index, outputs=[])
-    blank_button.click(add_blank_page, outputs=[])
-    rearrange_order.submit(rearrange_pages, inputs=rearrange_order, outputs=[])
+    def upload_files(files):
+        paths = []
+        for file in files:
+            paths = merger.add_document(file)
+        return paths
+
+    def merge_and_export():
+        return merger.export_merged_pdf()
+
+    def delete_page(index):
+        return merger.delete_page(index)
+
+    def add_blank_page():
+        return merger.add_blank_page()
+
+    def reorder_pages(order):
+        return merger.rearrange_pages(order)
+
+    file_upload.upload(upload_files, inputs=file_upload, outputs=preview_gallery)
+    delete_index.submit(delete_page, inputs=delete_index, outputs=preview_gallery)
+    blank_button.click(add_blank_page, outputs=preview_gallery)
+    preview_gallery.change(reorder_pages, inputs=preview_gallery, outputs=preview_gallery)
     merge_button.click(merge_and_export, outputs=output_file)
 
-demo.launch()
+demo.launch()