Add next gen stats page

src/pages/11_Next_Gen_Stats.py

@@ -0,0 +1,72 @@
+import streamlit as st
+
+from config import DEFAULT_ICON
+from shared_page import common_page_config
+
+from streamlit_filter import get_multiselect_for_df_column
+from queries.nflverse.github_data import get_nextgen_stats, get_current_tables, SEASON
+
+
+hide_columns = [
+    "season",
+    "season_type",
+    "player_gsis_id",
+    "player_first_name",
+    "player_last_name",
+    "player_jersey_number",
+    "player_short_name",
+]
+
+
+def get_page():
+    page_title = f"Next Gen Stats - {SEASON}"
+    st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
+    common_page_config()
+    st.title(page_title)
+
+    stat_category = st.selectbox("Stat Category", ["Passing", "Rushing", "Receiving"])
+    ngs_table_name = f"nextgen_stats_ngs_{stat_category.lower()}"
+    current_tables_list = get_current_tables()
+
+    if ngs_table_name not in current_tables_list:
+        st.write("Data not loaded.")
+        st.write("Check loaded data [here](./Load_Data)")
+        return
+    data = get_nextgen_stats(SEASON, stat_category)
+
+    season_or_week = st.selectbox("Season or Weekly Stats", ["Season", "Week"])
+    if season_or_week == "Season":
+        data = data[data["week"] == 0]
+        data.drop(columns=["week"], inplace=True)
+    else:
+        data = data[data["week"] > 0]
+        week_selection = st.slider(
+            "Filter Week Range:",
+            min_value=data["week"].min(),
+            max_value=data["week"].max(),
+            value=(data["week"].min(), data["week"].max()),
+            step=1,
+        )
+        data = data[data["week"].between(*week_selection)]
+
+    data.drop(columns=hide_columns, inplace=True)
+    positions_selected = get_multiselect_for_df_column(data, "player_position")
+    teams_selected = get_multiselect_for_df_column(data, "team_abbr")
+
+    data = data[(data["player_position"].isin(positions_selected) & data["team_abbr"].isin(teams_selected))]
+
+    with st.container():
+        filtered_data = data
+        st.dataframe(
+            filtered_data,
+            hide_index=True,
+            # height=35 * (len(filtered_data) + 1) + 12,
+            use_container_width=False,
+            column_config={
+                "team_name": st.column_config.TextColumn(label="League Team", help="Name of fantasy League team."),
+            },
+        )
+
+
+if __name__ == "__main__":
+    get_page()

src/queries/nflverse/github_data.py

@@ -63,6 +63,11 @@ def get_pbp_participation(season_int: int) -> pd.DataFrame:
     return df
 
 
+def get_nextgen_stats(season_int: int, stat_category: str) -> pd.DataFrame:
+    df = duckdb.sql(f"SELECT * from nextgen_stats_ngs_{stat_category} where season = {season_int}").df()
+    return df
+
+
 SEASON = "2023"
 
 NFLVERSE_ASSETS = [

src/streamlit_filter.py

@@ -88,3 +88,15 @@ def filter_dataframe(df: pd.DataFrame, force_on: bool = False, force_on_columns:
                     df = df[df[column].astype(str).str.contains(user_text_input)]
 
     return df
+
+
+def get_multiselect_for_df_column(df: pd.DataFrame, column_name: str) -> list:
+    options_list = sorted(df[column_name].unique().tolist())
+    if len(options_list) > 1:
+        selected = (
+            st.multiselect(column_name.title(), options_list, placeholder=f"Select a {column_name} to filter")
+            or options_list
+        )
+    else:
+        selected = options_list
+    return selected