import os
import shutil
import threading
import tkinter as tk
from tkinter import filedialog, messagebox, simpledialog, ttk
from PIL import Image, ImageTk

class ImageToPDFApp:
    def __init__(self, root):
        self.root = root
        self.root.title("圖片轉(zhuǎn)PDF工具")
        self.root.geometry("800x600")
        self.image_paths = []
        self.modified_images = {}
        self.last_dir = os.path.expanduser("~")
        self.renamed_dir = os.path.join(os.getcwd(), "renamed_images")

        self.build_ui()

    def build_ui(self):
        top_frame = tk.Frame(self.root, bg="#f0f0f0")
        top_frame.pack(fill=tk.X, padx=5, pady=5)

        buttons = [
            ("添加圖片", self.add_images),
            ("刪除選中", self.remove_selected),
            ("上移", self.move_up),
            ("下移", self.move_down),
            ("清空列表", self.clear_list),
            ("轉(zhuǎn)換為PDF", self.convert_to_pdf)
        ]

        for text, cmd in buttons:
            b = tk.Button(top_frame, text=text, command=cmd, width=12, font=("Times New Roman", 12))
            b.pack(side=tk.LEFT, padx=3, pady=2)

        middle_frame = tk.Frame(self.root)
        middle_frame.pack(fill=tk.BOTH, expand=True, padx=5, pady=5)

        left_frame = tk.Frame(middle_frame)
        left_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)

        self.lst_images = tk.Listbox(left_frame)
        self.lst_images.pack(fill=tk.BOTH, expand=True)
        self.lst_images.bind("<<ListboxSelect>>", self.update_preview)
        
        #增加快捷鍵
        self.lst_images.bind("<Up>", self.on_key_up)
        self.lst_images.bind("<Down>", self.on_key_down)
        self.lst_images.bind("<Delete>", self.on_key_delete)

        preview_frame = tk.Frame(middle_frame)
        preview_frame.pack(side=tk.RIGHT, fill=tk.BOTH, expand=True)

        self.preview_label = tk.Label(preview_frame, text="圖片預(yù)覽", relief=tk.SUNKEN, bg="white")
        self.preview_label.pack(side=tk.LEFT, expand=True, fill=tk.BOTH, padx=5)

        ctrl_frame = tk.Frame(preview_frame)
        ctrl_frame.pack(side=tk.RIGHT, fill=tk.Y, padx=5)

        ctrl_buttons = [
            ("向左旋轉(zhuǎn) (90°)", lambda: self.transform_image('rotate_left')),
            ("向右旋轉(zhuǎn) (90°)", lambda: self.transform_image('rotate_right')),
            ("水平翻轉(zhuǎn)", lambda: self.transform_image('flip_h')),
            ("垂直翻轉(zhuǎn)", lambda: self.transform_image('flip_v')),
            ("重置", self.reset_image),
            ("批量重命名", self.batch_rename_images)  # 添加按鈕
        ]

        for text, cmd in ctrl_buttons:
            btn = tk.Button(ctrl_frame, text=text, command=cmd, width=14, font=("Times New Roman", 12))
            btn.pack(fill=tk.X, pady=3)

        self.img_count_label = tk.Label(ctrl_frame, text="已添加 0 張圖片", font=("Times New Roman", 12), anchor='w')
        self.img_count_label.pack(side=tk.BOTTOM, pady=5, fill=tk.X)

        self.progress = ttk.Progressbar(self.root, mode="determinate")
        self.progress.pack(fill=tk.X, padx=5, pady=5)
        
    def on_key_up(self, event):
        self.move_up()

    def on_key_down(self, event):
        self.move_down()

    def on_key_delete(self, event):
        self.remove_selected()

    def add_images(self):
        files = filedialog.askopenfilenames(filetypes=[("Image files", "*.png *.jpg *.jpeg *.bmp *.gif *.tiff")],
                                            initialdir=self.last_dir)
        if files:
            for f in files:
                if f not in self.image_paths:
                    self.image_paths.append(f)
                    self.lst_images.insert(tk.END, os.path.basename(f))
            self.img_count_label.config(text=f"已添加 {len(self.image_paths)} 張圖片")
            self.lst_images.select_set(0)
            self.update_preview()

    def remove_selected(self):
        selected = self.lst_images.curselection()
        for i in reversed(selected):
            del self.image_paths[i]
            self.lst_images.delete(i)
        self.img_count_label.config(text=f"已添加 {len(self.image_paths)} 張圖片")
        self.preview_label.config(image='', text='圖片預(yù)覽')

    def move_up(self):
        i = self.lst_images.curselection()
        if i and i[0] > 0:
            idx = i[0]
            self.image_paths[idx - 1], self.image_paths[idx] = self.image_paths[idx], self.image_paths[idx - 1]
            self.lst_images.delete(idx)
            self.lst_images.insert(idx - 1, os.path.basename(self.image_paths[idx - 1]))
            self.lst_images.select_set(idx - 1)
            self.update_preview()

    def move_down(self):
        i = self.lst_images.curselection()
        if i and i[0] < len(self.image_paths) - 1:
            idx = i[0]
            self.image_paths[idx + 1], self.image_paths[idx] = self.image_paths[idx], self.image_paths[idx + 1]
            self.lst_images.delete(idx)
            self.lst_images.insert(idx + 1, os.path.basename(self.image_paths[idx + 1]))
            self.lst_images.select_set(idx + 1)
            self.update_preview()

    def clear_list(self):
        self.image_paths.clear()
        self.lst_images.delete(0, tk.END)
        self.modified_images.clear()
        self.preview_label.config(image='', text='圖片預(yù)覽')
        self.img_count_label.config(text="已添加 0 張圖片")

    def update_preview(self, event=None):
        if not self.lst_images.curselection():
            return
        idx = self.lst_images.curselection()[0]
        path = self.image_paths[idx]
        img = self.modified_images.get(path, Image.open(path))
        img.thumbnail((300, 300))
        self.tkimg = ImageTk.PhotoImage(img)
        self.preview_label.config(image=self.tkimg, text='')

    def transform_image(self, action):
        idxs = self.lst_images.curselection()
        if not idxs:
            return
        idx = idxs[0]
        path = self.image_paths[idx]
        img = self.modified_images.get(path, Image.open(path))

        if action == 'rotate_left':
            img = img.rotate(90, expand=True)
        elif action == 'rotate_right':
            img = img.rotate(-90, expand=True)
        elif action == 'flip_h':
            img = img.transpose(Image.FLIP_LEFT_RIGHT)
        elif action == 'flip_v':
            img = img.transpose(Image.FLIP_TOP_BOTTOM)

        self.modified_images[path] = img
        self.update_preview()

    def reset_image(self):
        idx = self.lst_images.curselection()
        if not idx:
            return
        path = self.image_paths[idx[0]]
        self.modified_images.pop(path, None)
        self.update_preview()

    def batch_rename_images(self):
        if not self.image_paths:
            messagebox.showwarning("警告", "沒(méi)有圖片可重命名")
            return

        template = simpledialog.askstring("命名模板", "請(qǐng)輸入命名模板（如 Page_##.jpg）：", initialvalue="Page_##.jpg")
        if not template or "##" not in template:
            messagebox.showwarning("格式錯(cuò)誤", "命名模板必須包含 ## 作為編號(hào)占位符")
            return

        if os.path.exists(self.renamed_dir):
            shutil.rmtree(self.renamed_dir)
        os.makedirs(self.renamed_dir, exist_ok=True)

        new_paths = []
        for i, path in enumerate(self.image_paths):
            new_name = template.replace("##", f"{i+1:02}")
            new_path = os.path.join(self.renamed_dir, new_name)
            shutil.copy2(path, new_path)
            new_paths.append(new_path)

        self.image_paths = new_paths
        self.lst_images.delete(0, tk.END)
        for path in self.image_paths:
            self.lst_images.insert(tk.END, os.path.basename(path))
        self.lst_images.select_set(0)
        self.update_preview()
        messagebox.showinfo("完成", f"已重命名圖片，保存在: {self.renamed_dir}")

    def convert_to_pdf(self):
        if not self.image_paths:
            messagebox.showwarning("警告", "沒(méi)有圖片可轉(zhuǎn)換")
            return
        output_path = filedialog.asksaveasfilename(defaultextension=".pdf", filetypes=[("PDF文件", "*.pdf")])
        if not output_path:
            return

        def task():
            images = []
            for i, path in enumerate(self.image_paths):
                img = self.modified_images.get(path, Image.open(path))
                if img.mode != 'RGB':
                    img = img.convert('RGB')
                images.append(img)
                self.progress['value'] = int((i + 1) / len(self.image_paths) * 100)
                self.root.update_idletasks()

            if images:
                images[0].save(output_path, save_all=True, append_images=images[1:])
                messagebox.showinfo("成功", f"PDF已保存至: {output_path}")
                self.progress['value'] = 0

        threading.Thread(target=task).start()
if __name__ == '__main__':
    root = tk.Tk()
    app = ImageToPDFApp(root)
    root.mainloop()