minor updates

src/myfasthtml/controls/DataGridsManager.py

@@ -320,7 +320,7 @@ class DataGridsManager(SingleInstance, DatagridMetadataProvider):
   def _mk_tree(self):
     tree = TreeView(self, _id="-treeview")
     for element in self._state.elements:
-      parent_id = tree.ensure_path(element.namespace)
+      parent_id = tree.ensure_path(element.namespace, node_type="folder")
       tree.add_node(TreeNode(id=element.document_id,
                              label=element.name,
                              type=element.type,

src/myfasthtml/controls/TreeView.py

@@ -341,10 +341,11 @@ class TreeView(MultipleInstance):
     return self
   
   def _start_rename(self, node_id: str):
-    """Start renaming a node (sets editing state)."""
+    """Start renaming a node (sets editing state and selection)."""
     if node_id not in self._state.items:
       raise ValueError(f"Node {node_id} does not exist")
-    
+
+    self._state.selected = node_id
     self._state.editing = node_id
     return self