I'd prefer a UX where if I start typing, the chatbot immediately stops talking, with a play button which I can use to let it finish predicting whatever it was on about.

It's a small tweak, but it would be a significant improvement. Especially when I'm trying to type multi-line input and forget to hit Shift-Enter: there's this immediate context switch to get to the mouse and hit a small button, when what I want to do is just keep typing. Normally that would mean the bot wouldn't even start replying before the pause signal gets sent to it.