Streamlit中st.file_uploader嵌套按钮后返回None问题咨询

Streamlit的核心运行逻辑是每次用户交互（点击按钮、上传文件等）都会重新执行一遍整个脚本。你的代码里，点击"Upload Image"按钮后，脚本重新运行，进入if upload_button:块，渲染出文件上传组件。但当你选择文件上传时，页面会再次触发刷新，这时候upload_button的状态会被重置为False（因为Streamlit的按钮点击状态是一次性的，刷新后就丢失了），导致后续的脚本执行不会进入if块，文件上传组件根本没被渲染，所以image_file自然就是None。

方案1：直接展示文件上传组件（最简洁） #

既然点击按钮只是为了唤起上传功能，不如直接把file_uploader放在外层，不需要按钮触发，这样每次刷新都会渲染它，上传文件后状态也能保留：

import streamlit as st 
import PIL as pil 

st.title("Dogs and Cats")

# 直接显示文件上传组件，不需要按钮触发
image_file = st.file_uploader("Upload image", type=["jpg","jpeg"])

if image_file is not None:
    org_image = pil.Image.open(image_file, mode='r')
    st.text("Uploaded image")
    st.image(org_image, caption='Image for Prediction')
    pred_button = st.button("Perform Prediction")
    if pred_button:
        st.image(org_image, caption='Predicted Image')
        st.write("The class is : ")

方案2：用session_state保存按钮状态（保留按钮触发的交互逻辑） #

如果一定要保留"Upload Image"按钮的触发逻辑，可以用Streamlit的session_state来保存按钮的状态，确保刷新后依然能进入渲染文件上传组件的分支：

import streamlit as st 
import PIL as pil 

st.title("Dogs and Cats")

# 初始化session_state中的状态
if "show_uploader" not in st.session_state:
    st.session_state.show_uploader = False

# 点击按钮时切换状态
if st.button("Upload Image"):
    st.session_state.show_uploader = not st.session_state.show_uploader

# 根据session_state的状态决定是否显示上传组件
if st.session_state.show_uploader:
    image_file = st.file_uploader("Upload image", type=["jpg","jpeg"])

    if image_file is not None:
        org_image = pil.Image.open(image_file, mode='r')
        st.text("Uploaded image")
        st.image(org_image, caption='Image for Prediction')
        pred_button = st.button("Perform Prediction")
        if pred_button:
            st.image(org_image, caption='Predicted Image')
            st.write("The class is : ")

内容的提问来源于stack exchange，提问作者Muhammad Ali Saqib