For UI, I might suggest something more “inlined” for the prompt screen. You might find some inspiration from outline writing apps like OmniOutliner [0] or emacs’ org-mode. Those will let you move around heading and groups of rows easily. If you could expand a row into multiple heading rows generated by AI, I could see that being really “in flow” for a writer!
You could imagine someone writing a couple details, hitting some key combo that expands it into a few headings, and then they can move around to the heading they want to expand on and start writing in it. They can recursively expand sentences/rows when ever they need to flesh something out.
It’s hard to get keycombos and the feel right, but I think it would be worth it.