| import streamlit as st | |
| import psutil | |
| # ------------------------------- | |
| # Manage Station state variables | |
| # ------------------------------- | |
| def initialize_session_state(): | |
| defaults = { | |
| "audio_path": None, | |
| "audio_ready": False, | |
| "transcription": "", | |
| "whisper": None | |
| } | |
| for k, v in defaults.items(): | |
| if k not in st.session_state: | |
| st.session_state[k] = v | |
| # 🔍 Show memory info after | |
| def display_memory_once(): | |
| if 'memory_logged' not in st.session_state: | |
| mem = psutil.virtual_memory() | |
| st.markdown(f"**Memory Used:** {mem.percent}%") | |
| st.session_state.memory_logged = True | |
| # Reset the app | |
| def reset_session_state_except_model(): | |
| keys_to_keep = {"classifier", "whisper"} | |
| for key in list(st.session_state.keys()): | |
| if key not in keys_to_keep: | |
| del st.session_state[key] | |