hand_gesturer_ecognition/gui.py

import threading
from tkinter import *
from tkinter import ttk
from PIL import Image, ImageTk
import numpy as np  # 导入 numpy
from gesture_recognition import start_recognition, stop_recognition, release_camera, keep_running

WINDOW_WIDTH = 800
WINDOW_HEIGHT = 705


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 keep_running, paused, popup_open
    if not keep_running:
        status_label.config(text="正在识别...")
        paused = False
        popup_open = False
        threading.Thread(
            target=lambda: start_recognition(callback=lambda data: root.after(0, update_canvas, canvas, data))).start()


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


def exit_program():
    """退出程序"""
    global window
    stop_recognition()
    release_camera()
    window.destroy()


def update_canvas(canvas, data):
    """更新画布显示图像或处理手指数"""
    if isinstance(data, np.ndarray):  # 确保传入的data是图像数据
        img = Image.fromarray(data)
        imgtk = ImageTk.PhotoImage(image=img)
        canvas.create_image(0, 0, anchor=NW, image=imgtk)
        canvas.image = imgtk
        canvas.update()
    else:
        handle_finger_detection(data)


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

def show_stop_recognition_window():
    """显示停止识别确认窗口"""

    def on_continue():
        """继续识别"""
        global paused, popup_open
        paused = False
        popup_open = False  # 关闭弹窗后将标志设置为False
        stop_window.destroy()
        start_thread(root, canvas, status_label)

    def on_stop():
        """停止识别"""
        global popup_open
        stop_recognition()
        popup_open = False  # 关闭弹窗后将标志设置为False
        stop_window.destroy()

    stop_window = Tk()
    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()