#!/usr/bin/env python3

import sys
sys.path.insert(0, '/MASTERFOLDER/Tools/text-polish/src')

import pyperclip, keyboard
from model_loader import load_model, polish

model, tokenizer = load_model()

def on_hotkey():
    text = pyperclip.paste()
    result = polish(model, tokenizer, text)
    pyperclip.copy(result)

keyboard.add_hotkey('ctrl+alt+p', on_hotkey)
keyboard.wait()

if __name__ == "__main__":
    print("Testing main.py implementation...")
    print("Main module loaded successfully")