Update app.py

app.py

@@ -48,13 +48,13 @@ sim_all_hold = pd.DataFrame(columns=['Player', 'Prop type', 'Prop', 'Mean_Outcom
 
 def add_c(new_df: pd.DataFrame | None = None):
     if new_df is not None:
-        if new_df.equals(raw_stats_disp):
+        if new_df.equals(st.session_state["raw_stats_disp"]):
             return
-        raw_stats_disp = new_df
+        st.session_state["raw_stats_disp"] = new_df
 
-    df = raw_stats_disp
-    df["2P"] = df["Minutes"] + df["FG2M"]
-    raw_stats_disp = df
+    df = st.session_state["raw_stats_disp"]
+    df["2P"] = df["Minutes"] + df["2FGM"]
+    st.session_state["raw_stats_disp"] = df
     st.experimental_rerun()
 
 @st.cache_resource(ttl = 299)
@@ -650,6 +650,10 @@ with tab6:
     elif split_var6 == 'All':
         team_var6 = player_stats.Team.values.tolist()
     raw_stats_disp = raw_baselines[raw_baselines['Team'].isin(team_var6)]
+    if "raw_stats_disp" not in st.session_state:
+        st.session_state.raw_stats_disp = raw_stats_disp
+    add_c()
+    
     editable_df = st.data_editor(
         raw_stats_disp,
         key="data",