Allow novel model refresh

novelwriter/core/index.py

@@ -120,10 +120,25 @@ class Index:
             self._generateNovelModel(tHandle)
         return self._novelModels.get(tHandle)
 
+    def refreshNovelModel(self, tHandle: str) -> None:
+        """Refresh a novel model."""
+        if model := self.getNovelModel(tHandle):
+            model.beginResetModel()
+            model.clear()
+            self._appendSubTreeToModel(tHandle, model)
+            model.endResetModel()
+        return
+
     def _generateNovelModel(self, tHandle: str) -> None:
         """Generate a novel model for a specific handle."""
         if (item := SHARED.project.tree[tHandle]) and item.isRootType() and item.isNovelLike():
             model = NovelModel()
+            self._appendSubTreeToModel(tHandle, model)
+            self._novelModels[tHandle] = model
+        return
+
+    def _appendSubTreeToModel(self, tHandle: str, model: NovelModel) -> None:
+        """Append all active novel documents to a novel model."""
         for handle in SHARED.project.tree.subTree(tHandle):
             if (
                 (node := self._itemIndex[handle])
@@ -131,7 +146,6 @@ class Index:
                 and node.item.isActive
             ):
                 model.append(node)
-            self._novelModels[tHandle] = model
         return
 
     ##

novelwriter/gui/noveltree.py

@@ -323,7 +323,9 @@ class GuiNovelToolBar(QWidget):
     @pyqtSlot()
     def _refreshNovelTree(self) -> None:
         """Rebuild the current tree."""
-        self.novelView.setCurrentNovel(SHARED.project.data.getLastHandle("novelTree"))
+        if rootHandle := self.novelValue.handle:
+            SHARED.project.index.refreshNovelModel(rootHandle)
+        # self.novelView.setCurrentNovel(SHARED.project.data.getLastHandle("novelTree"))
         return
 
     @pyqtSlot()