Start and stop a streaming fragment
Streamlit lets you turn functions into fragments, which can rerun independently from the full script. Additionally, you can tell Streamlit to rerun a fragment at a set time interval. This is great for streaming data or monitoring processes. You may want the user to start and stop this live streaming. To do this, programmatically set the run_every
parameter for your fragment.
Applied concepts
- Use a fragment to stream live data.
- Start and stop a fragment from automatically rerunning.
Prerequisites
-
The following must be installed in your Python environment:
streamlit>=1.37.0
-
You should have a clean working directory called
your-repository
. -
You should have a basic understanding of fragments.
Summary
In this example, you'll build an app that streams two data series in a line chart. Your app will gather recent data on the first load of a session and statically display the line chart. Two buttons in the sidebar will allow users to start and stop data streaming to update the chart in real time. You'll use a fragment to manage the frequency and scope of the live updates.
Here's a look at what you'll build:
import streamlit as st
import pandas as pd
import numpy as np
from datetime import datetime, timedelta
def get_recent_data(last_timestamp):
"""Generate and return data from last timestamp to now, at most 60 seconds."""
now = datetime.now()
if now - last_timestamp > timedelta(seconds=60):
last_timestamp = now - timedelta(seconds=60)
sample_time = timedelta(seconds=0.5) # time between data points
next_timestamp = last_timestamp + sample_time
timestamps = np.arange(next_timestamp, now, sample_time)
sample_values = np.random.randn(len(timestamps), 2)
data = pd.DataFrame(sample_values, index=timestamps, columns=["A", "B"])
return data
if "data" not in st.session_state:
st.session_state.data = get_recent_data(datetime.now() - timedelta(seconds=60))
if "stream" not in st.session_state:
st.session_state.stream = False
def toggle_streaming():
st.session_state.stream = not st.session_state.stream
st.title("Data feed")
st.sidebar.slider(
"Check for updates every: (seconds)", 0.5, 5.0, value=1.0, key="run_every"
)
st.sidebar.button(
"Start streaming", disabled=st.session_state.stream, on_click=toggle_streaming
)
st.sidebar.button(
"Stop streaming", disabled=not st.session_state.stream, on_click=toggle_streaming
)
if st.session_state.stream is True:
run_every = st.session_state.run_every
else:
run_every = None
@st.fragment(run_every=run_every)
def show_latest_data():
last_timestamp = st.session_state.data.index[-1]
st.session_state.data = pd.concat(
[st.session_state.data, get_recent_data(last_timestamp)]
)
st.session_state.data = st.session_state.data[-100:]
st.line_chart(st.session_state.data)
show_latest_data()
Build the example
Initialize your app
-
In
your_repository
, create a file namedapp.py
. -
In a terminal, change directories to
your_repository
and start your app.streamlit run app.py
Your app will be blank since you still need to add code.
-
In
app.py
, write the following:import streamlit as st import pandas as pd import numpy as np from datetime import datetime, timedelta
You'll be using these libraries as follows:
- You'll work with two data series in a
pandas.DataFrame
. - You'll generate random data with
numpy
. - The data will have
datetime.datetime
index values.
- You'll work with two data series in a
-
Save your
app.py
file and view your running app. -
Click "Always rerun" or hit your "A" key in your running app.
Your running preview will automatically update as you save changes to
app.py
. Your preview will still be blank. Return to your code.
Build a function to generate random, recent data
To begin with, you'll define a function to randomly generate some data for two time series, which you'll call "A" and "B." It's okay to skip this section if you just want to copy the function.
def get_recent_data(last_timestamp):
"""Generate and return data from last timestamp to now, at most 60 seconds."""
now = datetime.now()
if now - last_timestamp > timedelta(seconds=60):
last_timestamp = now - timedelta(seconds=60)
sample_time = timedelta(seconds=0.5) # time between data points
next_timestamp = last_timestamp + sample_time
timestamps = np.arange(next_timestamp, now, sample_time)
sample_values = np.random.randn(len(timestamps), 2)
data = pd.DataFrame(sample_values, index=timestamps, columns=["A", "B"])
return data
-
Start your function definition.
def get_recent_data(last_timestamp): """Generate and return data from last timestamp to now, at most 60 seconds."""
You'll pass the timestamp of your most recent datapoint to your data-generating function. Your function will use this to only return new data.
-
Get the current time and adjust the last timestamp if it is over 60 seconds ago.
now = datetime.now() if now - last_timestamp > timedelta(seconds=60): last_timestamp = now - timedelta(seconds=60)
By updating the last timestamp, you'll ensure the function never returns more than 60 seconds of data.
-
Declare a new variable,
sample_time
, to define the time between datapoints. Calculate the timestamp of the first, new datapoint.sample_time = timedelta(seconds=0.5) # time between data points next_timestamp = last_timestamp + sample_time
-
Create a
datetime.datetime
index and generate two data series of the same length.timestamps = np.arange(next_timestamp, now, sample_time) sample_values = np.random.randn(len(timestamps), 2)
-
Combine the data series with the index into a
pandas.DataFrame
and return the data.data = pd.DataFrame(sample_values, index=timestamps, columns=["A", "B"]) return data
-
(Optional) Test out your function by calling it and displaying the data.
data = get_recent_data(datetime.now() - timedelta(seconds=60)) data
Save your
app.py
file to see the preview. Delete these two lines when finished.
Initialize Session State values for your app
Since you will dynamically change the run_every
parameter of @st.fragment()
, you'll need to initialize the associated variables and Session State values before defining your fragment function. Your fragment function will also read and update values in Session State, so you can define those now to make the fragment function easier to understand.
-
Initialize your data for the first app load in a session.
if "data" not in st.session_state: st.session_state.data = get_recent_data(datetime.now() - timedelta(seconds=60))
Your app will display this initial data in a static line chart before a user starts streaming data.
-
Initialize
"stream"
in Session State to turn streaming on and off. Set the default to off (False
).if "stream" not in st.session_state: st.session_state.stream = False
-
Create a callback function to toggle
"stream"
betweenTrue
andFalse
.def toggle_streaming(): st.session_state.stream = not st.session_state.stream
-
Add a title to your app.
st.title("Data feed")
-
Add a slider to the sidebar to set how frequently to check for data while streaming.
st.sidebar.slider( "Check for updates every: (seconds)", 0.5, 5.0, value=1.0, key="run_every" )
-
Add buttons to the sidebar to turn streaming on and off.
st.sidebar.button( "Start streaming", disabled=st.session_state.stream, on_click=toggle_streaming ) st.sidebar.button( "Stop streaming", disabled=not st.session_state.stream, on_click=toggle_streaming )
Both functions use the same callback to toggle
"stream"
in Session State. Use the current value"stream"
to disable one of the buttons. This ensures the buttons are always consistent with the current state; "Start streaming" is only clickable when streaming is off, and "Stop streaming" is only clickable when streaming is on. The buttons also provide status to the user by highlighting which action is available to them. -
Create and set a new variable,
run_every
, that will determine whether or not the fragment function will rerun automatically (and how fast).if st.session_state.stream is True: run_every = st.session_state.run_every else: run_every = None
Build a fragment function to stream data
To allow the user to turn data streaming on and off, you must set the run_every
parameter in the @st.fragment()
decorator.
@st.fragment(run_every=run_every)
def show_latest_data():
last_timestamp = st.session_state.data.index[-1]
st.session_state.data = pd.concat(
[st.session_state.data, get_recent_data(last_timestamp)]
)
st.session_state.data = st.session_state.data[-100:]
st.line_chart(st.session_state.data)
-
Use an
@st.fragment
decorator and start your function definition.@st.fragment(run_every=run_every) def show_latest_data():
Use the
run_every
variable declared above to set the parameter of the same name. -
Retrieve the timestamp of the last datapoint in Session State.
last_timestamp = st.session_state.data.index[-1]
-
Update the data in Session State and trim to keep only the last 100 timestamps.
st.session_state.data = pd.concat( [st.session_state.data, get_recent_data(last_timestamp)] ) st.session_state.data = st.session_state.data[-100:]
-
Show the data in a line chart.
st.line_chart(st.session_state.data)
Your fragment-function definition is complete.
Call and test out your fragment function
-
Call your function at the bottom of your code.
show_latest_data()
-
Test out your app by clicking "Start streaming." Try adjusting the frequency of updates.
Next steps
Try adjusting the frequency of data generation or how much data is kept in Session State. Within get_recent_data
try setting sample_time
with a widget.
Try using st.plotly_chart or st.altair_chart to add labels to your chart.
Still have questions?
Our forums are full of helpful information and Streamlit experts.