Update app.py

app.py

@@ -46,16 +46,11 @@ 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(st.session_state["raw_stats_disp"]):
-            return
-        st.session_state["raw_stats_disp"] = new_df
-
-    df = st.session_state["raw_stats_disp"]
-    df["2P"] = df["Minutes"] + df["FG2M"]
-    st.session_state["raw_stats_disp"] = df
-    st.experimental_rerun()
+def add_column(df):
+    return_df = df
+    return_df = return_df["Minutes"] * return_df["FG2M"]
+    
+    return return_df
 
 @st.cache_resource(ttl = 299)
 def init_baselines():
@@ -650,16 +645,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)]
-    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",
         hide_index=True,
     )
-    add_c(editable_df)
+    st.dataframe(add_column(editable_df))
     st.download_button(
         label="Export Customizable Model",
         data=convert_df_to_csv(player_stats),