Add menu item for deleting instances beyond frame limit

sleap/gui/app.py

@@ -788,6 +788,12 @@ def new_instance_menu_action():
             "Delete Predictions beyond Max Instances...",
             self.commands.deleteInstanceLimitPredictions,
         )
+        add_menu_item(
+            labelMenu,
+            "delete frame limit predictions new item",
+            "Delete Predictions beyond Frame Limit new item...",
+            self.commands.deleteFrameLimitPredictions,
+        )
 
         ### Tracks Menu ###
 

sleap/gui/commands.py

@@ -494,6 +494,10 @@ def deleteInstanceLimitPredictions(self):
         """Gui for deleting instances beyond some number in each frame."""
         self.execute(DeleteInstanceLimitPredictions)
 
+    def deleteFrameLimitPredictions(self):
+        """Gui for deleting instances beyond some frame number."""
+        self.execute(DeleteFrameLimitPredictions)
+
     def completeInstanceNodes(self, instance: Instance):
         """Adds missing nodes to given instance."""
         self.execute(AddMissingInstanceNodes, instance=instance)
@@ -2468,6 +2472,34 @@ def ask(cls, context: CommandContext, params: dict) -> bool:
             return super().ask(context, params)
 
 
+class DeleteFrameLimitPredictions(InstanceDeleteCommand):
+    @staticmethod
+    def get_frame_instance_list(context: CommandContext, params: Dict):
+        predicted_instances = []
+        # Select the instances to be deleted
+        for lf in context.labels.find(context.state["video"]):
+            if lf.frame_idx >= params["frame_idx_threshold"]:
+                predicted_instances.extend(
+                    [(lf, inst) for inst in lf.predicted_instances]
+                )
+        return predicted_instances
+
+    @classmethod
+    def ask(cls, context: CommandContext, params: Dict) -> bool:
+        current_video = context.state["video"]
+        frame_idx_thresh, okay = QtWidgets.QInputDialog.getInt(
+            context.app,
+            "Delete Instance beyond Frame Number...",
+            "Frame number after which instances to be deleted:",
+            1,
+            1,
+            len(current_video),
+        )
+        if okay:
+            params["frame_idx_threshold"] = frame_idx_thresh
+            return super().ask(context, params)
+
+
 class TransposeInstances(EditCommand):
     topics = [UpdateTopic.project_instances, UpdateTopic.tracks]