Streamlit侧边栏输入时自动重置问题求助

问题场景 #

开发Streamlit应用时，侧边栏初始显示文件上传器和Submit按钮，上传文件并点击Submit后，会展示更多输入选项（input1、input2、input3）以及Reset、Final Submit按钮。但在第二步操作时，选择input1等输入项，侧边栏会重置回第一步，之前显示的选项被隐藏。

问题原因 #

Streamlit的核心运行机制是每次用户交互（比如选择下拉框、滑动滑块）都会重新执行整个脚本。原代码仅通过submit = co1.button("Submit")的布尔值控制第二步UI渲染，但按钮的状态不会跨会话保留——当你选择selectbox触发页面重跑时，submit会被重新赋值为False，导致第二步的UI不再渲染，看起来就像重置回了第一步。

解决方案 #

使用Streamlit的st.session_state持久化步骤状态，跟踪当前处于哪个操作阶段，代替单纯依赖按钮的一次性触发状态。

修改后完整代码 #

import streamlit as st
from pathlib import Path

def compo():
    c1,c2 = st.sidebar.columns(2)
    input1 = c1.selectbox(label = "Input 1",options = ['Hello',"You"])
    input2 = c2.selectbox(label = "Input 2", options = ["There","there"])
    input3 = st.sidebar.slider("Number",min_value=1, max_value=10, value=1, step=1)
    st.sidebar.markdown("---")
    return input1, input2, input3

# 初始化会话状态的步骤
if "step" not in st.session_state:
    st.session_state.step = 1

file = st.sidebar.file_uploader('Upload',type=["pdf","txt"])

# Step 1: 初始状态
if st.session_state.step == 1:
    if file is not None:
        # 保存上传文件
        save_folder = 'F:/tmp'
        save_path = Path(save_folder, file.name)
        with open(save_path, mode='wb') as w:
            w.write(file.getvalue())  # 修复原代码变量名错误
        with st.sidebar:
            if save_path.exists():
                st.write("File uploaded")

    co1, co2= st.sidebar.columns([1, 1])
    if co1.button("Submit", use_container_width=True):
        st.session_state.step = 2
        st.rerun()  # 立即重跑切换到Step 2

# Step 2: 提交后的状态
elif st.session_state.step == 2:
    input1, input2, input3 = compo()
    st.sidebar.markdown("---")
    c1,c2 = st.sidebar.columns([1,1])
    
    if c1.button("Reset", use_container_width=True):
        st.session_state.step = 1
        st.rerun()
    
    if c2.button("Final Submit", use_container_width=True):
        placeholder = st.empty()
        if file is not None:
            if (input1 == "Hello") or (input2 == "There"):
                placeholder.empty()
                # 调用处理函数并展示结果
                st.success("处理成功！")
            else:
                placeholder.empty()
                st.error("Sorry!")
        else:
            if (input1 == "You") or (input2 == "there"):
                placeholder.empty()
                # 调用处理函数并展示结果
                st.success("处理成功！")
            else:
                placeholder.empty()
                st.error("Sorry!")

关键修改点 #

1. 会话状态跟踪：用st.session_state.step记录当前操作步骤，初始化为1，点击Submit后切换为2，点击Reset后重置为1
2. 即时刷新：使用st.rerun()在步骤切换时立即重跑脚本，确保UI及时更新
3. 逻辑拆分：将不同步骤的UI逻辑分离，避免重跑时状态丢失
4. 变量修复：修正原代码中uploaded_file.getvalue()的变量名错误，改为file.getvalue()

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