oh this is rad! i made a google colab that did this a while ago but i applaud you for doing it directly in js :D