Support for post annotation label reassignement.

copienator_gui/cut_helper.py

@@ -45,21 +45,29 @@ def cut_operator(percent: float, keep: int, mode: str) -> str:
 
 
 class CutHelper(tk.Toplevel):
-    def __init__(self, parent, path: Path, on_accept, initial=None):
-        # Load the source before creating a window so invalid PDFs leave no dialog.
-        with pymupdf.open(path) as document:
-            if not len(document):
-                raise ValueError("Le PDF est vide.")
-            self.heights = [page.rect.height for page in document]
-            width = min(850, parent.winfo_screenwidth() - 100)
-            self.scale = min(1.5, width / max(page.rect.width for page in document))
-            rendered = []
-            for page in document:
-                pix = page.get_pixmap(matrix=pymupdf.Matrix(self.scale, self.scale), alpha=False,
-                                      colorspace=pymupdf.csRGB)
-                rendered.append(Image.frombytes("RGB", (pix.width, pix.height), pix.samples))
+    def __init__(self, parent, path: Path | Image.Image, on_accept, initial=None):
+        # Body edits pass a raster without headers or supplementary pages.
+        width = min(850, parent.winfo_screenwidth() - 100)
+        if isinstance(path, Image.Image):
+            self.heights = [path.height]
+            self.scale = min(1.5, width / path.width)
+            rendered = [path.resize((round(path.width * self.scale),
+                                     round(path.height * self.scale)), Image.Resampling.LANCZOS)]
+            title = "Corps de la réponse annotée"
+        else:
+            with pymupdf.open(path) as document:
+                if not len(document):
+                    raise ValueError("Le PDF est vide.")
+                self.heights = [page.rect.height for page in document]
+                self.scale = min(1.5, width / max(page.rect.width for page in document))
+                rendered = []
+                for page in document:
+                    pix = page.get_pixmap(matrix=pymupdf.Matrix(self.scale, self.scale), alpha=False,
+                                          colorspace=pymupdf.csRGB)
+                    rendered.append(Image.frombytes("RGB", (pix.width, pix.height), pix.samples))
+            title = path.name
         super().__init__(parent)
-        self.title(f"Cut — {path.name}")
+        self.title(f"Cut — {title}")
         self.geometry(f"{width + 45}x{min(850, parent.winfo_screenheight() - 100)}")
         self.transient(parent.winfo_toplevel())
         self.on_accept = on_accept
@@ -125,7 +133,7 @@ class CutHelper(tk.Toplevel):
         operator = cut_operator(self.percent, self.keep.get(), self.mode.get())
         rounded = float(operator.split("{")[1].split("}")[0])
         if not 0 < rounded < 100:
-            messagebox.showerror("Coupe invalide", "Chaque partie doit contenir une portion du PDF.", parent=self)
+            messagebox.showerror("Coupe invalide", "Chaque partie doit contenir une portion de la réponse.", parent=self)
             return
         self.destroy()
         self.on_accept(operator)