Streamlit状态切换异常：上传文件或点击按钮后旧输出未消失

核心问题 #

• 文件上传器状态被覆盖：直接将st.file_uploader()赋值给st.session_state['Uploaded']，导致每次页面交互（比如点击按钮）时，上传器重置为未上传状态，覆盖之前的session_state值。
• 状态分支逻辑冲突：当button_name为generate_zip时，代码直接修改button_name为download_zip_or_run_pipeline，同一轮渲染中两个状态分支的代码都会执行，导致内容同时显示。
• 按钮点击后状态未正确维持：点击"Continue without Downloading"后，页面刷新时上传器重置，导致Uploaded被设为None，状态逻辑出错。

修复后的代码 #

import streamlit as st

# 初始化session_state
if 'button_name' not in st.session_state:
    st.session_state['button_name'] = None
if 'Uploaded' not in st.session_state:
    st.session_state['Uploaded'] = None
if 'zipped_path' not in st.session_state:
    st.session_state['zipped_path'] = "dummy.zip"  # 替换为实际生成的zip路径


class DummyFile:
    def __init__(self, name):
        self.name = name


# 仅在有新文件上传时更新session_state，避免交互时覆盖状态
uploaded_file = st.file_uploader("", type=["pdf", "zip"])
if uploaded_file is not None:
    st.session_state['Uploaded'] = uploaded_file

# 仅在button_name未手动设置时，根据上传文件更新状态
if st.session_state['Uploaded'] is not None and st.session_state['button_name'] is None:
    if st.session_state['Uploaded'].name.endswith(".zip"):
        st.session_state['button_name'] = "main"
    elif st.session_state['Uploaded'].name.endswith(".pdf"):
        st.session_state['button_name'] = "generate_zip"


# 使用elif控制流程，确保同一轮渲染只执行一个状态分支
if st.session_state['button_name'] == "main":
    st.markdown("main")

elif st.session_state['button_name'] == "generate_zip":
    st.markdown("generate_zip")
    # 修改状态后立即刷新页面，避免同时显示两个分支内容
    st.session_state['button_name'] = "download_zip_or_run_pipeline"
    st.rerun()

elif st.session_state['button_name'] == "download_zip_or_run_pipeline":
    col3, col4 = st.columns(2)

    with col3:
        if st.button("Continue without Downloading"):
            st.session_state['Uploaded'] = DummyFile("dummy.zip")
            st.session_state['button_name'] = "main"
            st.rerun()  # 立即进入main流程

    with col4:
        # 增加路径存在判断，避免报错
        if st.session_state['zipped_path']:
            with open(st.session_state['zipped_path'], 'rb') as file_data:
                st.download_button('Download Zipped Images', file_data, file_name="images.zip")
        else:
            st.markdown("Zip文件路径未配置")

st.write(st.session_state)

关键修复点 #

• 文件上传状态管理：将上传器返回值先存入临时变量，仅当有新文件时更新session_state，避免交互时覆盖已有状态。
• 流程分支控制：用elif替代独立if，确保同一轮渲染只执行一个分支；状态变更后调用st.rerun()，立即刷新页面使新状态生效。
• 按钮点击状态维持：点击按钮修改状态后调用st.rerun()，跳过上传器重置的影响，直接进入目标流程。

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