generate_image.py