What AI tool are you using for the image generation?
It would be also great to have a small icon that indicates wether you already have an image in the album or not. Like an checkmark. And also a counter that tell you how many images there are left to collect (x/y).