insert_delete_pg_pdf.py

# main code
import fitz
import tkinter as tk
from tkinter import filedialog, messagebox
from PIL import Image, ImageTk, ExifTags
import os
import PyPDF2

class PDFEditorApp:

    def __init__(self, root):
        self.root = root
        self.root.title("PDF Editor")
        self.root.configure(bg="black")

        self.pdf_document = None
        self.current_page = 0
        self.temp_images = []

        self.create_gui()

    def create_gui(self):

        self.button_frame = tk.Frame(self.root, bg="black")
        self.button_frame.pack(side=tk.RIGHT, fill=tk.Y, padx=10, pady=10)
        self.root.bind("<Left>", lambda event: self.prev_page())
        self.root.bind("<Right>", lambda event: self.next_page())
        self.root.bind("<Return>", lambda event: self.go_to_page())

        self.prev_button = tk.Button(self.button_frame, text="Previous Page", command=self.prev_page, bg="white", fg="black")
        self.next_button = tk.Button(self.button_frame, text="Next Page", command=self.next_page, bg="white", fg="black")
        self.add_page_button = tk.Button(self.button_frame, text="Add Page", command=self.add_page, bg="white", fg="black")
        self.add_to_last_page_button = tk.Button(self.button_frame, text="Add to Last Page", command=self.add_to_last_page, bg="white", fg="black")
        self.delete_page_button = tk.Button(self.button_frame, text="Delete Page", command=self.delete_page, bg="white", fg="black")
        self.make_pdf_button = tk.Button(self.button_frame, text="Make PDF", command=self.save_pdf, bg="white", fg="black")

        self.prev_button.pack(fill=tk.X, pady=5)
        self.next_button.pack(fill=tk.X, pady=5)
        self.add_page_button.pack(fill=tk.X, pady=5)
        self.add_to_last_page_button.pack(fill=tk.X, pady=5)
        self.delete_page_button.pack(fill=tk.X, pady=5)
        self.make_pdf_button.pack(fill=tk.X, pady=5)

        self.go_to_page_button = tk.Button(self.button_frame, text="Go to Page", command=self.go_to_page, bg="white", fg="black")
        self.go_to_page_button.pack(side=tk.BOTTOM, fill=tk.X, pady=5)
        
        
        self.page_entry_var = tk.StringVar()
        self.page_entry = tk.Entry(self.button_frame, textvariable=self.page_entry_var, bg="white", fg="black")
        self.page_entry.pack(side=tk.BOTTOM, fill=tk.X, pady=5)

        self.page_entry_label = tk.Label(self.button_frame, text="Enter Page Number:", bg="white", fg="black")
        self.page_entry_label.pack(side=tk.BOTTOM, fill=tk.X, pady=5)
        

        self.page_number_label = tk.Label(self.button_frame, text="Page 1", bg="white", fg="black")
        self.page_number_label.pack(side=tk.BOTTOM, fill=tk.X, pady=5)

        self.canvas = tk.Canvas(self.root, bg="black")
        self.canvas.pack(side=tk.LEFT, expand=True, fill=tk.BOTH)

    def load_pdf(self):

        file_path = filedialog.askopenfilename(filetypes=[("PDF files", "*.pdf")])
        is_encrypted = False

        if not file_path:
            messagebox.showinfo("Error","No PDF selected, exiting")
            self.root.destroy()
            exit()
        
        with open(file_path, 'rb') as pdf_file:
            pdf_reader = PyPDF2.PdfReader(pdf_file)
            is_encrypted = pdf_reader.is_encrypted

        if is_encrypted:
            messagebox.showinfo("Error","The selected PDF is encrypted hence, exiting")
            self.root.destroy()
            exit()

        if file_path:
            self.pdf_document = fitz.open(file_path)
            self.current_page = 0
            self.show_page()
            self.update_page_number_label()
            self.root.deiconify()

    def show_page(self):

        if self.pdf_document is not None:
            page = self.pdf_document.load_page(self.current_page)
            pix = page.get_pixmap(matrix=fitz.Matrix(1, 1))
            img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)

            self.tk_image = ImageTk.PhotoImage(img)
            canvas_width = self.root.winfo_width()
            canvas_height = self.root.winfo_height()
            self.canvas.config(width=canvas_width, height=canvas_height)
            self.canvas.create_image(0, 0, anchor=tk.NW, image=self.tk_image)

    def update_page_number_label(self):

        if self.pdf_document is not None:
            total_pages = len(self.pdf_document)
            self.page_number_label.config(text=f"Page {self.current_page + 1}/{total_pages}")

    def prev_page(self):

        if self.current_page > 0:
            self.current_page -= 1
            self.show_page()
            self.update_page_number_label()

    def next_page(self):

        if self.pdf_document is not None and self.current_page < len(self.pdf_document) - 1:
            self.current_page += 1
            self.show_page()
            self.update_page_number_label()

    def go_to_page(self):

        if self.pdf_document is not None:
            try:
                page_number = int(self.page_entry_var.get())
                if 1 <= page_number <= len(self.pdf_document):
                    self.current_page = page_number - 1
                    self.show_page()
                    self.update_page_number_label()
                    self.page_entry.delete(0, tk.END)
                else:
                    messagebox.showwarning("Invalid Page Number", f"Please enter a page number between 1 and {len(self.pdf_document)}.")
                    self.page_entry.delete(0, tk.END)
            except ValueError:
                messagebox.showwarning("Invalid Input", "Please enter a valid page number.")
                self.page_entry.delete(0, tk.END)

    def add_page(self):

        if self.pdf_document is not None:
            file_path = filedialog.askopenfilename(filetypes=[("Image files", "*.png *.jpg *.jpeg *.JPG *.PNG *.JPEG")])
            if file_path:
                self.insert_page(file_path)
                self.show_page()
                self.update_page_number_label()

    def add_to_last_page(self):

        if self.pdf_document is not None:
            file_path = filedialog.askopenfilename(filetypes=[("Image files", "*.png *.jpg *.jpeg *.JPG *.PNG *.JPEG")])
            if file_path:
                self.insert_page(file_path, page_index=len(self.pdf_document))
                self.show_page()
                self.update_page_number_label()

    def delete_all(self):

        if self.pdf_document is not None:
            total_pages = len(self.pdf_document)
            if total_pages == 1:
                messagebox.showinfo("All Pages Deleted", "All pages have been deleted. Quitting the application.")
                self.pdf_document.close()
                self.root.quit()
            else:
                remaining_images = len(self.temp_images) - total_pages
                if remaining_images == 0:
                    messagebox.showinfo("Images Remaining", f"There are {remaining_images} images remaining. Quitting the application.")
                self.pdf_document.close()
                self.root.quit()

    def delete_page(self):

        if self.pdf_document is not None:
            if len(self.pdf_document) > 0:
                self.pdf_document.delete_page(self.current_page)
                if self.current_page < len(self.temp_images):
                    self.temp_images.pop(self.current_page)

                if self.current_page == len(self.pdf_document):
                    self.current_page -= 1

                if len(self.pdf_document) > 0:
                    self.show_page()
                    self.update_page_number_label()
                else:
                    self.delete_all()

    def insert_page(self, image_path, page_index=None):

        if image_path:
            if page_index is None:
                page_index = self.current_page

            new_page = self.pdf_document.new_page(page_index, width=500, height=700)
            rect = fitz.Rect(0, 0, 500, 700)

            img = Image.open(image_path)
            exif_data = img._getexif()
            if exif_data:
                for tag, value in exif_data.items():
                    if tag in ExifTags.TAGS and ExifTags.TAGS[tag] == "Orientation":
                        if value == 3:
                            new_page.set_rotation(180)
                        elif value == 6:
                            new_page.set_rotation(90)
                        elif value == 8:
                            new_page.set_rotation(270)

            image_pixmap = fitz.Pixmap(image_path)
            new_page.insert_image(rect, pixmap=image_pixmap)

            temp_image = Image.open(image_path)
            self.temp_images.insert(page_index, temp_image)



    def save_pdf(self):

        if self.pdf_document is not None:
            save_path = filedialog.asksaveasfilename(
                filetypes=[("PDF files", "*.pdf")],
                defaultextension=".pdf",
            )

            if save_path:
                self.pdf_document.save(save_path)
                self.pdf_document.close()
                messagebox.showinfo("Message", "PDF has been saved successfully as "+save_path.split("/")[-1] + " at " + os.path.dirname(save_path))
                self.root.quit()
        
            else:
                messagebox.showinfo("Error", "No Location Selected")
                return
            
if __name__ == "__main__":

    root = tk.Tk()
    screen_width = root.winfo_screenwidth()
    screen_height = root.winfo_screenheight()
    root.geometry(f"{screen_width}x{screen_height}+0+0")
    app = PDFEditorApp(root)
    app.root.withdraw()
    app.load_pdf()
    root.mainloop()