Added functionality to remove pages from a PDF

app.py

@@ -86,6 +86,44 @@ def split_pdf(file_path, start_page, end_page):
     zipf.close()
     return zip_path, start_page + 1, end_page + 1
 
+def remove_pages(file, pages_to_remove):
+    if file is None:
+        return "❌ No file uploaded.", None
+
+    input_pdf = PdfReader(open(file.name, "rb"))
+    writer = PdfWriter()
+    total_pages = len(input_pdf.pages)
+
+    # Parse page numbers
+    pages_to_remove = pages_to_remove.replace(" ", "")
+    remove_set = set()
+
+    try:
+        for part in pages_to_remove.split(","):
+            if "-" in part:
+                start, end = map(int, part.split("-"))
+                remove_set.update(range(start, end + 1))
+            else:
+                remove_set.add(int(part))
+    except:
+        return "❌ Invalid page format.", None
+
+    # Keep valid pages only
+    remove_set = {p for p in remove_set if 0 <= p < total_pages}
+
+    for i in range(total_pages):
+        if i not in remove_set:
+            writer.add_page(input_pdf.pages[i])
+
+    output_dir = tempfile.mkdtemp()
+    output_path = os.path.join(output_dir, "pages_removed.pdf")
+
+    with open(output_path, "wb") as f:
+        writer.write(f)
+
+    status = f"✅ Removed pages: {sorted(remove_set)}"
+    return status, output_path
+
 def process_pdf(file, start_page, end_page):
     if file is None:
         return "❌ No file uploaded.", None
@@ -152,6 +190,31 @@ with gr.Blocks(title="PDF Utility") as demo:
                 inputs=[pdf_uploads],
                 outputs=[merge_status, merged_file]
             )
+            
+        with gr.TabItem("Remove Pages"):
+            gr.Markdown(
+                "Remove one or more pages from a PDF.\n\n"
+                "**Examples:** `2`, `1,3,5`, `2-6`, `1,3,5-7`\n\n"
+                "⚠️ Page numbers are **0-based**."
+            )
+
+            remove_file = gr.File(label="Upload PDF", file_types=[".pdf"])
+
+            pages_input = gr.Textbox(
+                label="Pages to remove",
+                placeholder="e.g. 1,3,5-7"
+            )
+
+            remove_button = gr.Button("🗑 Remove Pages")
+
+            remove_status = gr.Textbox(label="Status", lines=2)
+            removed_pdf = gr.File(label="Download PDF")
+
+            remove_button.click(
+                fn=remove_pages,
+                inputs=[remove_file, pages_input],
+                outputs=[remove_status, removed_pdf]
+            )
+
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()