Refactor player ID dictionary creation in Streamlit app to filter by site (Draftkings and Fanduel), improving data accuracy and clarity in lineup initialization.

src/streamlit_app.py

@@ -176,10 +176,10 @@ def convert_pm_df(array):
     return array.to_csv(index=False).encode('utf-8')
 
 roo_data, sd_roo_data, timestamp = 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['Player'], roo_data['player_id']))
-fd_id_dict_sd = dict(zip(sd_roo_data['Player'], sd_roo_data['player_id']))
+dk_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Draftkings']['Player'], roo_data[roo_data['Site'] == 'Draftkings']['player_id']))
+dk_id_dict_sd = dict(zip(sd_roo_data[sd_roo_data['Site'] == 'Draftkings']['Player'], sd_roo_data[sd_roo_data['Site'] == 'Draftkings']['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_sd = dict(zip(sd_roo_data[sd_roo_data['Site'] == 'Fanduel']['Player'], sd_roo_data[sd_roo_data['Site'] == 'Fanduel']['player_id']))
 hold_display = roo_data
 lineup_display = []
 check_list = []
@@ -196,10 +196,10 @@ with tab1:
                 # i.e. clear values from both square and cube
                 st.cache_data.clear()
                 roo_data, sd_roo_data, timestamp = 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['Player'], roo_data['player_id']))
-                fd_id_dict_sd = dict(zip(sd_roo_data['Player'], sd_roo_data['player_id']))
+                dk_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Draftkings']['Player'], roo_data[roo_data['Site'] == 'Draftkings']['player_id']))
+                dk_id_dict_sd = dict(zip(sd_roo_data[sd_roo_data['Site'] == 'Draftkings']['Player'], sd_roo_data[sd_roo_data['Site'] == 'Draftkings']['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_sd = dict(zip(sd_roo_data[sd_roo_data['Site'] == 'Fanduel']['Player'], sd_roo_data[sd_roo_data['Site'] == 'Fanduel']['player_id']))
                 dk_lineups = init_DK_lineups('Regular')
                 fd_lineups = init_FD_lineups('Regular')
                 hold_display = roo_data
@@ -249,10 +249,10 @@ with tab2:
         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['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['Player'], roo_data['player_id']))
-            fd_id_dict_sd = dict(zip(sd_roo_data['Player'], sd_roo_data['player_id']))
+            dk_id_dict = dict(zip(roo_data[roo_data['Site'] == 'Draftkings']['Player'], roo_data[roo_data['Site'] == 'Draftkings']['player_id']))
+            dk_id_dict_sd = dict(zip(sd_roo_data[sd_roo_data['Site'] == 'Draftkings']['Player'], sd_roo_data[sd_roo_data['Site'] == 'Draftkings']['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_sd = dict(zip(sd_roo_data[sd_roo_data['Site'] == 'Fanduel']['Player'], sd_roo_data[sd_roo_data['Site'] == 'Fanduel']['player_id']))
             hold_display = roo_data
             dk_lineups = init_DK_lineups('Regular')
             fd_lineups = init_FD_lineups('Regular')