Inline text editing with AI

I'm currently writing an extensive document/narrative that is about 40 pages long and I need to have AI help me create it. What I am looking for is something where I can live edit a document like a canvas and at the same time also chat with AI with requests to alter, rewrite, or revise certain sentences of the document as well. Does anybody know of any tools that can do that? So far, I've tried context.ai and it seemed promising. However, its functionality seems too limited after I signed up for it and doesn't exactly work the way I had intended it to. I also tried artifacts in Claude, but I've been having some difficulty there as well.

Does anyone know of any software/AI Platforms that have this functionality of live-text editing manually, coupled with real-time edits by AI chat? Preferably, also having a function where I can upload knowledge to it for referencing certain data points?