如何将回调函数返回值关联到Streamlit动态生成的按钮？

问题描述 #

从谷歌表格读取电影标题，通过循环批量生成Streamlit按钮，所有按钮共用同一个回调函数。该回调函数会返回海报路径，希望将返回值存入变量并用于展示电影海报。现有代码如下，询问如何将点击事件与动态生成的按钮关联，能否在生成按钮的同一循环内实现？

rh_movie_list = data[user].to_list()
for title in rh_movie_list:
    if type(title) is float:
        break
    with col1:
        st.button(
            title, key=title, on_click=get_movie_art, args=(title,),
            use_container_width=True
        )

def get_movie_art(poster):
    poster_path = f'{movie_art_endpoint}{poster}'
    return poster_path

解决方案 #

Streamlit的按钮回调函数无法直接返回值给主程序，必须借助**会话状态（st.session_state）**来存储交互后的结果，同时可以在生成按钮的同一循环内完成点击事件的关联，具体步骤如下：

1. 初始化会话状态变量 #

在脚本开头添加代码，用来存储选中的海报路径：

# 初始化会话状态，存储选中的海报路径
if 'selected_poster' not in st.session_state:
    st.session_state.selected_poster = None

2. 修改回调函数，存储路径到会话状态 #

将原回调函数的返回逻辑改为写入会话状态，这样点击按钮后，路径会被保存下来：

def get_movie_art(title):
    # 修正原函数参数名错误：参数应为电影标题title，而非poster
    poster_path = f'{movie_art_endpoint}{title}'
    # 将路径存入会话状态
    st.session_state.selected_poster = poster_path

3. 保留循环生成按钮的逻辑（关联点击事件） #

你原本的循环生成按钮的代码已经实现了点击事件的关联——通过on_click指定回调函数，args传递电影标题参数，只需优化空值判断逻辑（用更严谨的pd.isna替代类型判断）：

import pandas as pd

rh_movie_list = data[user].to_list()
for title in rh_movie_list:
    # 用pandas的空值判断替代类型检查，适配表格空值场景
    if pd.isna(title):
        break
    with col1:
        st.button(
            title, key=title, on_click=get_movie_art, args=(title,),
            use_container_width=True
        )

4. 展示选中的海报 #

在页面任意位置（比如按钮下方）添加代码，根据会话状态中的路径展示海报：

# 当会话状态中有选中的海报路径时，展示海报
if st.session_state.selected_poster:
    st.image(st.session_state.selected_poster, caption="选中的电影海报")

关键说明 #

• 可以在生成按钮的同一循环内完成点击事件关联：循环中每个按钮通过on_click和args参数，已经和回调函数绑定，点击后会触发对应逻辑。
• 会话状态是Streamlit中存储跨交互状态的核心方式，所有用户交互后的结果都需要通过它来传递，而非普通函数返回值。

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