Add fantasy positions filter and set explicit filters on snap counts

src/pages/7_Snap_Counts.py

@@ -6,14 +6,17 @@ from shared_page import common_page_config
 
 from queries.footballguys.constants import YEAR
 from queries.nflverse.github_data import get_snap_counts
-from streamlit_filter import filter_dataframe
 
 
 @st.cache_data(ttl=60 * 60 * 24)
 def load_data():
     data = get_snap_counts(YEAR)
+    data = data[data.fantasy_position]
+    teams_list = sorted(data.team.unique())
+    position_list = data.position.unique()
+    weeks_list = sorted(data.week.unique())
     data_load_time_str = datetime.datetime.utcnow().strftime("%m/%d/%Y %I:%M %p")
-    return data, data_load_time_str
+    return data, teams_list, position_list, weeks_list, data_load_time_str
 
 
 def get_page():
@@ -23,11 +26,18 @@ def get_page():
     st.title(page_title)
     if st.button("Refresh Data"):
         st.cache_data.clear()
-    data, data_load_time_str = load_data()
+    data, teams_list, position_list, weeks_list, data_load_time_str = load_data()
     st.write(f"Data loaded as of: {data_load_time_str} UTC")
+    teams_selected = st.multiselect("Team:", teams_list, placeholder="Select a team to filter") or teams_list
+    positions_selected = (
+        st.multiselect("Position:", position_list, placeholder="Select a position to filter") or position_list
+    )
+    weeks_selected = st.multiselect("Week:", weeks_list, placeholder="Select a week to filter") or weeks_list
 
     with st.container():
-        filtered_data = filter_dataframe(data, force_on=True, force_on_columns=["team", "position", "week", "opponent"])
+        filtered_data = data[
+            (data.team.isin(teams_selected) & data.position.isin(positions_selected) & data.week.isin(weeks_selected))
+        ]
         st.dataframe(
             filtered_data,
             hide_index=True,

src/queries/nflverse/github_data.py

@@ -7,8 +7,20 @@ def get_parquet_github(season_int: int, parquet_prefix: str):
     return df
 
 
+FANTASY_POSITIONS = [
+    "QB",
+    "RB",
+    "WR",
+    "TE",
+    "FB",
+    "K",
+]
+
+
 def get_snap_counts(season_int: int) -> pd.DataFrame:
-    return get_parquet_github(season_int, "snap_counts/snap_counts")
+    df = get_parquet_github(season_int, "snap_counts/snap_counts")
+    df["fantasy_position"] = df["position"].isin(FANTASY_POSITIONS)
+    return df
 
 
 def get_play_by_play(season_int: int) -> pd.DataFrame: