I think related to this, many teachers have mentioned to me a more robust comment writing history so that it is easier to track suggested edits of tone, length, and grammar without starting over. When engaging with the AI in the PR - these quick actions, combined with a version history, and then an ability to accept or reject specific suggestions, would help.