Streamlit中按钮触发file_uploader后上传组件消失问题求助

问题说明 #

• 需求：页面初始仅显示“Upload File”按钮，点击按钮后展示file_uploader组件，上传文件后能正常显示选中的文件名，且组件不会消失
• 异常现象：点击按钮后file_uploader可正常显示，但上传文件后组件消失，无法展示文件名，终端首次打印uploaded_file为None；相同逻辑用于selectbox时，选择选项后下拉菜单也会消失

原代码 #

import streamlit as st

if st.button('Upload File'):
    uploaded_file = st.file_uploader("Choose a file")
    print(uploaded_file)
    if uploaded_file is not None:
        st.write("You selected the file:", uploaded_file.name)

问题根源 #

Streamlit基于脚本重新运行的机制处理交互：每次用户点击按钮、上传文件等操作，整个Python脚本都会重新执行一遍。原代码中，file_uploader仅在按钮点击的那次脚本运行中被渲染；当上传文件触发脚本重跑时，st.button()的状态已重置，if条件不成立，file_uploader就不再被渲染，导致组件消失。

解决方案 #

使用Streamlit的session_state持久化按钮的触发状态，以此控制file_uploader的显示与否。session_state会在脚本多次重跑间保留状态值，确保组件不会因交互操作消失。

修正后的代码 #

import streamlit as st

# 初始化会话状态，默认隐藏上传组件
if "show_uploader" not in st.session_state:
    st.session_state.show_uploader = False

# 点击按钮时切换上传组件的显示状态
if st.button('Upload File'):
    st.session_state.show_uploader = not st.session_state.show_uploader

# 根据会话状态决定是否渲染上传组件
if st.session_state.show_uploader:
    uploaded_file = st.file_uploader("Choose a file")
    if uploaded_file is not None:
        st.write("You selected the file:", uploaded_file.name)

代码说明 #

• 通过session_state存储show_uploader布尔值，确保脚本重跑时状态不丢失
• 点击按钮时切换show_uploader的值，实现上传组件的显示/隐藏切换
• 上传文件触发脚本重跑时，session_state.show_uploader仍为True，file_uploader会持续渲染，可正常展示选中的文件名

内容的提问来源于stack exchange，提问作者emily.mi