Author here: I could make this open source if anyone wants to collaborate (would need to tidy up code first, I made this in my noob years and its messy as hell)