Implement conditional data export logic in app.py: enhance the data export functionality by incorporating slate type (Regular or Showdown) for both Draftkings and Fanduel, ensuring accurate player ID mapping and improving data handling during export.

app.py

@@ -420,13 +420,27 @@ with tab2:
                 elif player_var1 == 'Full Slate':
                         player_var2 = fd_raw.Player.values.tolist()
         if st.button("Prepare data export", key='data_export'):
-            data_export = st.session_state.working_seed.copy()
+    
             if site_var1 == 'Draftkings':
-                for col_idx in range(8):
-                    data_export[:, col_idx] = np.array([id_dict.get(player, player) for player in data_export[:, col_idx]])
+                if slate_type_var1 == 'Regular':
+                    data_export = init_DK_lineups(slate_var1)
+                    for col_idx in range(8):
+                        data_export[:, col_idx] = np.array([id_dict.get(player, player) for player in data_export[:, col_idx]])
+                elif slate_type_var1 == 'Showdown':
+                    data_export = init_DK_SD_lineups(slate_var1)
+                    for col_idx in range(5):
+                        data_export[:, col_idx] = np.array([id_dict_sd.get(player, player) for player in data_export[:, col_idx]])
+                
             elif site_var1 == 'Fanduel':
-                for col_idx in range(9):
-                    data_export[:, col_idx] = np.array([id_dict.get(player, player) for player in data_export[:, col_idx]])
+                if slate_type_var1 == 'Regular':
+                    data_export = init_FD_lineups(slate_var1)
+                    for col_idx in range(9):
+                        data_export[:, col_idx] = np.array([id_dict.get(player, player) for player in data_export[:, col_idx]])
+                elif slate_type_var1 == 'Showdown':
+                    data_export = init_FD_SD_lineups(slate_var1)
+                    for col_idx in range(5):
+                        data_export[:, col_idx] = np.array([id_dict_sd.get(player, player) for player in data_export[:, col_idx]])
+                
             st.download_button(
                 label="Export optimals set",
                 data=convert_df(data_export),