Yep, asking questions directly in the notebook, then editing the answer and executing it would be very powerful. I think `jupyter-ai` is close to this, judging by description, but I didn't try it.