Add player ID mappings for DraftKings and FanDuel in app.py to enhance data export functionality, ensuring accurate player identification in exported files.

app.py

@@ -70,7 +70,9 @@ def init_baselines():
     roo_data['Salary'] = roo_data['Salary'].astype(int)
     
     dk_roo = roo_data[roo_data['Site'] == 'Draftkings']
+    dk_id_map = dict(zip(dk_roo['player_ID'], dk_roo['Player']))
     fd_roo = roo_data[roo_data['Site'] == 'Fanduel']
+    fd_id_map = dict(zip(fd_roo['player_ID'], fd_roo['Player']))
 
     collection = db["Player_SD_Range_Of_Outcomes"] 
     cursor = collection.find()
@@ -103,7 +105,7 @@ def init_baselines():
     scoring_percentages['DK LevX'] = scoring_percentages['Top Score'].rank(pct=True).astype(float) - scoring_percentages['DK Own%'].rank(pct=True).astype(float)
     scoring_percentages['FD LevX'] = scoring_percentages['Top Score'].rank(pct=True).astype(float) - scoring_percentages['FD Own%'].rank(pct=True).astype(float)
     
-    return roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo
+    return roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo, dk_id_map, fd_id_map
 
 @st.cache_data(ttl = 60)
 def init_DK_lineups(type_var, slate_var):  
@@ -259,7 +261,7 @@ col1, col2 = st.columns([1, 9])
 with col1:
     if st.button("Load/Reset Data", key='reset'):
         st.cache_data.clear()
-        roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo = init_baselines()
+        roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo, dk_id_map, fd_id_map = init_baselines()
         hold_display = roo_data
         dk_lineups = init_DK_lineups('Regular', 'Main')
         fd_lineups = init_FD_lineups('Regular', 'Main')
@@ -276,7 +278,7 @@ with col2:
 
 tab1, tab2, tab3 = st.tabs(["Scoring Percentages", "Player ROO", "Optimals"])
 
-roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo = init_baselines()
+roo_data, sd_roo_data, scoring_percentages, dk_roo, fd_roo, dk_id_map, fd_id_map = init_baselines()
 hold_display = roo_data
 
 with tab1:
@@ -489,12 +491,12 @@ with tab3:
             st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
 
     export_file = st.session_state.data_export_display.copy()
-    # if site_var == 'Draftkings':
-    #     for col_idx in range(6):
-    #         export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict)
-    # elif site_var == 'Fanduel':
-    #     for col_idx in range(6):
-    #         export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict)
+    if site_var == 'Draftkings':
+        for col_idx in range(10):
+            export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(dk_id_map)
+    elif site_var == 'Fanduel':
+        for col_idx in range(9):
+            export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(fd_id_map)
             
     with st.container():
         if st.button("Reset Optimals", key='reset3'):