sudacode/owocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add pause at startup, display current engine at startup

Browse Source
This commit is contained in:
AuroraWright
2023-12-09 10:01:33 +01:00
parent 2bec213534
commit 3eb8db3a48
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
manga_ocr/run.py
View File

@@ -53,6 +53,7 @@ def run(read_from='clipboard',
force_cpu=False, force_cpu=False,
delay_secs=0.5, delay_secs=0.5,
engine='mangaocr', engine='mangaocr',
start_paused=False,
verbose=False verbose=False
): ):
""" """
@@ -65,6 +66,7 @@ def run(read_from='clipboard',
:param force_cpu: If True, OCR will use CPU even if GPU is available. :param force_cpu: If True, OCR will use CPU even if GPU is available.
:param delay_secs: How often to check for new images, in seconds. :param delay_secs: How often to check for new images, in seconds.
:param engine: OCR engine to use. Available: "mangaocr", "gvision", "avision", "azure", "easyocr", "paddleocr". :param engine: OCR engine to use. Available: "mangaocr", "gvision", "avision", "azure", "easyocr", "paddleocr".
:param start_paused: Pause at startup.
:param verbose: If True, unhides all warnings. :param verbose: If True, unhides all warnings.
""" """
@@ -109,13 +111,16 @@ def run(read_from='clipboard',
if read_from == 'clipboard': if read_from == 'clipboard':
from PIL import ImageGrab from PIL import ImageGrab
logger.info('Reading from clipboard')
paused = False
global just_unpaused global just_unpaused
global tmp_paused
paused = start_paused
just_unpaused = True just_unpaused = True
tmp_paused = False
img = None img = None
logger.opt(ansi=True).info(f"Reading from clipboard using <cyan>{get_engine_name(engine)}</cyan>{' (paused)' if paused else ''}")
def on_key_press(key): def on_key_press(key):
global tmp_paused global tmp_paused
if key == keyboard.Key.cmd_r or key == keyboard.Key.ctrl_r: if key == keyboard.Key.cmd_r or key == keyboard.Key.ctrl_r:
@@ -128,9 +133,6 @@ def run(read_from='clipboard',
tmp_paused = False tmp_paused = False
just_unpaused = True just_unpaused = True
global tmp_paused
tmp_paused = False
tmp_paused_listener = keyboard.Listener( tmp_paused_listener = keyboard.Listener(
on_press=on_key_press, on_press=on_key_press,
on_release=on_key_release) on_release=on_key_release)
@@ -140,7 +142,7 @@ def run(read_from='clipboard',
if not read_from.is_dir(): if not read_from.is_dir():
raise ValueError('read_from must be either "clipboard" or a path to a directory') raise ValueError('read_from must be either "clipboard" or a path to a directory')
logger.info(f'Reading from directory {read_from}') logger.opt(ansi=True).info(f'Reading from directory {read_from} using <cyan>{get_engine_name(engine)}</cyan>')
old_paths = set() old_paths = set()
for path in read_from.iterdir(): for path in read_from.iterdir():