import threading
from tkinter import *
from tkinter import ttk
from PIL import Image, ImageTk
import numpy as np
from gesture_recognition import start_recognition, stop_recognition, release_camera, reset_hand_detection

WINDOW_WIDTH = 800
WINDOW_HEIGHT = 705
main_launched = False
status_label = None
root = None
popup_open = False
paused = False

def main():
    """显示欢迎屏幕"""
    show_welcome_screen()

def set_window_position(window, width, height):
    """设置窗口位置居中"""
    screen_width = window.winfo_screenwidth()
    screen_height = window.winfo_screenheight()
    x = (screen_width - width) // 2
    y = (screen_height - height) // 2
    window.geometry(f'{width}x{height}+{x}+{y}')

def show_welcome_screen():
    """显示欢迎界面"""
    welcome = Tk()
    welcome.title("欢迎使用")
    set_window_position(welcome, WINDOW_WIDTH, WINDOW_HEIGHT)

    bg_image = Image.open("1.png")
    bg_image = bg_image.resize((WINDOW_WIDTH, WINDOW_HEIGHT), Image.Resampling.LANCZOS)
    bg_image = ImageTk.PhotoImage(bg_image)

    background_label = Label(welcome, image=bg_image)
    background_label.image = bg_image
    background_label.place(relwidth=1, relheight=1)

    label = Label(welcome, text="欢迎使用手势识别系统", font=('Helvetica', 24, 'bold'), bg="lightblue", fg="white")
    label.place(relx=0.5, rely=0.4, anchor=CENTER)

    btn_continue = Button(welcome, text="继续", command=lambda: [welcome.destroy(), show_main_screen()], font=('Helvetica', 14), bg="yellow", fg="black", relief=SOLID)
    btn_continue.place(relx=0.5, rely=0.6, anchor=CENTER)

    welcome.mainloop()

def show_main_screen():
    """显示主界面"""
    global window, canvas, root, paused, popup_open, status_label
    window = Tk()
    root = window
    window.title("手势识别")
    set_window_position(window, WINDOW_WIDTH, WINDOW_HEIGHT)
    style = ttk.Style(window)
    style.theme_use('clam')

    frame_controls = Frame(window, bg='lightgray', pady=10)
    frame_controls.pack(fill=X)

    frame_video = Frame(window, bg='white')
    frame_video.pack(fill=BOTH, expand=True)

    canvas = Canvas(frame_video, bg='black')
    canvas.pack(fill=BOTH, expand=True, padx=20, pady=20)

    status_label = Label(frame_controls, text="等待开始...", font=('Helvetica', 14), bg='lightgray')
    status_label.pack(side=LEFT, padx=(10, 20))

    btn_start = ttk.Button(frame_controls, text="开始", command=lambda: start_thread(root, canvas, status_label))
    btn_start.pack(side=LEFT, padx=10)

    btn_stop = ttk.Button(frame_controls, text="停止", command=lambda: stop_recognition_with_label(status_label))
    btn_stop.pack(side=LEFT, padx=10)

    btn_exit = ttk.Button(frame_controls, text="退出", command=lambda: exit_program())
    btn_exit.pack(side=LEFT, padx=10)

    style = ttk.Style(window)
    style.configure('TButton', font=('Helvetica', 14))

    window.mainloop()

def start_thread(root, canvas, status_label):
    """启动识别线程"""
    global paused, popup_open
    if paused:
        reset_hand_detection()  # 重置手部检测状态
    status_label.config(text="正在识别...")
    paused = False
    popup_open = False
    threading.Thread(target=lambda: start_recognition(callback=lambda fingers, img: root.after(0, update_canvas, canvas, fingers, img))).start()

def stop_recognition_with_label(status_label):
    """停止识别并更新状态标签"""
    stop_recognition()
    status_label.config(text="已停止")

def exit_program():
    """退出程序"""
    global window
    stop_recognition()
    release_camera()
    if window is not None:
        window.destroy()
    window = None

def update_canvas(canvas, fingers, img):
    """更新画布显示图像或处理手指数"""
    if root is None or not root.winfo_exists():
        return
    if isinstance(img, np.ndarray):
        img = Image.fromarray(img)
        imgtk = ImageTk.PhotoImage(image=img)
        canvas.create_image(0, 0, anchor=NW, image=imgtk)
        canvas.image = imgtk
        canvas.update()
    handle_finger_detection(fingers)

def handle_finger_detection(finger_count):
    """处理检测到的手指数"""
    global paused, popup_open
    if finger_count == 5:
        if not paused and not popup_open:
            paused = True
            popup_open = True
            show_stop_recognition_window()

def show_stop_recognition_window():
    """显示停止识别确认窗口"""
    def on_continue():
        """继续识别"""
        global paused, popup_open, status_label
        paused = False
        popup_open = False
        stop_window.destroy()
        start_thread(root, canvas, status_label)

    def on_stop():
        """停止识别"""
        global popup_open
        stop_recognition()
        popup_open = False
        stop_window.destroy()

    if root is not None and root.winfo_exists():
        stop_window = Toplevel(root)
        stop_window.title("停止识别")

        label = Label(stop_window, text="您竖起了五根手指，是否停止识别？", font=('Helvetica', 24, 'bold'))
        label.pack(pady=20)

        continue_button = Button(stop_window, text="继续识别", command=on_continue)
        continue_button.pack(side=LEFT, padx=10, pady=10)

        stop_button = Button(stop_window, text="停止识别", command=on_stop)
        stop_button.pack(side=RIGHT, padx=10, pady=10)

        stop_window.protocol("WM_DELETE_WINDOW", on_continue)
        stop_window.mainloop()

if __name__ == "__main__":
    main()