AI game screen reference control workflow - critique wanted

I built a small workflow/control system for AI-generated game screen references. The main problem I am trying to solve: direct image prompts often make nice-looking screens that are hard to turn into