Smoother Writing Assistant editing

Writing Assistant now handles Esc, command deletion, command rewriting, and focus more naturally so you can keep typing without losing your place.