import os
import tkinter as tk
from tkinter import ttk, filedialog, messagebox
from PIL import Image
from pathlib import Path
 
SUPPORTED_FORMATS = ["JPEG", "PNG", "BMP", "GIF", "TIFF", "WEBP"]
 
class ImageConverterApp:
    def __init__(self, root):
        self.root = root
        self.root.title("圖片格式轉(zhuǎn)換工具")
        self.root.geometry("800x600")
 
        # 初始化變量
        self.selected_files = []
        self.output_format = tk.StringVar(value="JPEG")
        self.output_path = tk.StringVar()
        self.quality = tk.IntVar(value=85)
        self.resize_enabled = tk.BooleanVar(value=False)
        self.new_width = tk.IntVar(value=0)
        self.new_height = tk.IntVar(value=0)
        self.keep_ratio = tk.BooleanVar(value=True)
        self.keep_metadata = tk.BooleanVar(value=False)
 
        self.create_widgets()
 
    def create_widgets(self):
        # 文件選擇區(qū)域
        file_frame = ttk.LabelFrame(self.root, text="選擇圖片")
        file_frame.pack(padx=10, pady=5, fill="x")
 
        ttk.Button(file_frame, text="添加文件", command=self.add_files).pack(side="left", padx=5)
        ttk.Button(file_frame, text="添加文件夾", command=self.add_folder).pack(side="left", padx=5)
        ttk.Button(file_frame, text="清空列表", command=self.clear_files).pack(side="right", padx=5)
 
        # 文件列表
        self.file_list = tk.Listbox(self.root, selectmode=tk.EXTENDED)
        self.file_list.pack(padx=10, pady=5, fill="both", expand=True)
 
        # 設置區(qū)域
        settings_frame = ttk.LabelFrame(self.root, text="轉(zhuǎn)換設置")
        settings_frame.pack(padx=10, pady=5, fill="x")
 
        # 輸出格式
        ttk.Label(settings_frame, text="輸出格式:").grid(row=0, column=0, sticky="w")
        format_combo = ttk.Combobox(settings_frame, textvariable=self.output_format, 
                                  values=SUPPORTED_FORMATS, state="readonly")
        format_combo.grid(row=0, column=1, padx=5, sticky="ew")
 
        # 輸出路徑
        ttk.Label(settings_frame, text="輸出路徑:").grid(row=1, column=0, sticky="w")
        ttk.Entry(settings_frame, textvariable=self.output_path).grid(row=1, column=1, columnspan=2, padx=5, sticky="ew")
        ttk.Button(settings_frame, text="瀏覽...", command=self.select_output_path).grid(row=1, column=3, padx=5)
 
        # 質(zhì)量設置
        ttk.Label(settings_frame, text="圖片質(zhì)量 (1-100):").grid(row=2, column=0, sticky="w")
        ttk.Scale(settings_frame, from_=1, to=100, variable=self.quality, 
                 command=lambda v: self.quality.set(int(float(v)))).grid(row=2, column=1, padx=5, sticky="ew")
        ttk.Label(settings_frame, textvariable=self.quality).grid(row=2, column=2, padx=5)
 
        # 尺寸調(diào)整
        resize_frame = ttk.Frame(settings_frame)
        resize_frame.grid(row=3, column=0, columnspan=3, sticky="ew", pady=5)
        ttk.Checkbutton(resize_frame, text="調(diào)整尺寸", variable=self.resize_enabled).pack(side="left", padx=5)
        ttk.Entry(resize_frame, textvariable=self.new_width, width=5).pack(side="left", padx=5)
        ttk.Label(resize_frame, text="x").pack(side="left")
        ttk.Entry(resize_frame, textvariable=self.new_height, width=5).pack(side="left", padx=5)
        ttk.Checkbutton(resize_frame, text="保持比例", variable=self.keep_ratio).pack(side="left", padx=5)
 
        # 其他選項
        ttk.Checkbutton(settings_frame, text="保留元數(shù)據(jù)", variable=self.keep_metadata).grid(row=4, column=0, columnspan=3, sticky="w")
 
        # 操作按鈕
        button_frame = ttk.Frame(self.root)
        button_frame.pack(padx=10, pady=10, fill="x")
 
        ttk.Button(button_frame, text="開始轉(zhuǎn)換", command=self.start_conversion).pack(side="right", padx=5)
        ttk.Button(button_frame, text="退出", command=self.root.destroy).pack(side="right", padx=5)
 
    def add_files(self):
        files = filedialog.askopenfilenames(
            filetypes=[("圖片文件", "*.jpg *.jpeg *.png *.bmp *.gif *.tiff *.webp")]
        )
        if files:
            self.selected_files.extend(files)
            self.update_file_list()
 
    def add_folder(self):
        folder = filedialog.askdirectory()
        if folder:
            for ext in ("*.jpg", "*.jpeg", "*.png", "*.bmp", "*.gif", "*.tiff", "*.webp"):
                self.selected_files.extend(Path(folder).glob(ext))
            self.update_file_list()
 
    def clear_files(self):
        self.selected_files = []
        self.file_list.delete(0, tk.END)
 
    def update_file_list(self):
        self.file_list.delete(0, tk.END)
        for f in self.selected_files:
            self.file_list.insert(tk.END, str(f))
 
    def select_output_path(self):
        path = filedialog.askdirectory()
        if path:
            self.output_path.set(path)
 
    def start_conversion(self):
        if not self.selected_files:
            messagebox.showwarning("警告", "請先選擇要轉(zhuǎn)換的圖片文件！")
            return
    
        output_path = self.output_path.get() or os.path.dirname(self.selected_files[0])
        output_format = self.output_format.get()
    
        # 創(chuàng)建輸出目錄（如果不存在）
        os.makedirs(output_path, exist_ok=True)
    
        for file_path in self.selected_files:
            try:
                img = Image.open(file_path)
                
                # 處理尺寸調(diào)整
                if self.resize_enabled.get():
                    width = self.new_width.get()
                    height = self.new_height.get()
                    original_width, original_height = img.size
 
                    if self.keep_ratio.get():
                        if width > 0 and height == 0:
                            ratio = width / original_width
                            height = int(original_height * ratio)
                        elif height > 0 and width == 0:
                            ratio = height / original_height
                            width = int(original_width * ratio)
                        else:
                            ratio = min(width/original_width, height/original_height)
                            width = int(original_width * ratio)
                            height = int(original_height * ratio)
                    
                    if width > 0 and height > 0:
                        img = img.resize((width, height), Image.Resampling.LANCZOS)
    
                # 構(gòu)建輸出路徑
                filename = os.path.splitext(os.path.basename(file_path))[0]
                output_file = os.path.join(output_path, f"{filename}.{output_format.lower()}")
    
                # 檢查文件是否存在并提示
                if os.path.exists(output_file):
                    response = messagebox.askyesnocancel(
                        "文件已存在",
                        f"目標文件已存在：\n{output_file}\n\n"
                        "請選擇操作：\n"
                        "? 是：覆蓋當前文件\n"
                        "? 否：跳過當前文件\n"
                        "? 取消：中止全部轉(zhuǎn)換",
                        parent=self.root
                    )
                    
                    if response is None:  # 取消
                        return
                    if not response:      # 跳過
                        continue
    
                # 構(gòu)建輸出路徑
                filename = os.path.splitext(os.path.basename(file_path))[0]
                output_file = os.path.join(output_path, f"{filename}.{output_format.lower()}")
 
                # 保存參數(shù)
                save_args = {'format': output_format}
                if output_format == 'JPEG':
                    save_args['quality'] = self.quality.get()
                    save_args['optimize'] = True
                elif output_format == 'PNG':
                    save_args['compress_level'] = 9 - int(self.quality.get() / 11.1)
 
                # 保存圖片
                if not self.keep_metadata.get():
                    img.info.pop('exif', None)
                img.save(output_file, **save_args)
    
            except Exception as e:
                messagebox.showerror("錯誤", f"處理文件 {file_path} 時出錯：\n{str(e)}")
                continue
    
        messagebox.showinfo("完成", "圖片轉(zhuǎn)換完成！")
 
if __name__ == "__main__":
    root = tk.Tk()
    app = ImageConverterApp(root)
    root.mainloop()