You can have a look at graphic-walker behind pygwalker, which provides more API to interact. graphic-walker is a react component and pygwalker is the python binding of it.