如何在Streamlit中通过按钮点击终止while loop

你的问题核心在于Streamlit的运行机制和线程阻塞：

1. Streamlit的重跑机制：用户每次点击按钮，整个脚本会从头重新执行一遍。当Record Audio按钮触发while循环后，脚本会一直卡在循环里，主线程被完全阻塞，根本没机会处理Stop Recording按钮的点击事件——因为Stop按钮的代码逻辑要等循环结束才会执行，而循环本身又停不下来。
2. 变量作用域问题：普通变量Record_stop无法跨Streamlit的多次脚本运行实例保存状态，就算Stop按钮修改了它的值，也只是在那次单独的脚本重跑中生效，和之前卡在循环里的脚本实例完全是两个上下文。

下面提供两种可行的实现方式，核心思路都是用Streamlit的Session State管理跨运行的状态，同时避免阻塞主线程。

方案一：线程+Session State（推荐） #

把录音逻辑放到子线程中执行，主线程留出来处理UI交互，通过st.session_state控制线程的启停：

import threading
import streamlit as st

# 初始化Session State变量，用于跨脚本运行保存状态
if "recording" not in st.session_state:
    st.session_state.recording = False
if "audio_data" not in st.session_state:
    st.session_state.audio_data = []

# 录音逻辑封装到函数，交给子线程执行
def record_audio(recorder):
    recorder.start()
    st.session_state.audio_data = []
    while st.session_state.recording:
        frame = recorder.read()
        st.session_state.audio_data.extend(frame)
        print('Recording')
    recorder.stop()

# 假设你已经初始化了recorder对象和col1、col2
with col1:
    if st.button('Record Audio') and not st.session_state.recording:
        st.write('Recording starts')
        st.session_state.recording = True
        # 启动守护线程执行录音，程序退出时线程自动终止
        threading.Thread(target=record_audio, args=(recorder,), daemon=True).start()

with col2:
    if st.button('Stop Recording') and st.session_state.recording:
        st.session_state.recording = False
        st.write('Recording stopped')

方案二：利用Streamlit重跑机制（无线程） #

通过st.experimental_rerun()触发脚本重跑，每次重跑时读取一帧音频，同时检查recording状态：

import streamlit as st

# 初始化Session State变量
if "recording" not in st.session_state:
    st.session_state.recording = False
if "audio_data" not in st.session_state:
    st.session_state.audio_data = []
if "status_display" not in st.session_state:
    st.session_state.status_display = st.empty()

# 假设你已经初始化了recorder对象和col1、col2
with col1:
    if st.button('Record Audio'):
        st.session_state.recording = True
        st.session_state.audio_data = []
        recorder.start()
        st.session_state.status_display.write('Recording starts')

with col2:
    if st.button('Stop Recording'):
        st.session_state.recording = False
        recorder.stop()
        st.session_state.status_display.write('Recording stopped')

# 如果正在录音，读取一帧后触发重跑，继续循环
if st.session_state.recording:
    frame = recorder.read()
    st.session_state.audio_data.extend(frame)
    print('Recording')
    st.experimental_rerun()

• 确保recorder对象是全局的或者存入st.session_state，避免每次脚本重跑都重新初始化。
• 方案一中的守护线程（daemon=True）能保证当Streamlit程序关闭时，录音线程会自动终止，避免资源泄漏。

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