visualization – Visualizing directories–a cleaner view of a helpful older post

I am looking to make graphics of a folder hierarchy, and this old post was very helpful. However, this method gives the whole file path for each node, which makes a complete mess in the directory I want to use it. It must not be too tricky, but I cannot figure out how to get just the last part of the path from each node of the tree structure.

In the abstract, if I have tree = a(b(c,d,e),g(h)), I believe I want some way of performing F(tree) = F(a)(F(b)(F(c),F(d),F(e)),F(g)(F(h))). I am struggling to implement.

In the concrete, I mildly adapted the code from the linked question above:

readDir(currentDirectory_, 0) := FileNameTake(currentDirectory);

readDir(currentDirectory_, level_) := 
Module({}, SetDirectory(currentDirectory);
joinedFiles = 
FileNameJoin({currentDirectory, #}) & /@ FileNames();
unjoinedFiles = FileNames();

(*do a logical test;
if the thing in a directory is a directory,*)
perFile(file_) := 
If(DirectoryQ(file),(*condition*)
 file @@ readDir(file, level - 1),(*true*)Sequence @@ {});(*false*)
perFile /@ joinedFiles);

treeDir(dir_) := TreeForm(dir @@ readDir(dir, 11))

treeDir("/Users/vironevaeh/a")

This yields the nice tree, but with the less-nice filepaths built in:
enter image description here

Can I either: (1) apply a function to the nested expression that gives me the first graphic, or (2) rewrite the generative code in another way, to give me the following from the directory input:

enter image description here