Streamlit多按钮与函数会话管理问题：页面重复执行函数

需求 #

开发功能实现用户输入URL后点击Analyze按钮触发do_something函数，成功后显示聊天输入框，提交聊天内容时执行ans函数并输出结果。

提供代码 #

import streamlit as st
from streamlit_chat import message
import logging

# Custom functions

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

st.markdown("<h1>VTest</h1>", unsafe_allow_html=True)
st.sidebar.header("Input parameter")

def do_something(url):
    st.write("Calc Length")
    slen = len(url)
    return slen

def get_text():
    input_text = st.text_area("You:")
    return input_text

def ans(len,user_input):
    return user_input+" Test "+str(len)

if "analyze" not in st.session_state:
    st.session_state["analyze"] = False

if "submit" not in st.session_state:
    st.session_state["submit"] = False

with st.sidebar.form(key="my_form"):
    URL = st.text_input("Enter URL :")
    analyze_button = st.form_submit_button(label="Analyze")


if str(URL) == "":
    st.warning("Awaiting URL input in sidebar for chat.")

if analyze_button:
    st.session_state["analyze"] = not st.session_state["analyze"]
    if str(URL) != "":
        with st.spinner(
            "Patience ..."
        ):
            slen = do_something(URL)
            
            if slen>0 and st.session_state["analyze"]:
                user_input = get_text()
                submit = st.button("Submit")
                
                if user_input:
                    with st.form(key='my_form'):
                        submit_button = st.form_submit_button('Submit')
                        if submit_button:
                            response = ans(len,user_input)
                            st.write(response)


                else:
                    st.warning("Enter your question above")
            else:
                st.error(
                    f"The greatest teacher, failure is...\n"
                )
                    

    else:
        st.sidebar.warning("Please provide URL ")
        raise Exception("You must provide a URL !")

现象 #

• 预期行为：点击Submit按钮时执行ans函数并在页面输出结果。
• 实际行为：点击后页面重载变空白，do_something函数被重复执行，不符合预期。

问题原因 #

1. 表单Key冲突：侧边栏表单和聊天提交表单使用了相同的key="my_form"，引发组件冲突，导致页面刷新异常。
2. 状态未持久化：do_something的返回值slen未存入st.session_state，每次页面交互都会重新执行该函数。
3. 逻辑结构错误：聊天输入框和提交按钮的渲染被嵌套在analyze_button的点击分支中，Streamlit每次交互都会重新运行整个脚本，导致状态丢失。
4. 参数命名冲突：ans函数的参数len是Python内置函数名，易引发语法错误。

修复后的代码 #

import streamlit as st
import logging

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

st.markdown("<h1>VTest</h1>", unsafe_allow_html=True)
st.sidebar.header("Input parameter")

def do_something(url):
    st.write("Calc Length")
    slen = len(url)
    return slen

def ans(slen, user_input):
    return f"{user_input} Test {slen}"

# 初始化会话状态
if "analyzed" not in st.session_state:
    st.session_state["analyzed"] = False
if "url_length" not in st.session_state:
    st.session_state["url_length"] = 0
if "chat_response" not in st.session_state:
    st.session_state["chat_response"] = ""

# 侧边栏URL输入表单
with st.sidebar.form(key="url_form"):
    URL = st.text_input("Enter URL :")
    analyze_button = st.form_submit_button(label="Analyze")

# 处理Analyze按钮点击
if analyze_button:
    if not URL.strip():
        st.sidebar.warning("Please provide URL ")
    else:
        with st.spinner("Patience ..."):
            slen = do_something(URL)
            if slen > 0:
                st.session_state["analyzed"] = True
                st.session_state["url_length"] = slen
            else:
                st.error("Failed to process URL")

# 显示URL输入提示
if not st.session_state["analyzed"]:
    st.warning("Awaiting URL input in sidebar for chat.")

# 分析成功后显示聊天输入区域
if st.session_state["analyzed"]:
    user_input = st.text_area("You:")
    
    with st.form(key="chat_form"):
        submit_button = st.form_submit_button("Submit")
        if submit_button:
            if user_input.strip():
                st.session_state["chat_response"] = ans(st.session_state["url_length"], user_input)
            else:
                st.warning("Enter your question above")
    
    # 显示聊天结果
    if st.session_state["chat_response"]:
        st.write(st.session_state["chat_response"])

修复说明 #

1. 表单Key区分：将侧边栏表单改为url_form，聊天提交表单改为chat_form，避免组件冲突。
2. 状态持久化：将URL长度slen存入st.session_state["url_length"]，分析状态存入st.session_state["analyzed"]，确保页面刷新时数据不丢失。
3. 重构逻辑结构：将聊天输入区域的渲染逻辑移到analyzed状态判断中，而非按钮点击分支，保证状态持续有效。
4. 修正参数命名：将ans函数的参数len改为slen，避免与内置函数冲突。
5. 优化交互流程：统一使用表单处理提交逻辑，避免混合按钮和表单导致的刷新问题。

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