Update history_manager.py

Add undo/redo for smart annotation

Author: woshimajintao

annotation_tool/controllers/history_manager.py

@@ -90,6 +90,31 @@ def _apply_state_change(self, cmd, is_undo):
                 if path in self.model.manual_annotations: del self.model.manual_annotations[path]
             else: self.model.manual_annotations[path] = copy.deepcopy(data)
             self.main.refresh_ui_after_undo_redo(path)
+
+        # ==========================================
+        # [NEW] Handle batch annotation confirm
+        # ==========================================
+        elif ctype == CmdType.BATCH_ANNOTATION_CONFIRM:
+            batch_changes = cmd['batch_changes'] # Retrieve the packed dictionary
+            
+            # Loop through every video that was modified in this batch
+            for path, changes in batch_changes.items():
+                data = changes['old_data'] if is_undo else changes['new_data']
+                
+                # Apply the data
+                if data:
+                    self.model.manual_annotations[path] = copy.deepcopy(data)
+                else:
+                    if path in self.model.manual_annotations:
+                        del self.model.manual_annotations[path]
+                        
+                # Update the checkmark status in the Tree UI for this video
+                self.main.update_action_item_status(path)
+                
+            # Refresh the right panel if the currently selected item was affected
+            self._refresh_active_view()
+        # ==========================================
+
 
         elif ctype == CmdType.UI_CHANGE:
             path = cmd['path']
@@ -378,4 +403,4 @@ def _apply_state_change(self, cmd, is_undo):
                     if evt.get('head') == head and evt.get('label') == src:
                         evt['label'] = dst
 
-            self._refresh_active_view()
+            self._refresh_active_view()