OpenAI gym custom reinforcement learning env help

I need please write your own way to animate the env from scratch, all other files (env, init...) can stay the same, provide a function that takes screenshots of the episodes using the camera [login to view URL] here is a custom RL env with openAI gym, as you can see, he animated the 2D env. I need someone to ignore that, and write up their own code to animate it by tomorrow if possible. so your own code to produce the same results