Would it be simple to find a way to use the library outside of Jupyter ?
I don't know anything about HTML/CSS/js, hence cannot really modify the visualize function, but I guess that generating a standalone HTML page would then allow to render it and convert it to an image.