Refactor data export functionality in app.py by replacing hardcoded column indices with named column mappings for DraftKings and FanDuel, improving code readability and maintainability.

app.py

@@ -443,12 +443,14 @@ with tab3:
 
         if st.button("Prepare data export", key='data_export'):
             data_export = st.session_state.working_seed.copy()
-            # if site_var == 'Draftkings':
-            #     for col_idx in range(6):
-            #         data_export[:, col_idx] = np.array([id_dict.get(player, player) for player in data_export[:, col_idx]])
-            # elif site_var == 'Fanduel':
-            #     for col_idx in range(6):
-            #         data_export[:, col_idx] = np.array([id_dict.get(player, player) for player in data_export[:, col_idx]])
+            if site_var == 'Draftkings':
+                map_columns = ['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3']
+                for col_idx in map_columns:
+                    data_export[col_idx] = data_export[col_idx].map(dk_id_map)
+            elif site_var == 'Fanduel':
+                map_columns = ['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL']
+                for col_idx in map_columns:
+                    data_export[col_idx] = data_export[col_idx].map(fd_id_map)
             st.download_button(
                 label="Export optimals set",
                 data=convert_df(data_export),
@@ -492,11 +494,13 @@ with tab3:
 
     export_file = st.session_state.data_export_display.copy()
     if site_var == 'Draftkings':
-        for col_idx in range(10):
-            export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(dk_id_map)
+        map_columns = ['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3']
+        for col_idx in map_columns:
+            export_file[col_idx] = export_file[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)
+        map_columns = ['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL']
+        for col_idx in map_columns:
+            export_file[col_idx] = export_file[col_idx].map(fd_id_map)
             
     with st.container():
         if st.button("Reset Optimals", key='reset3'):