Update app.py to consistently use session_state for display data management, enhancing filtering logic for site and slate types, and ensuring proper data presentation in 'Simple' and 'Advanced' views.

app.py

@@ -173,31 +173,45 @@ 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':
-        st.session_slate['display_data'] = roo_data.copy()
+        st.session_state['display_data'] = roo_data.copy()
         if site_var2 == 'Draftkings':
-            st.session_slate['display_data']['Site'] = 'Draftkings'
+            st.session_state['display_data']['Site'] = 'Draftkings'
+
+            if pos_var2 == 'All':
+                pass
+            elif pos_var2 == 'Pitchers':
+                st.session_state['display_data'] = st.session_state['display_data'][st.session_state['display_data']['Position'] == 'SP']
+            elif pos_var2 == 'Hitters':
+                st.session_state['display_data'] = st.session_state['display_data'][st.session_state['display_data']['Position'] != 'SP']
         elif site_var2 == 'Fanduel':
-            st.session_slate['display_data']['Site'] = 'Fanduel'
+            st.session_state['display_data']['Site'] = 'Fanduel'
+
+            if pos_var2 == 'All':
+                pass
+            elif pos_var2 == 'Pitchers':
+                st.session_state['display_data'] = st.session_state['display_data'][st.session_state['display_data']['Position'] == 'P']
+            elif pos_var2 == 'Hitters':
+                st.session_state['display_data'] = st.session_state['display_data'][st.session_state['display_data']['Position'] != 'P']
         
         if slate_var2 == 'Main':
-            st.session_slate['display_data'] = st.session_slate['display_data'][st.session_slate['display_data']['Slate'] == 'Main']
+            st.session_state['display_data'] = st.session_state['display_data'][st.session_state['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']
+            st.session_state['display_data'] = st.session_state['display_data'][st.session_state['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']
+            st.session_state['display_data'] = st.session_state['display_data'][st.session_state['display_data']['Slate'] == 'Auxiliary']
             
     elif slate_type_var2 == 'Showdown':
-        st.session_slate['display_data'] = sd_roo_data.copy()
+        st.session_state['display_data'] = sd_roo_data.copy()
         if site_var2 == 'Draftkings':
-            st.session_slate['display_data']['Site'] = 'Draftkings'
+            st.session_state['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'])
+            st.session_state['display_data']['Site'] = 'Fanduel'
+    st.session_state['display_data'] = st.session_state['display_data'].drop(columns=['site', 'slate', 'version', 'timestamp'])
     if view_var == "Simple":
-        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)
+        st.session_state['display_data'] = st.session_state['display_data'][['Player', 'Position', 'Salary', 'Median', 'Ceiling', 'Own']]
+        st.dataframe(st.session_state['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(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)
+        st.dataframe(st.session_state['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"):