Streamlit表单需两次提交才更新session_state？求答疑与解决方法

1. 这是否属于预期行为？ #

不属于预期行为，问题源于代码写法不符合Streamlit的运行逻辑，并非框架本身的设计缺陷。

2. 是否是我的操作存在错误？ #

是的，两个示例均存在逻辑问题：

• 示例1错误：直接将st.text_input的返回值赋值给st.session_state['name']，但Streamlit表单内的组件值在提交前不会同步到全局状态。首次提交时，页面重运行过程中st.text_input返回的仍是初始空值，因此session_state中只会存入空字符串；第二次提交时，组件已缓存第一次提交的值，才会正确写入。
• 示例2错误：回调函数中使用的name变量是页面重运行时的旧值（初始空值）。Streamlit回调函数的触发时机早于组件值的更新，此时name还未被赋值为新输入的内容，导致第一次提交时session_state仍为空。

3. 提交后立即更新session_state的解决方案 #

有两种可靠方式实现提交后立即同步：

方式一：给组件指定key，自动绑定session_state #

Streamlit组件的key参数会自动将组件值与st.session_state[key]绑定，表单提交后会立即同步最新值：

import streamlit as st

# 查看session_state中的所有键值对
s = [f"{k}: {v}" for k, v in st.session_state.items()]
st.write(s)

# 定义表单，给text_input指定key
with st.form("my_form"):
    st.text_input("Name", key="name")
    st.form_submit_button("Submit")

运行效果：首次提交输入内容后，session_state会立即显示["name: Chris"]。

方式二：在回调函数中通过session_state获取组件值 #

如果需要自定义回调逻辑，不要直接使用组件返回的变量，而是通过st.session_state[key]获取最新值（需给组件指定key）：

import streamlit as st

# 查看session_state中的所有键值对
s = [f"{k}: {v}" for k, v in st.session_state.items()]
st.write(s)

# 定义表单
with st.form("my_form"):
    def update():
        # 从session_state中获取组件的最新值
        st.session_state['saved_name'] = st.session_state['input_name']

    st.text_input("Name", key="input_name")
    st.form_submit_button("Submit", on_click=update)

运行效果：首次提交后，session_state会同时包含input_name和saved_name的最新值。

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