39 lines
2.2 KiB
Python
39 lines
2.2 KiB
Python
"""Re-render saved handwriting interpretation with current scores and remarks."""
|
|
from copienator import EvaluationWorkspace, ExitCode, evaluation_parser, execute, workspace_from_args, utils
|
|
from copienator.review_snapshot import load, FILENAME
|
|
from copienator.commands.reading_annotations import apply_actions_and_regenerate
|
|
from copienator.commands.reading_grouped_annotations import apply_actions_and_regenerate_grouped
|
|
|
|
|
|
def run(workspace: EvaluationWorkspace, mode: str):
|
|
directories = sorted((p for p in (workspace.root / mode).glob('Copie*')
|
|
if p.is_dir() and p.name[5:].isdigit() and (p / 'score.json').is_file()))
|
|
# Validate every snapshot before starting to publish any copy.
|
|
for directory in directories:
|
|
if not (directory / FILENAME).is_file():
|
|
raise ValueError(f'{directory.name} : résultat de lecture sauvegardé absent ; relisez les annotations manuscrites.')
|
|
load(workspace.root, directory)
|
|
status = ExitCode.SUCCESS
|
|
for directory in directories:
|
|
cid = directory.name[5:]
|
|
saved, data, notes = load(workspace.root, directory)
|
|
if saved['engine'] == 'standard':
|
|
result = apply_actions_and_regenerate(workspace, {cid: data}, cid, [], None,
|
|
utils.read_all_labels(workspace.root), update_score=True,
|
|
label_notes={label: note['img'] for label, note in notes.items()},
|
|
review_metadata=saved['bnote'])
|
|
else:
|
|
result, log = apply_actions_and_regenerate_grouped(workspace, {cid: data}, cid, [], notes,
|
|
utils.read_all_labels(workspace.root), update_score=True, annotation_dir=mode,
|
|
force_filtered=set(saved.get('force_filtered', [])))
|
|
print(log, flush=True)
|
|
if result != ExitCode.SUCCESS:
|
|
status = ExitCode.PARTIAL
|
|
return status
|
|
|
|
|
|
def main(argv=None):
|
|
parser = evaluation_parser('Regenerate from saved reviewed inputs without rescanning handwriting')
|
|
parser.add_argument('--annotation-dir', choices=('Bnot', 'BGnot'), required=True)
|
|
return execute(parser, argv, lambda args: run(workspace_from_args(args), args.annotation_dir))
|