st.experimental_dialog feels a bit laggy when closing

[lukasmasuch]

Checklist

• I have searched the existing issues for similar issues.
• I added a very descriptive title to this issue.
• I have provided sufficient information below to help reproduce this issue.

Summary

When closing a dialog created via st.experimental_dialog, there is a short lag when closing the dialog:

Screen.Recording.2024-05-23.at.14.39.33.mov

The same doesn't happen with other native dialogs (e.g. about dialog):

Screen.Recording.2024-05-23.at.14.42.41.mov

Reproducible Code Example

import streamlit as st

@st.experimental_dialog("Hello")
def hello_world():
    st.write("Hello World")

hello_world()

Steps To Reproduce

No response

Expected Behavior

No response

Current Behavior

No response

Is this a regression?

• Yes, this used to work in a previous version.

Debug info

• Streamlit version: 1.34
• Python version: 3.10
• Operating System: MacOS
• Browser: Chrome

Additional Information

No response

[github-actions]

If this issue affects you, please react with a 👍 (thumbs up emoji) to the initial post.

Your feedback helps us prioritize which bugs to investigate and address first.