Streamlit弹窗如何实现孤立等待用户输入而非全页重跑？

开发Streamlit应用时，希望通过弹窗获取用户临时输入，但无法实现弹窗孤立重跑。当前代码如下：

utils.py中的弹窗代码：

@st.fragment
@st.dialog("Enter Your Input")
def get_user_input_popup(prompt_text):
    
    if 'dialog_waiting' not in st.session_state:
        st.session_state.dialog_waiting = True
    
    if st.session_state.dialog_waiting:
        with st.form("popup_form"):
            user_input = st.text_input("Your response:", placeholder="Type here...")
            
            col1, col2 = st.columns(2)
            with col1:
                submit = st.form_submit_button("Submit", type="primary")
            with col2:
                cancel = st.form_submit_button("Cancel")
            
            if submit and user_input.strip():
                st.session_state.dialog_result = user_input.strip()
                st.session_state.dialog_waiting = False
                st.rerun(scope="fragment")
            
            elif cancel:
                st.session_state.dialog_result = "CANCELLED"
                st.session_state.dialog_waiting = False
                st.rerun(scope="fragment")
            
            elif submit:
                st.error("Please enter some text!")
        
        return None
    
    else:
        return st.session_state.dialog_result

page.py中调用代码：

with st.spinner("Waiting for input..."):
    response = get_user_input_popup("Need more info:")

print("\n\n", response, "\n\n", type(response))

原本以为用st.fragment可仅重跑弹窗，但实际触发全页重跑；删除st.rerun代码后，函数直接返回None。如何实现类似常规对话框的等待用户输入功能？

解决方案 #

核心问题在于st.dialog和st.fragment的叠加逻辑不符合预期，且st.rerun(scope="fragment")无法在对话框场景下限制重跑范围。通过调整会话状态判断逻辑，结合对话框的渲染特性，可实现预期的弹窗等待输入效果：

修改后的utils.py代码 #

def get_user_input_popup(prompt_text):
    # 初始化会话状态
    if 'dialog_result' not in st.session_state:
        st.session_state.dialog_result = None
    if 'dialog_waiting' not in st.session_state:
        st.session_state.dialog_waiting = True

    # 若已有结果，返回并清理状态（方便下次调用）
    if not st.session_state.dialog_waiting and st.session_state.dialog_result is not None:
        result = st.session_state.dialog_result
        st.session_state.dialog_result = None
        st.session_state.dialog_waiting = True
        return result

    # 定义对话框内容
    @st.dialog("Enter Your Input")
    def dialog_content():
        with st.form("popup_form"):
            user_input = st.text_input("Your response:", placeholder="Type here...")
            
            col1, col2 = st.columns(2)
            with col1:
                submit = st.form_submit_button("Submit", type="primary")
            with col2:
                cancel = st.form_submit_button("Cancel")
            
            if submit and user_input.strip():
                st.session_state.dialog_result = user_input.strip()
                st.session_state.dialog_waiting = False
                st.rerun()
            elif cancel:
                st.session_state.dialog_result = "CANCELLED"
                st.session_state.dialog_waiting = False
                st.rerun()
            elif submit:
                st.error("Please enter some text!")

    # 仅在等待状态下渲染对话框
    if st.session_state.dialog_waiting:
        dialog_content()
        return None
    else:
        return st.session_state.dialog_result

修改后的page.py调用代码 #

response = get_user_input_popup("Need more info:")

# 仅获取到结果后执行后续逻辑
if response is not None:
    print("\n\n", response, "\n\n", type(response))
    # 此处添加你的业务逻辑
else:
    st.spinner("Waiting for input...")

关键调整说明 #

1. 移除st.fragment：对话框本身是独立UI组件，无需额外包裹fragment，fragment更适合非对话框场景的局部UI更新。
2. 调整状态判断顺序：先检查是否已有输入结果，有则返回并清理状态；无则渲染对话框等待输入。
3. 简化重跑逻辑：使用st.rerun()触发全页重跑，但由于对话框仅在等待状态下渲染，重跑后会直接返回结果，用户感知上和弹窗等待一致。
4. 优化调用逻辑：仅当获取到有效结果时才执行后续操作，避免加载提示一直显示。

内容的提问来源于stack exchange，提问作者user9343456