Add team filtering functionality to Player Stat and Prop Trend tables

app.py

@@ -118,10 +118,16 @@ prop_display, prop_trends, pick_frame, team_dict = pull_baselines()
 tab1, tab2, tab3 = st.tabs(["Player Stat Table", 'Prop Trend Table', 'Stat Specific Simulations'])
 
 with tab1:
-    if st.button("Reset Data", key='reset1'):
-              st.cache_data.clear()
-              prop_display, prop_trends, pick_frame, team_dict = pull_baselines()
-    prop_frame = prop_display
+    with st.expander("Info and Filters"):
+        if st.button("Reset Data", key='reset1'):
+            st.cache_data.clear()
+            prop_display, prop_trends, pick_frame, team_dict = pull_baselines()
+        team_var = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='team_var')
+        if team_var == 'Specific Teams':
+            team_var = st.multiselect('Which teams would you like to include in the tables?', options = prop_display['Team'].unique(), key='team_var')
+        elif team_var == 'All':
+            team_var = prop_display['Team'].unique()
+    prop_frame = prop_display[prop_display['Team'].isin(team_var)]
     st.dataframe(prop_frame.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
     
     st.download_button(
@@ -133,10 +139,11 @@ with tab1:
     )
 
 with tab2:
-    if st.button("Reset Data", key='reset3'):
-              st.cache_data.clear()
-              prop_display, prop_trends, pick_frame, team_dict = pull_baselines()
-    with st.expander("Filter Options"):
+    with st.expander("Info and Filters"):
+        if st.button("Reset Data", key='reset3'):
+                st.cache_data.clear()
+                prop_display, prop_trends, pick_frame, team_dict = pull_baselines()
+    
         split_var5 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var5')
         if split_var5 == 'Specific Teams':
             team_var5 = st.multiselect('Which teams would you like to include in the tables?', options = prop_trends['Team'].unique(), key='team_var5')