Update app.py

app.py

@@ -126,12 +126,14 @@ with tab1:
     with col2:
         if st.button("Load Seed Frame", key='seed_frame_load'):
             if site_var1 == 'Draftkings':
+                
                 DK_seed = init_DK_seed_frame()
                 DK_seed_parse = DK_seed[DK_seed['Team'].isin(team_var2)]
                 DK_seed_parse = DK_seed_parse[DK_seed_parse['Team_count'].isin(stack_var2)]
-                st.session_state.data_export_display = DK_seed_parse.head(1000)
-                st.session_state.data_export = DK_seed_parse
-                st.session_state.data_export_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.data_export.iloc[:,0:9].values, return_counts=True)),
+                data_export_display = DK_seed_parse.head(1000)
+                st.session_state.data_export_display = data_export_display.copy()
+                st.session_state.data_export = data_export_display.copy()
+                st.session_state.data_export_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.data_export_display.iloc[:,0:9].values, return_counts=True)),
                                             columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
                 st.session_state.data_export_freq['Freq'] = st.session_state.data_export_freq['Freq'].astype(int)
                 st.session_state.data_export_freq['Exposure'] = st.session_state.data_export_freq['Freq']/(len(DK_seed_parse['Team']))
@@ -143,18 +145,24 @@ with tab1:
                             file_name='MLB_optimals_export.csv',
                             mime='text/csv',
                         )
-                st.dataframe(st.session_state.data_export_display.style.format(precision=2), height=500, use_container_width=True)
-                st.dataframe(st.session_state.data_export_freq.style.format(percentages_format, precision=2), height=500, use_container_width=True)
+                with st.container():
+                    if 'data_export_display' in st.session.state:
+                        st.dataframe(st.session_state.data_export_display.style.format(precision=2), height=500, use_container_width=True)
+                with st.container():
+                    if 'data_export_freq' in st.session.state:
+                        st.dataframe(st.session_state.data_export_freq.style.format(percentages_format, precision=2), height=500, use_container_width=True)
+            
             elif site_var1 == 'Fanduel':
                 FD_seed = init_DK_seed_frame()
                 FD_seed_parse = FD_seed[FD_seed['Team'].isin(team_var2)]
                 FD_seed_parse = FD_seed_parse[FD_seed_parse['Team_count'].isin(stack_var2)]
-                st.session_state.data_export_display = FD_seed_parse.head(1000)
-                st.session_state.data_export = FD_seed_parse
-                st.session_state.data_export_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.data_export.iloc[:,0:8].values, return_counts=True)),
+                data_export_display = DK_seed_parse.head(1000)
+                st.session_state.data_export_display = data_export_display.copy()
+                st.session_state.data_export = data_export_display.copy()
+                st.session_state.data_export_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.data_export_display.iloc[:,0:8].values, return_counts=True)),
                                             columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
                 st.session_state.data_export_freq['Freq'] = st.session_state.data_export_freq['Freq'].astype(int)
-                st.session_state.data_export_freq['Exposure'] = st.session_state.data_export_freq['Freq']/(len(FD_seed_parse['Team']))
+                st.session_state.data_export_freq['Exposure'] = st.session_state.data_export_freq['Freq']/(len(DK_seed_parse['Team']))
                 
                 if 'data_export' in st.session_state:
                         st.download_button(
@@ -163,8 +171,12 @@ with tab1:
                             file_name='MLB_optimals_export.csv',
                             mime='text/csv',
                         )
-                st.dataframe(st.session_state.data_export_display.style.format(precision=2), height=500, use_container_width=True)
-                st.dataframe(st.session_state.data_export_freq.style.format(percentages_format, precision=2), height=500, use_container_width=True)
+                with st.container():
+                    if 'data_export_display' in st.session.state:
+                        st.dataframe(st.session_state.data_export_display.style.format(precision=2), height=500, use_container_width=True)
+                with st.container():
+                    if 'data_export_freq' in st.session.state:
+                        st.dataframe(st.session_state.data_export_freq.style.format(percentages_format, precision=2), height=500, use_container_width=True)
             
 with tab2:
     col1, col2 = st.columns([1, 7])