If you reconnect via WebSockets, grab the whole current state and update the interface.  Also a bit of cleanup.