caching dataframe loading and plotting by row

src/display.py

@@ -69,8 +69,8 @@ def display_app():
 
     with column2:
         if grid_response['selected_rows'] is not None and len(grid_response['selected_rows']) > 0:
-            #figure = plot_radar_chart_rows(rows=grid_response['selected_rows'])
-            figure = plot_radar_chart_name(dataframe= dataframe, model_name=grid_response['selected_rows'][0]["model_name"])
+            figure = plot_radar_chart_rows(rows=grid_response['selected_rows'])
+            #figure = plot_radar_chart_name(dataframe= dataframe, model_name=grid_response['selected_rows'][0]["model_name"])
             st.plotly_chart(figure, use_container_width=True)
         else:
             if len(subdata)>0:

src/load_data.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 
-#@st.cache_data
+@st.cache_data
 def load_dataframe() -> pd.DataFrame:
     """
     Load dataframe from the csv file in public directory
@@ -13,7 +13,7 @@ def load_dataframe() -> pd.DataFrame:
     dataframe = dataframe.drop(columns = "Unnamed: 0")
     return dataframe
 
-#@st.cache_data
+@st.cache_data
 def sort_by(dataframe: pd.DataFrame, column_name: str, ascending:bool = False) -> pd.DataFrame:
     """
     Sort the dataframe by column_name