Update app.py

app.py

@@ -100,44 +100,35 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.TabItem("Split PDF"):
             inp = gr.File(label="PDF")
             ranges = gr.Textbox(label="Page ranges", placeholder="e.g. 1-3,6-8")
-            btn = gr.Button("Split")
-            out, err = gr.File(), gr.Textbox(visible=False)
-            btn.click(lambda f,r: (*split_pdf(f,r),), [inp,ranges],[out,err,err])
+            btn = gr.Button("Split PDF")
+            out, err = gr.File(label="Download Split PDF"), gr.Textbox(label="Error", visible=False)
+            btn.click(lambda f,r: (*split_pdf(f,r),), [inp,ranges], [out,err,err])
 
-                # Merge
+        # Merge
         with gr.TabItem("Merge PDF"):
-            files = gr.Files(label="Upload PDFs to merge (max 1 GB each)")
-            reorder = gr.Sortable([], label="Drag to reorder files by filename")
+            files = gr.Files(label="Upload & reorder PDFs (drag to reorder)")
+            btn_m = gr.Button("Merge PDFs")
+            out_m, err_m = gr.File(label="Download Merged PDF"), gr.Textbox(label="Error", visible=False)
 
-            def update_reorder(file_list):
-                return [os.path.basename(f.name) for f in file_list] if file_list else []
-            files.change(update_reorder, inputs=files, outputs=reorder)
-
-            btn_m = gr.Button("Merge PDFs", variant="primary")
-            out_m = gr.File(label="Download Merged PDF")
-            err_m = gr.Textbox(label="Error", interactive=False, visible=False)
-
-            def run_merge(file_list, order_list):
+            def run_merge(file_list):
                 if not file_list or len(file_list) < 2:
                     return None, "Please upload at least two PDFs.", True
-                name_to_file = {os.path.basename(f.name): f for f in file_list}
-                ordered_files = [name_to_file[name] for name in order_list if name in name_to_file]
-                out, err = merge_pdfs(ordered_files)
+                out, err = merge_pdfs(file_list)
                 return out, err, err is not None
 
-            btn_m.click(run_merge, [files, reorder], [out_m, err_m, err_m])
+            btn_m.click(run_merge, inputs=[files], outputs=[out_m, err_m, err_m])
 
         # Compress
         with gr.TabItem("Compress PDF"):
             comp_file = gr.File(label="PDF to compress")
             method = gr.Radio(choices=["Stream only","PikePDF"], value="PikePDF", label="Method")
             quality = gr.Slider(10,100, value=75, step=5, label="JPEG Quality")
-            btn_c = gr.Button("Compress")
-            out_c, err_c = gr.File(), gr.Textbox(visible=False)
-            def do_compress(f,m,q):
-                if m=="PikePDF": return compress_pdf_pike(f,q)
-                return compress_pdf_stream(f)
-            btn_c.click(do_compress, [comp_file,method,quality], [out_c,err_c,err_c])
+            btn_c = gr.Button("Compress PDF")
+            out_c, err_c = gr.File(label="Download Compressed PDF"), gr.Textbox(label="Error", visible=False)
+            btn_c.click(lambda f,m,q: compress_pdf_pike(f,q) if m=="PikePDF" else compress_pdf_stream(f),
+                        inputs=[comp_file,method,quality], outputs=[out_c,err_c,err_c])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT",7860)))
+ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT",7860)))