Add check for caption existence to ensure backward compatibility with…

… label files.
healthonrails · Oct 17, 2024 · 47b3afe · 47b3afe
1 parent ea46fc6
commit 47b3afe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/annolid/gui/app.py b/annolid/gui/app.py
@@ -2323,7 +2323,8 @@ def loadPredictShapes(self, frame_number, filename):
                 if self.labelFile:
                     self.loadLabels(self.labelFile.shapes)
                     caption = self.labelFile.get_caption()
-                    self.canvas.setCaption(caption)
+                    if caption is not None:
+                        self.canvas.setCaption(caption)
             except Exception as e:
                 print(e)
 

diff --git a/annolid/gui/label_file.py b/annolid/gui/label_file.py
@@ -97,7 +97,7 @@ def load(self, filename):
             imagePath = None
             caption = None
 
-            if data['caption'] is not None:
+            if 'caption' in data and data['caption'] is not None:
                 caption = data['caption']
 
             if data["imageData"] is not None: