Upload 2 files

app.py

@@ -0,0 +1,59 @@
+import gradio as gr
+from PyPDF2 import PdfReader, PdfWriter
+import tempfile
+import os
+import atexit
+
+def merge_pdfs(pdf_files, order):
+    pdf_writer = PdfWriter()
+    
+    # Sort the PDF files based on the specified order
+    sorted_pdfs = [pdf for _, pdf in sorted(zip(order, pdf_files))]
+    
+    # Read and add each PDF file to the writer in the specified order
+    for pdf in sorted_pdfs:
+        pdf_reader = PdfReader(pdf.name)
+        for page in pdf_reader.pages:
+            pdf_writer.add_page(page)
+    
+    # Create a temporary file for the merged PDF
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as temp_file:
+        pdf_writer.write(temp_file)
+        temp_file_path = temp_file.name
+    
+    return temp_file_path
+
+# Create Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# PDF Merger and Viewer")
+    pdf_input = gr.File(label="Upload PDF Files to Merge", file_types=[".pdf"], file_count="multiple")
+    order_input = gr.Textbox(label="Enter the order of PDFs (comma-separated numbers, e.g., 2,1,3)", placeholder="1,2,3,...")
+    merge_button = gr.Button("Merge PDFs")
+    merged_result = gr.File(label="Download Merged PDF")
+    
+    # Set button action to merge PDFs and then display the first page
+    def merge_and_preview(pdf_files, order):
+        if not order:
+            order = list(range(1, len(pdf_files) + 1))
+        else:
+            order = [int(x.strip()) for x in order.split(',')]
+        
+        if len(order) != len(pdf_files):
+            return gr.Error("The number of order values must match the number of uploaded PDFs.")
+        
+        merged_pdf_path = merge_pdfs(pdf_files, order)
+        return merged_pdf_path
+    
+    merge_button.click(merge_and_preview, inputs=[pdf_input, order_input], outputs=[merged_result])
+
+# Launch the Gradio app
+demo.launch()
+
+# Clean up temporary files
+def cleanup_temp_files():
+    temp_dir = tempfile.gettempdir()
+    for filename in os.listdir(temp_dir):
+        if filename.endswith('.pdf'):
+            os.remove(os.path.join(temp_dir, filename))
+
+atexit.register(cleanup_temp_files)

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+PyPDF2