If the author reads this we’ve used VPython for simple simulation visualizations such as this and it has worked great both inside and outside of Jupyter notebooks.

http://www.glowscript.org/docs/VPythonDocs/index.html