Я не знаю, какое именно всплывающее окно вы используете, но в любом случае, я расскажу вам в целом, как вы можете достичь того, о чем вы просили.
Вы можете отправить событие нажатия кнопки во всплывающее окно, а когда оно закроется, отправить его обратно в свой компонент, а затем использовать
element.focus();
Чтобы снова сосредоточиться на этом элементе.