When rewriting and editing, it would be nice to have AI output in one when its selected for rewrite so you can pick and choose. Overwriting in place makes it very hard to compare.

One screen for original text the second one has the not merged AI response of that text. Think 4-5 paragraphs at a time if not more in the second screen.