Refine data processing in app.py by filtering for 'Main Slate' and 'overall' versions, and enhance data export functionality with player ID mapping for DraftKings and FanDuel. Improved session state handling for working seeds.

app.py

@@ -108,6 +108,8 @@ def init_baselines():
     cursor = collection.find()
 
     raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[raw_display['slate'] == 'Main Slate']
+    raw_display = raw_display[raw_display['version'] == 'overall']
     dk_raw = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '2x%', '3x%', '4x%',
                           'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
     dk_raw['STDev'] = (dk_raw['Ceiling'] - dk_raw['Floor']) / 4
@@ -116,6 +118,8 @@ def init_baselines():
     cursor = collection.find()
 
     raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[raw_display['slate'] == 'Main Slate']
+    raw_display = raw_display[raw_display['version'] == 'overall']
     fd_raw = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '2x%', '3x%', '4x%',
                           'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
     fd_raw['STDev'] = (fd_raw['Ceiling'] - fd_raw['Floor']) / 4
@@ -174,6 +178,8 @@ def sim_contest(Sim_size, seed_frame, maps_dict, sharp_split, Contest_Size):
     return Sim_Winners
 
 dk_raw, fd_raw = init_baselines()
+dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
+fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
 
 tab1, tab2 = st.tabs(['Contest Sims', 'Data Export'])
 with tab2:
@@ -186,6 +192,8 @@ with tab2:
               DK_seed = init_DK_seed_frames()
               FD_seed = init_FD_seed_frames()
               dk_raw, fd_raw = init_baselines()
+              dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
+              fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
               
         slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'))
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
@@ -233,7 +241,16 @@ with tab2:
         
 
         if st.button("Prepare data export", key='data_export'):
+                if 'working_seed' in st.session_state:
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
+                elif 'working_seed' not in st.session_state:
+                    st.session_state.working_seed = DK_seed.copy()
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                 data_export = st.session_state.working_seed.copy()
+                for col in range(9):
+                    data_export[:, col] = np.array([dk_id_dict.get(x, x) for x in data_export[:, col]])
                 st.download_button(
                     label="Export optimals set",
                     data=convert_df(data_export),
@@ -279,6 +296,9 @@ with tab1:
               DK_seed = init_DK_seed_frames()
               FD_seed = init_FD_seed_frames()
               dk_raw, fd_raw = init_baselines()
+              dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
+              fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
+
         sim_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'), key='sim_slate_var1')
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='sim_site_var1')
         if sim_site_var1 == 'Draftkings':
@@ -390,6 +410,8 @@ with tab1:
                 
                # Data Copying
                 st.session_state.Sim_Winner_Export = Sim_Winner_Frame.copy()
+                for col in st.session_state.Sim_Winner_Export.iloc[:, 0:9].columns:
+                    st.session_state.Sim_Winner_Export[col] = st.session_state.Sim_Winner_Export[col].map(dk_id_dict)
                 
                 # Data Copying
                 st.session_state.Sim_Winner_Display = Sim_Winner_Frame.copy()