Update app.py

app.py

@@ -46,6 +46,17 @@ prop_table_options = ['points', 'threes', 'rebounds', 'assists', 'blocks', 'stea
 all_sim_vars = ['points', 'rebounds', 'assists', 'threes', 'PRA', 'points+rebounds', 'points+assists', 'rebounds+assists']
 sim_all_hold = pd.DataFrame(columns=['Player', 'Prop type', 'Prop', 'Mean_Outcome', 'Imp Over', 'Over%', 'Imp Under', 'Under%', 'Bet?', 'Edge'])
 
+def add_c(new_df: pd.DataFrame | None = None):
+    if new_df is not None:
+        if new_df.equals(raw_stats_disp):
+            return
+        raw_stats_disp = new_df
+
+    df = raw_stats_disp
+    df["2P"] = df["Minutes"] + df["FG2M"]
+    raw_stats_disp = df
+    st.experimental_rerun()
+
 @st.cache_resource(ttl = 299)
 def init_baselines():
     sh = gcservice_account.open_by_url(master_hold)
@@ -639,8 +650,12 @@ with tab6:
     elif split_var6 == 'All':
         team_var6 = player_stats.Team.values.tolist()
     raw_stats_disp = raw_baselines[raw_baselines['Team'].isin(team_var6)]
-    raw_stats_disp['2P'] = raw_stats_disp['Minutes'] * raw_stats_disp['FG2M']
-    st.data_editor(raw_stats_disp, use_container_width = True)
+    editable_df = st.data_editor(
+        raw_stats_disp,
+        key="data",
+        hide_index=True,
+    )
+    add_c(editable_df)
     st.download_button(
         label="Export Customizable Model",
         data=convert_df_to_csv(player_stats),