import tkinter as tk
from tkinter import filedialog, messagebox
import base64
 
class Base64ImageConverter:
    def __init__(self, root):
        self.root = root
        self.root.title("圖片與Base64轉(zhuǎn)換工具")
        self.root.geometry("500x400")
 
        # 上傳圖片按鈕
        self.upload_button = tk.Button(root, text="上傳圖片", command=self.upload_image)
        self.upload_button.pack(pady=10)
 
        # 顯示圖片路徑
        self.image_path_label = tk.Label(root, text="未選擇圖片", fg="gray")
        self.image_path_label.pack(pady=5)
 
        # Base64 輸入框
        self.base64_input_label = tk.Label(root, text="輸入Base64字符串:")
        self.base64_input_label.pack(pady=5)
        self.base64_input = tk.Text(root, height=5, width=50)
        self.base64_input.pack(pady=5)
 
        # 轉(zhuǎn)換按鈕
        self.convert_button = tk.Button(root, text="轉(zhuǎn)換為Base64", command=self.convert_to_base64)
        self.convert_button.pack(pady=10)
 
        self.convert_button2 = tk.Button(root, text="轉(zhuǎn)換為圖片", command=self.convert_to_image)
        self.convert_button2.pack(pady=10)
 
        # 狀態(tài)顯示
        self.status_label = tk.Label(root, text="", fg="blue")
        self.status_label.pack(pady=10)
 
    def upload_image(self):
        """上傳圖片并顯示路徑"""
        file_path = filedialog.askopenfilename(filetypes=[("Image Files", "*.jpg;*.jpeg;*.png;*.bmp")])
        if file_path:
            self.image_path_label.config(text=file_path, fg="green")
            self.status_label.config(text="圖片已上傳", fg="blue")
 
    def convert_to_base64(self):
        """將圖片轉(zhuǎn)換為Base64"""
        image_path = self.image_path_label.cget("text")
        if image_path == "未選擇圖片":
            messagebox.showerror("錯(cuò)誤", "請(qǐng)先上傳圖片！")
            return
 
        try:
            with open(image_path, "rb") as image_file:
                encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
                self.base64_input.delete(1.0, tk.END)  # 清空輸入框
                self.base64_input.insert(tk.END, encoded_string)
                self.status_label.config(text="圖片已轉(zhuǎn)換為Base64", fg="green")
        except Exception as e:
            messagebox.showerror("錯(cuò)誤", f"轉(zhuǎn)換失敗: {str(e)}")
 
    def convert_to_image(self):
        """將Base64轉(zhuǎn)換為圖片并保存"""
        base64_string = self.base64_input.get(1.0, tk.END).strip()
        if not base64_string:
            messagebox.showerror("錯(cuò)誤", "Base64字符串不能為空！")
            return
 
        try:
            output_path = filedialog.asksaveasfilename(defaultextension=".jpg", filetypes=[("JPEG Files", "*.jpg"), ("PNG Files", "*.png")])
            if output_path:
                image_data = base64.b64decode(base64_string)
                with open(output_path, "wb") as image_file:
                    image_file.write(image_data)
                self.status_label.config(text=f"圖片已保存為: {output_path}", fg="green")
        except Exception as e:
            messagebox.showerror("錯(cuò)誤", f"轉(zhuǎn)換失敗: {str(e)}")
 
if __name__ == "__main__":
    root = tk.Tk()
    app = Base64ImageConverter(root)
    root.mainloop()