Used cache

app.py

@@ -10,18 +10,28 @@ import warnings
 
 warnings.filterwarnings('ignore')
 
+st.set_page_config(page_title="Sinhala Embedding Space", page_icon=":bar_chart:")
 # cluster PNG file
 image = Image.open('plots/clusters.png')
 
-chart_data = pd.read_csv(r"data/top_cluster_dataset.csv",dtype={'Headline': str, 'x': np.float64, 'y': np.float64, 'labels': str})
+# Load data
+@st.cache_data(persist='disk')
+def load_data():
+    chart_data = pd.read_csv(r"data/top_cluster_dataset.csv",dtype={'Headline': str, 'x': np.float64, 'y': np.float64, 'labels': str})
+    return chart_data
 
+chart_data = load_data()
 # Create a Streamlit app
-st.set_page_config(page_title="Sinhala Embedding Space", page_icon=":bar_chart:")
 
 # Define tabs
 tabs = ["Clustering Results","Sentences Similarity"]
 selected_tab = st.sidebar.radio("Select a Tab", tabs)
 
+@st.cache_data(persist='disk')
+def get_altair_chart():
+    chart = alt.Chart(chart_data).mark_circle(size=60).encode(x='x', y='y', color='labels', tooltip=['Headline']).interactive()
+    return chart
+
 # Main content
 if selected_tab == "Sentences Similarity":
     sample_sentences = chart_data['Headline'].sample(10, random_state=1).tolist()
@@ -66,7 +76,7 @@ elif selected_tab == "Clustering Results":
     with st.spinner('Loading Interactive Results...'):
         # Display Altair chart
         st.subheader("Interactive Chart")
-        chart = alt.Chart(chart_data).mark_circle(size=60).encode(x='x', y='y', color='labels', tooltip=['Headline']).interactive()
+        chart = get_altair_chart()
         st.altair_chart(chart, use_container_width=True)
     
     # Dropdown functionality to update DataFrame