Build an AI Image Generator Python