This is so lovely. I desperately wish I could color code all math!!
Even more so if it worked with PDFs.
I already have way too many projects on the backburner right now... wink wink nudge nudge to anyone who wants to take this.
More interesting, you can ask a vision model to color parts of speech in img2img and it works OK for frontier models.