Upload folder using huggingface_hub

candidate_window.py

@@ -0,0 +1,53 @@
+# candidate_window.py
+import tkinter as tk
+from caret import get_caret_rect
+
+class CandidateWindow:
+    def __init__(self):
+        self.root = tk.Tk()
+        self.root.overrideredirect(True)  # 无边框
+        self.root.attributes("-topmost", True)
+
+        self.label = tk.Label(
+            self.root,
+            text="",
+            bg="#222",
+            fg="white",
+            font=("Segoe UI", 12),
+            justify="left",
+            padx=8,
+            pady=4
+        )
+        self.label.pack()
+
+        self.visible = False
+
+    def set_text(self, text):
+        self.label.config(text=text)
+
+    def show(self):
+        if not self.visible:
+            self.visible = True
+            self.root.deiconify()
+
+    def hide(self):
+        self.visible = False
+        self.root.withdraw()
+
+    def update_position(self):
+        rect = get_caret_rect()
+
+        if rect:
+            x = rect.left
+            y = rect.bottom + 5  # 关键：显示在光标下方
+
+            self.root.geometry(f"+{x}+{y}")
+
+    def tick(self):
+        self.update_position()
+        self.root.after(10, self.tick)  # 高频刷新（输入法常见）
+
+    def start(self):
+        self.hide()
+        self.tick()
+        self.root.mainloop()

caret.py

@@ -0,0 +1,28 @@
+# caret.py
+import ctypes
+from ctypes import wintypes
+
+class GUITHREADINFO(ctypes.Structure):
+    _fields_ = [
+        ("cbSize", ctypes.c_ulong),
+        ("flags", ctypes.c_ulong),
+        ("hwndActive", ctypes.c_void_p),
+        ("hwndFocus", ctypes.c_void_p),
+        ("hwndCapture", ctypes.c_void_p),
+        ("hwndMenuOwner", ctypes.c_void_p),
+        ("hwndMoveSize", ctypes.c_void_p),
+        ("hwndCaret", ctypes.c_void_p),
+        ("rcCaret", wintypes.RECT),
+    ]
+
+def get_caret_rect():
+    info = GUITHREADINFO()
+    info.cbSize = ctypes.sizeof(GUITHREADINFO)
+
+    ctypes.windll.user32.GetGUIThreadInfo(
+        0,
+        ctypes.byref(info)
+    )
+
+    # rcCaret 是屏幕坐标（关键）
+    return info.rcCaret

main.py

@@ -0,0 +1,37 @@
+# main.py
+import keyboard
+from candidate_window import CandidateWindow
+from engine import Engine
+
+cw = CandidateWindow()
+engine = Engine()
+
+def on_key(e):
+    if e.event_type != "down":
+        return
+
+    # ESC 隐藏
+    if e.name == "esc":
+        cw.hide()
+        engine.reset()
+        return
+
+    # 普通字符
+    if len(e.name) == 1:
+        candidates = engine.input(e.name)
+
+        text = "\n".join([
+            f"{i+1}. {c}" for i, c in enumerate(candidates)
+        ])
+
+        cw.set_text(text)
+        cw.show()
+
+    # 空格选择（模拟）
+    if e.name == "space":
+        engine.reset()
+        cw.hide()
+
+keyboard.hook(on_key)
+
+cw.start()