Add focus back to chat input after message sent