Checkout Dr Racket, I believe you can have images in the comments and even as part of code.

https://docs.racket-lang.org/quick/