Miscs

plotting.py

@@ -9,7 +9,7 @@ from tkinter import messagebox
 from pathlib import Path
 from PIL import Image, ImageDraw, ImageFont, ImageTk
 
-print("o to open pdf, O original pdf, e to emacs part, click for coordinates")
+print("o to open pdf, O original pdf, e to emacs part, i to interro, click for coordinates")
 
 # --- Configuration & Globals ---
 padding = 60
@@ -54,7 +54,7 @@ def convert_list(l, group_id, json_schema):
         ll.append(ee)
     return ll
 
-def prepare_image(image_path: str, bounding_boxes, all_labels, nb_pages):
+def prepare_image(image_path: str, bounding_boxes, all_labels, nb_pages, last_label_index):
     im = Image.open(image_path)
     im.load()
     width, height = im.size
@@ -62,7 +62,6 @@ def prepare_image(image_path: str, bounding_boxes, all_labels, nb_pages):
     new_im.paste(im, (0, 0))
     draw = ImageDraw.Draw(new_im)
     bounding_boxes.sort(key=lambda b: (page_number(b["box_2d"], nb_pages), b["box_2d"][0]))
-    last_label_index = -1
 
     for bbox in bounding_boxes:
         raw_y_min = int(bbox["box_2d"][0] * height / 1000)
@@ -88,7 +87,7 @@ def prepare_image(image_path: str, bounding_boxes, all_labels, nb_pages):
                 draw.text((abs_x_min + 8, abs_y_min - 30), label, fill=color, font=font)
             else:
                 draw.text((abs_x_min + 8, abs_y_max + 6), label, fill=color, font=font)
-    return new_im
+    return (new_im, last_label_index)
 
 # --- Processing Logic (Worker Thread) ---
 
@@ -97,10 +96,15 @@ def worker_thread(base_dir, files_to_process, all_labels):
     Iterates through files, prepares VISUALS only, and puts metadata in queue.
     Does NOT write final JSON files anymore.
     """
+    previous_copie = None
+    last_label_index = None
     for img_path in files_to_process:
         json_path = base_dir / f"{img_path.stem}.json"
         copie_part = int(img_path.stem[-2:])
         copie = img_path.stem[:-3]
+        if copie != previous_copie:
+            last_label_index = -1
+            previous_copie = copie
         json_schema_path = base_dir / 'Cutleft' / f"{copie}_schema.json"
 
         try:
@@ -127,7 +131,8 @@ def worker_thread(base_dir, files_to_process, all_labels):
 
             try:
                 print(f"Buffering {img_path.name}...")
-                pil_image = prepare_image(str(img_path), bb_list, all_labels, nb_pages)
+                (pil_image, last_label_index) = \
+                    prepare_image(str(img_path), bb_list, all_labels, nb_pages, last_label_index)
 
                 metadata = {
                     "copie": copie,
@@ -169,6 +174,7 @@ class ImageViewer:
         self.root.bind('<Return>', self.on_enter)
         self.root.bind('e', self.on_edit)
         self.root.bind('o', self.on_open_pdf)
+        self.root.bind('i', self.on_open_interro)
         self.root.bind('O', self.on_open_ori_pdf)
         self.root.bind('<Escape>', lambda e: self.root.quit())
         self.label.bind('<Button-1>', self.on_click)
@@ -265,6 +271,12 @@ class ImageViewer:
             subprocess.Popen(['xdg-open', str(pdf_path.absolute())])
 
     def on_open_ori_pdf(self, event):
+        if self.is_viewing and self.current_json_path:
+            pdf_path = "/home/sebastien/Staging/Interro/" + base_dir + "pdf"
+            print(f"Opening {pdf_path}")
+            subprocess.Popen(['xdg-open', pdf_path])
+
+    def on_open_interro(self, event):
         if self.is_viewing and self.current_json_path:
             new_filename = self.current_json_path.stem.split('_')[0] + ".pdf"
             pdf_path = self.current_json_path.parent / "Copies Originales" / new_filename