updated the app.py

app.py

@@ -7,8 +7,8 @@ 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))]
+    # Sort the PDF files based on the specified order, skipping files marked with '0'
+    sorted_pdfs = [pdf_files[i-1] for i in order if i != 0]
     
     # Read and add each PDF file to the writer in the specified order
     for pdf in sorted_pdfs:
@@ -16,10 +16,10 @@ def merge_pdfs(pdf_files, order):
         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:
+    # Create a named temporary file for the merged PDF as 'combine.pdf'
+    temp_file_path = os.path.join(tempfile.gettempdir(), "combine.pdf")
+    with open(temp_file_path, 'wb') as temp_file:
         pdf_writer.write(temp_file)
-        temp_file_path = temp_file.name
     
     return temp_file_path
 
@@ -27,20 +27,29 @@ def merge_pdfs(pdf_files, order):
 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,...")
+    order_input = gr.Textbox(label="Enter the order of PDFs as comma-separated numbers, skip the number if you want to skip the file", placeholder="1,2,3,... or 3,1,2")
     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):
+        n = len(pdf_files)
+        
         if not order:
-            order = list(range(1, len(pdf_files) + 1))
+            # Default to natural order if order is empty
+            order = list(range(1, n + 1))
         else:
-            order = [int(x.strip()) for x in order.split(',')]
+            try:
+                # Convert the input string to a list of integers
+                order = [int(x.strip()) for x in order.split(',')]
+            except ValueError:
+                return gr.Error("Invalid order format. Ensure it is comma-separated numbers.")
         
-        if len(order) != len(pdf_files):
-            return gr.Error("The number of order values must match the number of uploaded PDFs.")
+        # Ensure the order does not reference non-existing files
+        if any(i < 0 or i > n for i in order):
+            return gr.Error(f"Order values must be between 0 and {n} (0 means to skip the file).")
         
+        # Adjust order by removing '0' entries and merging only the specified files
         merged_pdf_path = merge_pdfs(pdf_files, order)
         return merged_pdf_path