Update annotation_manager.py

Adapt to new /models

Author: woshimajintao

annotation_tool/controllers/classification/annotation_manager.py

@@ -30,11 +30,14 @@ def save_manual_annotation(self):
         self.main.update_action_item_status(path)
         self.main.update_save_export_button_state()
 
+        # [MV Fix] Auto-advance using QTreeView API
         tree = self.ui.classification_ui.left_panel.tree
-        curr = tree.currentItem()
-        nxt = tree.itemBelow(curr)
-        if nxt:
-            QTimer.singleShot(500, lambda: [tree.setCurrentItem(nxt), tree.scrollToItem(nxt)])
+        curr_idx = tree.currentIndex()
+        if curr_idx.isValid():
+            # Try to get index below
+            nxt_idx = tree.indexBelow(curr_idx)
+            if nxt_idx.isValid():
+                QTimer.singleShot(500, lambda: [tree.setCurrentIndex(nxt_idx), tree.scrollTo(nxt_idx)])
 
     def clear_current_manual_annotation(self):
         path = self.main.get_current_action_path()
@@ -67,6 +70,10 @@ def handle_ui_selection_change(self, head, new_val):
 
         self.model.push_undo(CmdType.UI_CHANGE, path=path, head=head, old_val=old_val, new_val=new_val)
 
+    # ... handle_add_label_head, handle_remove_label_head ...
+    # These methods below generally don't touch the TreeView, so they are fine as is, 
+    # but I'll include them to ensure the file is complete.
+
     def handle_add_label_head(self, name):
         clean = name.strip().replace(' ', '_').lower()
         if not clean or clean in self.model.label_definitions: return
@@ -145,4 +152,4 @@ def remove_custom_type(self, head, lbl):
         group = self.ui.classification_ui.right_panel.label_groups.get(head)
         if isinstance(group, DynamicSingleLabelGroup): group.update_radios(defn['labels'])
         else: group.update_checkboxes(defn['labels'])
-        self.display_manual_annotation(self.main.get_current_action_path())
+        self.display_manual_annotation(self.main.get_current_action_path())