Get league simulation working

src/pages/50_League_Simulation.py

@@ -1,9 +1,49 @@
+import pandas as pd
 import streamlit as st
 
-from config import DEFAULT_ICON
+from config import DEFAULT_ICON, SEASON
+from login_component import is_token_in_session
 from shared_page import common_page_config
+from simulate import create_simulate_summary, run_simulations
 
-from login_component import is_token_in_session
+
+@st.cache_resource(ttl=60 * 60 * 24)
+def get_all_league_settings_with_cache(season: int):
+    return st.session_state.yahoo_client.get_all_logged_in_user_league_settings(season=season)
+
+
+@st.cache_resource(ttl=60 * 60 * 24)
+def get_schedule_with_cache(league_key):
+    return st.session_state.yahoo_client.full_schedule_dataframe(league_key)
+
+
+def clear_all():
+    for session_key in st.session_state.keys():
+        if session_key[:13] == "match-filter-":
+            st.session_state[session_key] = None
+    return
+
+
+def show_week_matches(schedule: pd.DataFrame, week: int | str):
+    df_week = schedule[schedule.week == week]
+    columns = st.columns(df_week.match_index.nunique())
+    match_filter_map = {}
+    for match_col, (match_index, df) in zip(columns, df_week.groupby("match_index")):
+        with match_col:
+            match_winner = st.radio(
+                "Set winner",
+                key=f"match-filter-{match_index}",
+                options=df.team_name.tolist(),
+                index=None,
+            )
+            if match_winner:
+                match_filter_map[match_index] = match_winner
+    return match_filter_map
+
+
+@st.cache_resource(ttl=60 * 60 * 24)
+def get_sim_with_cache(schedule: pd.DataFrame, complete_weeks: int, n_sims: int, n_playoff: int):
+    return run_simulations(schedule, complete_weeks, n_sims, n_playoff)
 
 
 def get_page():
@@ -19,7 +59,37 @@ def get_page():
         )
 
     else:
-        st.write("Logged in.  Feature to go here")
+        selected_season = st.selectbox("Select Season", list(range(SEASON, 2012, -1)))
+        user_leagues = get_all_league_settings_with_cache(season=selected_season)
+        selected_league = st.selectbox("Select league", user_leagues, format_func=lambda x: x.name)
+        st.header(f"{selected_league.name} - {selected_league.season}")
+
+        df_schedule = get_schedule_with_cache(selected_league.league_key)
+        # st.dataframe(df_schedule)
+        st.write(f"Current Week: {selected_league.current_week}")
+        sim_result = pd.DataFrame()
+        n_sims = 10000
+
+        st.header("Match Scenarios")
+        st.write("Select match winners to filter simulation scenarios")
+        all_match_filters = {}
+        st.button("Reset", on_click=clear_all)
+        for week in range(selected_league.current_week, selected_league.playoff_start_week):
+            with st.expander(f"Week {week}"):
+                all_match_filters[week] = show_week_matches(df_schedule, week)
+
+        sim_result = get_sim_with_cache(
+            df_schedule, selected_league.current_week - 1, n_sims, selected_league.num_playoff_teams
+        )
+
+        if len(sim_result):
+            filtered_sims = sim_result.copy()
+            for week, filtered_match_dict in all_match_filters.items():
+                for match_index, match_winner in filtered_match_dict.items():
+                    filtered_sims = filtered_sims[filtered_sims[int(week)][match_index] == match_winner]
+
+            st.write(f"Number of Scenarios in filter: {len(filtered_sims)=} / {n_sims}")
+            st.dataframe(create_simulate_summary(filtered_sims))
 
 
 if __name__ == "__main__":