Update app.py

app.py

@@ -173,24 +173,38 @@ with tab1:
     with col2:
         if st.button("Load Data", key='load_data'):
             if site_var1 == 'Draftkings':
-                
-                st.session_state.working_seed = DK_seed.copy()
-                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:1000], columns=column_names)
-                
-                st.session_state.data_export_freq = pd.DataFrame(calculate_DK_value_frequencies(st.session_state.working_seed), columns=['Player', 'Exposure'])
-                st.session_state.data_export_freq = st.session_state.data_export_freq.sort_values(by='Exposure', ascending=False)
+                if 'working_seed' in st.session_state:
+                    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:1000], columns=column_names)
+                    
+                    st.session_state.data_export_freq = pd.DataFrame(calculate_DK_value_frequencies(st.session_state.working_seed), columns=['Player', 'Exposure'])
+                    st.session_state.data_export_freq = st.session_state.data_export_freq.sort_values(by='Exposure', ascending=False)
+                else:
+                    st.session_state.working_seed = DK_seed.copy()
+                    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:1000], columns=column_names)
+                    
+                    st.session_state.data_export_freq = pd.DataFrame(calculate_DK_value_frequencies(st.session_state.working_seed), columns=['Player', 'Exposure'])
+                    st.session_state.data_export_freq = st.session_state.data_export_freq.sort_values(by='Exposure', ascending=False)
                 
             elif site_var1 == 'Fanduel':
-                
-                st.session_state.working_seed = FD_seed.copy()
-                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:1000], columns=column_names)
-                
-                st.session_state.data_export_freq = pd.DataFrame(calculate_FD_value_frequencies(st.session_state.working_seed), columns=['Player', 'Exposure'])
-                st.session_state.data_export_freq = st.session_state.data_export_freq.sort_values(by='Exposure', ascending=False)
+                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[:, 12], stack_var2)]
+                    st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:1000], columns=column_names)
+                    
+                    st.session_state.data_export_freq = pd.DataFrame(calculate_FD_value_frequencies(st.session_state.working_seed), columns=['Player', 'Exposure'])
+                    st.session_state.data_export_freq = st.session_state.data_export_freq.sort_values(by='Exposure', ascending=False)
+                else:
+                    st.session_state.working_seed = FD_seed.copy()
+                    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:1000], columns=column_names)
+                    
+                    st.session_state.data_export_freq = pd.DataFrame(calculate_FD_value_frequencies(st.session_state.working_seed), columns=['Player', 'Exposure'])
+                    st.session_state.data_export_freq = st.session_state.data_export_freq.sort_values(by='Exposure', ascending=False)
                 
             with st.container():
                 if 'data_export_display' in st.session_state: