image2PDF.py

import tkinter as tk
from tkinter import filedialog, messagebox, simpledialog
from PIL import Image
import img2pdf
from pathlib import Path


def select_images():
    """Open file dialog to select one or more image files"""
    root = tk.Tk()
    root.withdraw()  # Hide the main window

    file_paths = filedialog.askopenfilenames(
        title="Select Image Files",
        filetypes=[
            ("Image files", "*.jpg *.jpeg *.png"),
            ("JPEG files", "*.jpg *.jpeg"),
            ("PNG files", "*.png"),
            ("All files", "*.*")
        ]
    )

    return list(file_paths) if file_paths else None


def get_dpi():
    """Prompt user for DPI value"""
    root = tk.Tk()
    root.withdraw()

    dpi = simpledialog.askinteger(
        "DPI Input",
        "Enter the DPI at which the images were scanned:",
        initialvalue=300,
        minvalue=72,
        maxvalue=2400
    )

    return dpi


def save_pdf():
    """Open save dialog for PDF output"""
    root = tk.Tk()
    root.withdraw()

    output_path = filedialog.asksaveasfilename(
        title="Save PDF As",
        defaultextension=".pdf",
        filetypes=[("PDF files", "*.pdf"), ("All files", "*.*")]
    )

    return output_path if output_path else None


def convert_images_to_pdf(image_paths, dpi, output_path):
    """Convert images to PDF-A format"""
    try:
        # Convert images to PDF with specified DPI
        # img2pdf will use the DPI metadata from images or the specified value
        with open(output_path, "wb") as f:
            # Pass DPI as a tuple for x and y resolution
            pdf_bytes = img2pdf.convert(
                image_paths,
                dpi=(dpi, dpi)
            )
            f.write(pdf_bytes)

        return True
    except Exception as e:
        messagebox.showerror("Error", f"Failed to create PDF:\n{str(e)}")
        return False


def main():
    """Main application flow"""
    # Step 1: Select images
    image_paths = select_images()

    if not image_paths:
        messagebox.showinfo("Cancelled", "No images selected. Exiting.")
        return

    # Step 2: Get DPI
    dpi = get_dpi()

    if not dpi:
        messagebox.showinfo("Cancelled", "No DPI provided. Exiting.")
        return

    # Step 3: Select output location
    output_path = save_pdf()

    if not output_path:
        messagebox.showinfo("Cancelled", "No output file selected. Exiting.")
        return

    # Convert images to PDF
    success = convert_images_to_pdf(image_paths, dpi, output_path)

    if success:
        messagebox.showinfo(
            "Success",
            f"PDF created successfully!\n\n"
            f"Images processed: {len(image_paths)}\n"
            f"Output: {output_path}"
        )


if __name__ == "__main__":
    main()