Refactor data handling in app.py to store display data in session state, enabling conditional filtering based on site and slate type selections, and improving data presentation for 'Simple' and 'Advanced' views.

app.py

@@ -173,15 +173,31 @@ with tab2:
             pos_var2 = st.selectbox("Which position group would you like to view?", ('All', 'Pitchers', 'Hitters'), key='pos_var2')
     
     if slate_type_var2 == 'Regular':
-        display_data = roo_data
+        st.session_slate['display_data'] = roo_data.copy()
+        if site_var2 == 'Draftkings':
+            st.session_slate['display_data']['Site'] = 'Draftkings'
+        elif site_var2 == 'Fanduel':
+            st.session_slate['display_data']['Site'] = 'Fanduel'
+        
+        if slate_var2 == 'Main':
+            st.session_slate['display_data'] = st.session_slate['display_data'][st.session_slate['display_data']['Slate'] == 'Main']
+        elif slate_var2 == 'Secondary':
+            st.session_slate['display_data'] = st.session_slate['display_data'][st.session_slate['display_data']['Slate'] == 'Secondary']
+        elif slate_var2 == 'Auxiliary':
+            st.session_slate['display_data'] = st.session_slate['display_data'][st.session_slate['display_data']['Slate'] == 'Auxiliary']
+            
     elif slate_type_var2 == 'Showdown':
-        display_data = sd_roo_data
-    display_data = display_data.drop(columns=['site', 'slate', 'version', 'timestamp'])
+        st.session_slate['display_data'] = sd_roo_data.copy()
+        if site_var2 == 'Draftkings':
+            st.session_slate['display_data']['Site'] = 'Draftkings'
+        elif site_var2 == 'Fanduel':
+            st.session_slate['display_data']['Site'] = 'Fanduel'
+    st.session_slate['display_data'] = st.session_slate['display_data'].drop(columns=['site', 'slate', 'version', 'timestamp'])
     if view_var == "Simple":
-        display_data = display_data[['Player', 'Position', 'Salary', 'Median', 'Ceiling', 'Own']]
-        st.dataframe(display_data.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(player_roo_format, precision=2), height=750, use_container_width = True, hide_index=True)
+        st.session_slate['display_data'] = st.session_slate['display_data'][['Player', 'Position', 'Salary', 'Median', 'Ceiling', 'Own']]
+        st.dataframe(st.session_slate['display_data'].style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(player_roo_format, precision=2), height=750, use_container_width = True, hide_index=True)
     elif view_var == "Advanced":
-        st.dataframe(display_data.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(player_roo_format, precision=2), height=750, use_container_width = True, hide_index=True)
+        st.dataframe(st.session_slate['display_data'].style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(player_roo_format, precision=2), height=750, use_container_width = True, hide_index=True)
 
 with tab3:
     with st.expander("Info and Filters"):