import tkinter as tk  # Python的标准图形用户界面库，用于创建图形用户界面。
from tkinter import filedialog
from PIL import Image, ImageTk  # Python图像处理库，提供图像处理功能。
from util import preprocess_image, load_file, get_config_value  # 引入预处理、加载和配置文件读取的方法

char_styles = get_config_value('char_styles')  # 字体样式
new_size = get_config_value('new_size')  # 新尺寸


class ImageClassifierApp:
    def __init__(self, model_path):
        # 加载最佳模型
        self.model = load_file("F1最高模型", model_path)

        # 创建主窗口
        self.root = tk.Tk()
        self.root.title('书法体识别器')
        self.root.geometry("300x200")

        # 创建一个按钮用于选择图像
        self.button = tk.Button(self.root, text='选择图像', command=self.select_image)
        self.button.pack()

        # 创建一个标签用于显示图像
        self.image_label = tk.Label(self.root)
        self.image_label.pack()

        # 创建一个标签用于显示预测的类别
        self.prediction_label = tk.Label(self.root)
        self.prediction_label.pack()

        # 启动GUI事件循环
        self.root.mainloop()

    def select_image(self):
        # 打开文件对话框以选择图像
        image_path = filedialog.askopenfilename()

        # 预处理图像
        image_preprocessed = preprocess_image(image_path, new_size)

        # 执行预测
        predicted_class = self.model.predict([image_preprocessed])

        # 读取图像
        pil_image = Image.open(image_path)
        w, h = pil_image.size
        self.root.geometry(f"{w}x{h+100}")

        # 将PIL图像转换为PhotoImage并更新标签
        image_tk = ImageTk.PhotoImage(pil_image)
        self.image_label.config(image=image_tk)
        self.image_label.image = image_tk

        # 更新预测标签
        self.prediction_label.config(text=f'预测类别: {char_styles[predicted_class[0]]}')


# 启动应用程序
app = ImageClassifierApp(f'{get_config_value("model_root")}/best_model')