Skip to content

Instantly share code, notes, and snippets.

@nockn
Forked from tvst/st_rerun.py
Last active August 18, 2021 06:02
Show Gist options
  • Save nockn/e6c8d1cd77e18ae95551507e713c5e06 to your computer and use it in GitHub Desktop.
Save nockn/e6c8d1cd77e18ae95551507e713c5e06 to your computer and use it in GitHub Desktop.
Rerun a Streamlit app from the top! See https://discuss.streamlit.io/t/update-sidebar/531
try:
from streamlit.ScriptRequestQueue import RerunData
from streamlit.ScriptRunner import RerunException
from streamlit.server.Server import Server
import streamlit.ReportThread as ReportThread
except:
from streamlit.script_request_queue import RerunData
from streamlit.script_runner import RerunException
from streamlit.server.server import Server
import streamlit.report_thread as ReportThread
def rerun():
"""Rerun a Streamlit app from the top!"""
widget_states = _get_widget_states()
raise RerunException(RerunData(widget_states))
def _get_widget_states():
# Hack to get the session object from Streamlit.
ctx = ReportThread.get_report_ctx()
session = None
session_infos = Server.get_current()._session_infos.values()
for session_info in session_infos:
if session_info.session._main_dg == ctx.main_dg:
session = session_info.session
if session is None:
raise RuntimeError(
"Oh noes. Couldn't get your Streamlit Session object"
'Are you doing something fancy with threads?')
# Got the session object!
return session._widget_states
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment