Improve description parsing by ensuring zone-shaped consistency

healthonrails · Nov 6, 2024 · 471256e · 471256e
1 parent 0d5d1f9
commit 471256e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/annolid/gui/app.py b/annolid/gui/app.py
@@ -2438,9 +2438,9 @@ def loadPredictShapes(self, frame_number, filename):
                     self.loadLabels(self.labelFile.shapes)
                     caption = self.labelFile.get_caption()
                     if caption is not None:
-                        # self.canvas.setCaption(caption)
-                        if self.caption_widget is not None:
-                            self.caption_widget.set_caption(caption)
+                        if self.caption_widget is None:
+                            self.openCaption()
+                        self.caption_widget.set_caption(caption)
             except Exception as e:
                 print(e)
 

diff --git a/annolid/postprocessing/tracking_results_analyzer.py b/annolid/postprocessing/tracking_results_analyzer.py
@@ -115,8 +115,10 @@ def load_zone_json(self):
         with open(self.zone_file, 'r') as f:
             self.zone_data = json.load(f)
         logger.info(f"Loading zones from {self.zone_file}")
+
         self.zone_shapes = [zone_shape for zone_shape in self.zone_data['shapes']
-                            if 'description' in zone_shape and 'zone' in zone_shape['description'].lower()
+                            if 'description' in zone_shape and zone_shape['description'] and 
+                             'zone' in zone_shape['description'].lower()
                             or 'zone' in zone_shape['label'].lower()]
         self.zone_time_dict = {shape['label']: 0 for shape in self.zone_shapes}