Enhance app.py by standardizing team abbreviations for Draftkings and Fanduel data, ensuring consistency in data representation. Additionally, clear session state on data export to prevent stale data issues. This improves data integrity and user experience in the application.

app.py

@@ -57,11 +57,13 @@ def init_baselines():
     DK_load_display = DK_load_display.drop_duplicates(subset=['Player'], keep='first')
     
     dk_raw = DK_load_display.dropna(subset=['Median'])
+    dk_raw['Team'] = dk_raw['Team'].replace(['TB', 'SJ', 'LA'], ['TBL', 'SJS', 'LAK'])
 
     FD_load_display = load_display[load_display['Site'] == 'Fanduel']
     FD_load_display = FD_load_display.drop_duplicates(subset=['Player'], keep='first')
     
     fd_raw = FD_load_display.dropna(subset=['Median'])
+    fd_raw['Team'] = fd_raw['Team'].replace(['TB', 'SJ', 'LA'], ['TBL', 'SJS', 'LAK'])
 
     return dk_raw, fd_raw
 
@@ -193,6 +195,8 @@ with tab2:
                     file_name='NHL_optimals_export.csv',
                     mime='text/csv',
                 )
+                for key in st.session_state.keys():
+                  del st.session_state[key]
             
     with col2:
         if st.button("Load Data", key='load_data'):