Update app.py

app.py

@@ -225,9 +225,13 @@ with tab2:
         sim_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'), key='sim_slate_var1')
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='sim_site_var1')
         if sim_site_var1 == 'Draftkings':
+            for key in st.session_state.keys():
+                  del st.session_state[key]
             raw_baselines = dk_raw
             column_names = dk_columns
         elif sim_site_var1 == 'Fanduel':
+            for key in st.session_state.keys():
+                  del st.session_state[key]
             raw_baselines = fd_raw
             column_names = fd_columns
             
@@ -341,13 +345,24 @@ with tab2:
                 st.session_state.stack_freq['Exposure'] = st.session_state.stack_freq['Freq']/(1000)
 
                 with st.container():
-                    st.dataframe(st.session_state.Sim_Winner_Frame)
+                    try:
+                        st.dataframe(st.session_state.Sim_Winner_Frame)
+                    except:
+                        st.write("resources low, waiting 5 seconds and trying again")
+                        time.sleep(5)
+                        st.dataframe(st.session_state.Sim_Winner_Frame)
                 
                 with st.container():
                     tab1, tab2, tab3, tab4 = st.tabs(['Overall Exposures', 'SP Exposures', 'Team Exposures', 'Stack Size Exposures'])
                     with tab1:
                         if 'player_freq' in st.session_state:
-                            st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
+                            
+                            try:
+                                st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
+                            except:
+                                st.write("resources low, waiting 5 seconds and trying again")
+                                time.sleep(5)
+                                st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
                             st.download_button(
                                 label="Export Exposures",
                                 data=st.session_state.player_freq.to_csv().encode('utf-8'),
@@ -357,31 +372,49 @@ with tab2:
                             )
                     with tab2:
                         if 'sp_freq' in st.session_state:
-                            st.dataframe(st.session_state.sp_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
+                            
+                            try:
+                                st.dataframe(st.session_state.sp_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
+                            except:
+                                st.write("resources low, waiting 5 seconds and trying again")
+                                time.sleep(5)
+                                st.dataframe(st.session_state.sp_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
                             st.download_button(
                                 label="Export Exposures",
                                 data=st.session_state.sp_freq.to_csv().encode('utf-8'),
-                                file_name='player_freq_export.csv',
+                                file_name='sp_freq.csv',
                                 mime='text/csv',
                                 key='sp'
                             )
                     with tab3:
                         if 'team_freq' in st.session_state:
-                            st.dataframe(st.session_state.team_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
+                            
+                            try:
+                                st.dataframe(st.session_state.team_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
+                            except:
+                                st.write("resources low, waiting 5 seconds and trying again")
+                                time.sleep(5)
+                                st.dataframe(st.session_state.team_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
                             st.download_button(
                                 label="Export Exposures",
                                 data=st.session_state.team_freq.to_csv().encode('utf-8'),
-                                file_name='player_freq_export.csv',
+                                file_name='team_freq.csv',
                                 mime='text/csv',
                                 key='team'
                             )
                     with tab4:
                         if 'stack_freq' in st.session_state:
-                            st.dataframe(st.session_state.stack_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
+                            
+                            try:
+                                st.dataframe(st.session_state.stack_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
+                            except:
+                                st.write("resources low, waiting 5 seconds and trying again")
+                                time.sleep(5)
+                                st.dataframe(st.session_state.stack_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
                             st.download_button(
                                 label="Export Exposures",
                                 data=st.session_state.stack_freq.to_csv().encode('utf-8'),
-                                file_name='player_freq_export.csv',
+                                file_name='stack_freq.csv',
                                 mime='text/csv',
                                 key='stack'
                             )
@@ -486,6 +519,7 @@ with tab2:
                     tab1, tab2, tab3, tab4 = st.tabs(['Overall Exposures', 'SP Exposures', 'Team Exposures', 'Stack Size Exposures'])
                     with tab1:
                         if 'player_freq' in st.session_state:
+                            
                             try:
                                 st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
                             except:
@@ -501,6 +535,7 @@ with tab2:
                             )
                     with tab2:
                         if 'sp_freq' in st.session_state:
+                            
                             try:
                                 st.dataframe(st.session_state.sp_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)
                             except:
@@ -516,6 +551,7 @@ with tab2:
                             )
                     with tab3:
                         if 'team_freq' in st.session_state:
+                            
                             try:
                                 st.dataframe(st.session_state.team_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
                             except:
@@ -531,6 +567,7 @@ with tab2:
                             )
                     with tab4:
                         if 'stack_freq' in st.session_state:
+                            
                             try:
                                 st.dataframe(st.session_state.stack_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
                             except: