Refactor player ID dictionary creation in Streamlit app to eliminate site-specific filtering for Draftkings and Fanduel, enhancing code simplicity and consistency. Reuse Draftkings dictionary for Fanduel ID mapping across data loading and reset functionalities.

src/streamlit_app.py

@@ -173,9 +173,9 @@ def convert_pm_df(array):
     return array.to_csv(index=False).encode('utf-8')
 
 roo_data, sd_roo_data = init_baselines()
-dk_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Draftkings']['Player'], roo_data[roo_data['Site'] == 'Draftkings']['player_id']))
+dk_id_dict = dict(zip(roo_data['Player'], roo_data['player_id']))
 dk_id_dict_sd = dict(zip(sd_roo_data['Player'], sd_roo_data['player_id']))
-fd_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Fanduel']['Player'], roo_data[roo_data['Site'] == 'Fanduel']['player_id']))
+fd_id_dict = dk_id_dict
 fd_id_dict_sd = dk_id_dict_sd
 hold_display = roo_data
 
@@ -184,9 +184,9 @@ with app_load_reset_column:
     if st.button("Load/Reset Data", key='reset_data_button'):
         st.cache_data.clear()
         roo_data, sd_roo_data = init_baselines()
-        dk_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Draftkings']['Player'], roo_data[roo_data['Site'] == 'Draftkings']['player_id']))
+        dk_id_dict = dict(zip(roo_data['Player'], roo_data['player_id']))
         dk_id_dict_sd = dict(zip(sd_roo_data['Player'], sd_roo_data['player_id']))
-        fd_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Fanduel']['Player'], roo_data[roo_data['Site'] == 'Fanduel']['player_id']))
+        fd_id_dict = dk_id_dict
         fd_id_dict_sd = dk_id_dict_sd
         dk_lineups = init_DK_lineups('Regular')
         fd_lineups = init_FD_lineups('Regular')
@@ -216,9 +216,9 @@ if selected_tab == "Player ROO":
         if st.button("Reset Data", key='reset1'):
                 st.cache_data.clear()
                 roo_data, sd_roo_data = init_baselines()
-                dk_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Draftkings']['Player'], roo_data[roo_data['Site'] == 'Draftkings']['player_id']))
+                dk_id_dict = dict(zip(roo_data['Player'], roo_data['player_id']))
                 dk_id_dict_sd = dict(zip(sd_roo_data['Player'], sd_roo_data['player_id']))
-                fd_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Fanduel']['Player'], roo_data[roo_data['Site'] == 'Fanduel']['player_id']))
+                fd_id_dict = dk_id_dict
                 fd_id_dict_sd = dk_id_dict_sd
                 dk_lineups = init_DK_lineups('Regular')
                 fd_lineups = init_FD_lineups('Regular')
@@ -276,15 +276,14 @@ if selected_tab == "Optimals":
     with st.expander("Info and Filters"):
         if st.button("Load/Reset Data", key='reset2'):
             st.cache_data.clear()
-            roo_data, sd_roo_data, timestamp = init_baselines()
-            dk_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Draftkings']['Player'], roo_data[roo_data['Site'] == 'Draftkings']['player_id']))
+            roo_data, sd_roo_data = init_baselines()
+            dk_id_dict = dict(zip(roo_data['Player'], roo_data['player_id']))
             dk_id_dict_sd = dict(zip(sd_roo_data['Player'], sd_roo_data['player_id']))
-            fd_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Fanduel']['Player'], roo_data[roo_data['Site'] == 'Fanduel']['player_id']))
+            fd_id_dict = dk_id_dict
             fd_id_dict_sd = dk_id_dict_sd
             hold_display = roo_data
             dk_lineups = init_DK_lineups('Regular')
             fd_lineups = init_FD_lineups('Regular')
-            t_stamp = f"Last Update: " + str(timestamp) + f" CST"
             for key in st.session_state.keys():
                 del st.session_state[key]