Feature Request: Return the nodes that have been expanded in a tree

It would be super helpful if the tree returned not just a list of the selected nodes but also a list of which nodes are expanded as is done in streamlit-tree-select (which has other shortcomings and is less supported, so would be great to see it here).

Thanks!