Как насчет сохранения выбранной плитки вместо логического значения, независимо от того, открыто диалоговое окно или нет? И если у вас есть выбранная плитка, вы отображаете содержимое в диалоговом окне?
Я изменил вашу песочницу, чтобы отразить это: https://codesandbox.io/s/nameless-cache-75sns