Add initial version of image generator
Add get_media in image generator module