I have been trying to do that and this is what I have been able to do:
You can find the code in my repo: https://github.com/omrastogi/dsa_questions/blob/master/cs5800-Algorithms/binary_search_trees/bst_visualization.py