Google ai studio has been doing this pretty well for mockups for me at least. I do see people running out of their already limited tokens using this though, unless they silo them to another collection of tokens.

Edit: I see they have a new token tier 'Claude Design' so that's good.