Streamlit 会话状态管理嵌套按钮：按顺序完成任务

2024-03-12 12:59:44

利用 Streamlit 会话状态管理嵌套按钮

简介

Streamlit 是一个开源 Python 库，用于构建交互式 Web 应用程序。会话状态是一种特殊类型的数据，用于在用户会话期间存储和检索信息。本教程将指导你如何使用会话状态来管理嵌套按钮，从而允许用户按顺序完成多项任务。

创建会话状态变量

首先，你需要创建会话状态变量来存储按钮的状态。这可以通过 st.session_state 对象来实现，如下所示：

if 'rec_btn' not in st.session_state:
    st.session_state.rec_btn = False

此代码检查会话状态中是否存在名为 rec_btn 的变量。如果没有，它将创建一个新的布尔变量并将其设置为 False。

回调函数

下一步，你需要创建一个回调函数，当按钮被点击时执行。该函数将更新会话状态变量以指示按钮已被点击。

def callback():
    st.session_state.rec_btn = True

创建嵌套按钮

现在，你可以使用会话状态变量来创建嵌套按钮。嵌套按钮是一个在另一个按钮内显示的按钮。以下是创建嵌套按钮的示例：

if st.button('RECOMMEND', key='rec_btn'):
    col1, col2, col3 = st.columns(3)
    with col1:
        st.image(output_images[0])
        st.markdown(output_names[0].upper())
        review = st.text_input(f"How much you liked the movie {output_names[0]}")
        if st.button('submit', on_click=callback):
            # 代码...

在上面的示例中，当 RECOMMEND 按钮被点击时，它将显示一个包含图像、文本框和 submit 按钮的列。当 submit 按钮被点击时，它将调用 callback 函数，该函数将更新会话状态变量 rec_btn。

错误解决

你提到的错误：“Values for st.button, st.download_button, st.file_uploader, and st.form cannot be set using st.session_state.” 是因为 Streamlit 限制使用 st.session_state 来设置某些组件的状态，包括按钮。要解决此错误，你应该使用 st.session_state 来存储按钮的状态，而不是直接设置按钮的状态。