Upload app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from PyPDF2 import PdfReader, PdfWriter, PageObject
+from pdf2image import convert_from_path
+import img2pdf
 import tempfile
 import os
 import atexit
@@ -33,50 +35,125 @@ def merge_pdfs(pdf_files, order, start_on_odd=False):
     
     return temp_file_path
 
+def pdf_to_images(pdf_file):
+    # Convert PDF to images
+    temp_dir = tempfile.mkdtemp()
+    images = convert_from_path(pdf_file.name, output_folder=temp_dir, fmt='jpg')
+    image_paths = []
+
+    for i, image in enumerate(images):
+        image_path = os.path.join(temp_dir, f"page_{i + 1}.jpg")
+        image.save(image_path, "JPEG")
+        image_paths.append(image_path)
+
+    return image_paths
+
+def images_to_pdf(image_files):
+    # Convert images to a single PDF
+    temp_file_path = os.path.join(tempfile.gettempdir(), "images_to_pdf.pdf")
+    with open(temp_file_path, "wb") as pdf_file:
+        pdf_file.write(img2pdf.convert([image.name for image in image_files]))
+    return temp_file_path
+
 # Create Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# PDF Merger")
-    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 as comma-separated numbers, skip the number if you want to skip the file", placeholder="1,2,3,... or 3,1,2")
-    
-    with gr.Row():
-        merge_button = gr.Button("Merge PDFs (Normal)")
-        merge_odd_button = gr.Button("Merge PDFs (Each PDF starts on odd page)")
-    
-    merged_result = gr.File(label="Download Merged PDF")
-    
-    def merge_and_preview(pdf_files, order, start_on_odd=False):
-        n = len(pdf_files)
-        
-        if not order:
-            # Default to natural order if order is empty
-            order = list(range(1, n + 1))
-        else:
-            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.")
-        
-        # 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).")
-        
-        # Merge PDFs with the specified start_on_odd option
-        merged_pdf_path = merge_pdfs(pdf_files, order, start_on_odd)
-        return merged_pdf_path
-    
-    merge_button.click(
-        lambda *args: merge_and_preview(*args, False),
-        inputs=[pdf_input, order_input],
-        outputs=[merged_result]
-    )
-    
-    merge_odd_button.click(
-        lambda *args: merge_and_preview(*args, True),
-        inputs=[pdf_input, order_input],
-        outputs=[merged_result]
-    )
+    gr.Markdown("# PDF Merger and Converter")
+
+    with gr.Tabs():
+        with gr.TabItem("PDF Merger"):
+            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 as comma-separated numbers, skip the number if you want to skip the file", placeholder="1,2,3,... or 3,1,2")
+            
+            with gr.Row():
+                merge_button = gr.Button("Merge PDFs (Normal)")
+                merge_odd_button = gr.Button("Merge PDFs (Each PDF starts on odd page)")
+            
+            merged_result = gr.File(label="Download Merged PDF")
+
+            def merge_and_preview(pdf_files, order, start_on_odd=False):
+                n = len(pdf_files)
+                
+                if not order:
+                    # Default to natural order if order is empty
+                    order = list(range(1, n + 1))
+                else:
+                    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.")
+                
+                # 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).")
+                
+                # Merge PDFs with the specified start_on_odd option
+                merged_pdf_path = merge_pdfs(pdf_files, order, start_on_odd)
+                return merged_pdf_path
+
+            merge_button.click(
+                lambda *args: merge_and_preview(*args, False),
+                inputs=[pdf_input, order_input],
+                outputs=[merged_result]
+            )
+            
+            merge_odd_button.click(
+                lambda *args: merge_and_preview(*args, True),
+                inputs=[pdf_input, order_input],
+                outputs=[merged_result]
+            )
+
+        with gr.TabItem("PDF to Image Converter"):
+            single_pdf_input = gr.File(label="Upload PDF File to Convert", file_types=[".pdf"], file_count="single")
+            image_output = gr.Gallery(label="Converted Images (JPG)", show_label=True)
+
+            def convert_pdf_to_images(pdf_file):
+                return pdf_to_images(pdf_file)
+
+            single_pdf_input.change(
+                convert_pdf_to_images,
+                inputs=[single_pdf_input],
+                outputs=[image_output]
+            )
+
+        with gr.TabItem("Image to PDF Converter"):
+            image_input = gr.File(label="Upload Images to Convert to PDF", file_types=[".jpg", ".png"], file_count="multiple")
+            order_option = gr.Radio(label="Select Order Type", choices=["Ordered", "Reverse", "Custom"], value="Ordered")
+            custom_order_input = gr.Textbox(label="Enter custom order (comma-separated indices)", visible=False)
+            image_gallery = gr.Gallery(label="Images Preview (Arrange Order)", show_label=True)
+            pdf_result = gr.File(label="Download PDF")
+
+            def update_custom_order_visibility(order_type):
+                return gr.update(visible=(order_type == "Custom"))
+
+            def sort_images(order_type, custom_order, images):
+                if order_type == "Reverse":
+                    return images[::-1]
+                elif order_type == "Custom":
+                    try:
+                        indices = [int(i.strip()) - 1 for i in custom_order.split(',')]
+                        return [images[i] for i in indices]
+                    except (ValueError, IndexError):
+                        return gr.Error("Invalid custom order. Ensure all indices are valid and within range.")
+                return images
+
+            order_option.change(
+                update_custom_order_visibility,
+                inputs=[order_option],
+                outputs=[custom_order_input]
+            )
+
+            gr.Button("Preview Sorted Images").click(
+                lambda order_type, custom_order, images: sort_images(order_type, custom_order, images),
+                inputs=[order_option, custom_order_input, image_input],
+                outputs=[image_gallery]
+            )
+
+            gr.Button("Generate PDF").click(
+                lambda order_type, custom_order, images: images_to_pdf(sort_images(order_type, custom_order, images)),
+                inputs=[order_option, custom_order_input, image_input],
+                outputs=[pdf_result]
+            )
 
 # Launch the Gradio app
 demo.launch()
@@ -85,7 +162,7 @@ demo.launch()
 def cleanup_temp_files():
     temp_dir = tempfile.gettempdir()
     for filename in os.listdir(temp_dir):
-        if filename.endswith('.pdf'):
+        if filename.endswith('.pdf') or filename.endswith('.jpg'):
             os.remove(os.path.join(temp_dir, filename))
 
-atexit.register(cleanup_temp_files)
+atexit.register(cleanup_temp_files)