Refactor data export logic in app.py to differentiate array indexing for Draftkings and Fanduel, ensuring correct data filtering based on selected site.

app.py

@@ -656,8 +656,12 @@ with tab2:
 
         if st.button("Prepare data export", key='data_export'):
                 if 'working_seed' in st.session_state:
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 10], stack_var2)]
+                    if site_var1 == 'Draftkings':
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], team_var2)]
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 13], stack_var2)]
+                    elif site_var1 == 'Fanduel':
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                     st.session_state.data_export_display = st.session_state.working_seed[0:lineup_num_var]
                 elif 'working_seed' not in st.session_state:
                     if site_var1 == 'Draftkings':
@@ -687,8 +691,12 @@ with tab2:
                             fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
                             raw_baselines = fd_raw
                             column_names = fd_columns
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 10], stack_var2)]
+                    if site_var1 == 'Draftkings':
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], team_var2)]
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 13], stack_var2)]
+                    elif site_var1 == 'Fanduel':
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                        st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                     st.session_state.data_export_display = st.session_state.working_seed[0:lineup_num_var]
                 data_export = st.session_state.working_seed.copy()
                 st.download_button(
@@ -703,8 +711,12 @@ with tab2:
     if st.button("Load Data", key='load_data'):
         if site_var1 == 'Draftkings':
             if 'working_seed' in st.session_state:
-                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 10], stack_var2)]
+                if site_var1 == 'Draftkings':
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 13], stack_var2)]
+                elif site_var1 == 'Fanduel':
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
+                    st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                 st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
             elif 'working_seed' not in st.session_state:
                 if slate_var1 == 'Main Slate':
@@ -714,14 +726,14 @@ with tab2:
                     raw_baselines = dk_raw
                     column_names = dk_columns
             
-                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 10], stack_var2)]
+                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], team_var2)]
+                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 13], stack_var2)]
                 st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
             
         elif site_var1 == 'Fanduel':
             if 'working_seed' in st.session_state:
                 st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 10], stack_var2)]
+                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                 st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
             elif 'working_seed' not in st.session_state:
                 if slate_var1 == 'Main Slate':
@@ -731,7 +743,7 @@ with tab2:
                     raw_baselines = fd_raw
                     column_names = fd_columns
                 st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
-                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 10], stack_var2)]
+                st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                 st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
             
     with st.container():