If the media entity has fields, instead of replacing the upload section with the generate section, the generate section appears below the field. This is unexpected and if there are a number of fields, the generate section will be added off screen.
Here is a screen recording of what I am seeing.
It's not the end of the world of course, but if you could figure out how to replace the upload section with the generate section in the same place, it would be a better user experience.
Thanks!
Active
1.0
Code
@thejimbirch I'm not sure I understand. The generate section does replace the upload section.
In your example wouldn't you want the caption field to still appear for a generated image?
Have a look at your field weights. The generate section has a weight of 2. If you are adding custom fields and want them to appear below the generate section, give them a weight that is higher than 2.