Can the AI rewrite just one paragraph instead of the whole chapter?

Yes — that's exactly what "Rewrite with AI" does. Select any passage in the editor, then choose Improve, Expand, Shorten, Simplify, or type your own instruction (e.g. "make it funnier"). Only the selected text is replaced.