Support for post annotation label reassignement.

copienator/commands/reading_annotations.py

@@ -9,20 +9,26 @@ import numpy as np
 from pdf2image import convert_from_path
 from PIL import Image, ImageChops, ImageDraw, ImageFilter
 
-from copienator.commands import annotating
-from copienator import configuration, utils
 from copienator import (
     EvaluationWorkspace,
     ExitCode,
     atomic_write_json,
+    configuration,
     evaluation_parser,
     execute,
     read_json,
+    utils,
     workspace_from_args,
 )
 from copienator.annotation_actions import apply_checkbox_actions, apply_score_overrides
-from copienator.answer_info import build_answer_info
 from copienator.annotation_data import AnnotationData, load_annotation_data
+from copienator.annotation_modifications import (
+    BASELINE,
+    output_removals,
+    stage_baseline,
+)
+from copienator.answer_info import build_answer_info
+from copienator.commands import annotating
 from copienator.filesystem import staged_files
 from copienator.return_answers import save_return_answer_options
 
@@ -168,6 +174,8 @@ def apply_actions_and_regenerate(
     scores = dict.fromkeys(all_labels, "")
     answer_labels: list[str] = []
     dirty_images: dict[str, Image.Image] = {}
+    header_heights: dict[str, int] = {}
+    filtered_labels: list[str] = []
     concatenated: list[Image.Image] = []
     filtered: list[Image.Image] = []
     incomplete = False
@@ -221,15 +229,18 @@ def apply_actions_and_regenerate(
                 final_image.paste(body, (0, new_header_height), mask=body)
 
         dirty_images[label] = final_image
+        header_heights[label] = new_header_height
         answer_labels.append(label)
         concatenated.append(final_image)
         if float(scores[label]) != 4.0 or result.get("feedback", []):
             filtered.append(final_image)
+            filtered_labels.append(label)
 
+    if incomplete:
+        return ExitCode.PARTIAL
     concat_image = concatenate(concatenated)
     filtered_image = concatenate(filtered)
-    with staged_files(output_dir, remove=("Concat.jpg", "Concat_F.jpg", "Concat_F.pdf",
-                                         "touched.json", "answer_labels.json")) as staging:
+    with staged_files(output_dir, remove=(*output_removals(output_dir), BASELINE)) as staging:
         for label, image in dirty_images.items():
             image.save(staging / f"{label}.jpg")
         if not preserve_score_file:
@@ -241,6 +252,9 @@ def apply_actions_and_regenerate(
             concat_image.save(staging / "Concat.jpg")
         if filtered_image is not None:
             filtered_image.save(staging / "Concat_F.jpg")
+        stage_baseline(workspace.root, "Bnot", output_dir, staging, dirty_images,
+                       header_heights, build_answer_info(scores, labels_data, answer_labels),
+                       filtered_labels, scores)
 
     if preserve_score_file:
         print(f"  Preserved existing score.json in {output_dir}")