Move all presentation controls to the sidebar and into the Parameters model

Author: Nick Canzoneri

src/app.py

@@ -7,7 +7,6 @@
     # additional_projections_chart,
     display_header,
     display_sidebar,
-    display_n_days_input,
     draw_census_table,
     draw_projected_admissions_table,
     draw_raw_sir_simulation_table,
@@ -39,10 +38,6 @@
     notes = "The total size of the susceptible population will be the entire catchment area for Penn Medicine entities (HUP, PAH, PMC, CCH)"
     show_more_info_about_this_tool(st=st, parameters=p, inputs=DEFAULTS, notes=notes)
 
-# PRESENTATION
-# Two more combination variable initialization / input element creation
-as_date = st.checkbox(label="Present result as dates instead of days", value=False)
-display_n_days_input(st, p, DEFAULTS)
 
 # begin format data
 admissions_df = build_admissions_df(p=p)  # p.n_days, *p.dispositions)
@@ -52,29 +47,29 @@
 st.subheader("New Admissions")
 st.markdown("Projected number of **daily** COVID-19 admissions at Penn hospitals")
 st.altair_chart(
-    new_admissions_chart(alt, admissions_df, parameters=p, as_date=as_date),
+    new_admissions_chart(alt, admissions_df, parameters=p),
     use_container_width=True,
 )
 if st.checkbox("Show Projected Admissions in tabular form"):
-    draw_projected_admissions_table(st, admissions_df, as_date=as_date)
+    draw_projected_admissions_table(st, admissions_df)
 st.subheader("Admitted Patients (Census)")
 st.markdown(
     "Projected **census** of COVID-19 patients, accounting for arrivals and discharges at Penn hospitals"
 )
 st.altair_chart(
-    admitted_patients_chart(alt=alt, census=census_df, parameters=p, as_date=as_date),
+    admitted_patients_chart(alt=alt, census=census_df, parameters=p),
     use_container_width=True,
 )
 if st.checkbox("Show Projected Census in tabular form"):
-    draw_census_table(st, census_df, as_date=as_date)
+    draw_census_table(st, census_df, parameters=p)
 st.markdown(
     """**Click the checkbox below to view additional data generated by this simulation**"""
 )
 if st.checkbox("Show Additional Projections"):
     show_additional_projections(
-        st, alt, additional_projections_chart, parameters=p, as_date=as_date
+        st, alt, additional_projections_chart, parameters=p
     )
     if st.checkbox("Show Raw SIR Simulation Data"):
-        draw_raw_sir_simulation_table(st, parameters=p, as_date=as_date)
+        draw_raw_sir_simulation_table(st, parameters=p)
 write_definitions(st)
 write_footer(st)

src/penn_chime/charts.py

@@ -7,11 +7,12 @@
 
 
 def new_admissions_chart(
-    alt, projection_admits: pd.DataFrame, parameters: Parameters, as_date: bool = False,
+    alt, projection_admits: pd.DataFrame, parameters: Parameters
 ) -> Chart:
     """docstring"""
     plot_projection_days = parameters.n_days - 10
     max_y_axis = parameters.max_y_axis
+    as_date = parameters.as_date
 
     y_scale = alt.Scale()
 
@@ -45,12 +46,13 @@ def new_admissions_chart(
 
 
 def admitted_patients_chart(
-    alt, census: pd.DataFrame, parameters: Parameters, as_date: bool = False
+    alt, census: pd.DataFrame, parameters: Parameters
 ) -> Chart:
     """docstring"""
 
     plot_projection_days = parameters.n_days - 10
     max_y_axis = parameters.max_y_axis
+    as_date = parameters.as_date
     if as_date:
         census = add_date_column(census)
         x_kwargs = {"shorthand": "date:T", "title": "Date"}
@@ -84,10 +86,16 @@ def admitted_patients_chart(
 
 
 def additional_projections_chart(
-    alt, i: np.ndarray, r: np.ndarray, as_date: bool = False, max_y_axis: int = None
+    alt, parameters: Parameters
 ) -> Chart:
+    i = parameters.infected_v
+    r = parameters.recovered_v
     dat = pd.DataFrame({"Infected": i, "Recovered": r})
     dat["day"] = dat.index
+
+    as_date = parameters.as_date
+    max_y_axis = parameters.max_y_axis
+
     if as_date:
         dat = add_date_column(dat)
         x_kwargs = {"shorthand": "date:T", "title": "Date"}

src/penn_chime/parameters.py

@@ -23,7 +23,8 @@ def __init__(
         icu: RateLos,
         ventilated: RateLos,
         max_y_axis: int = None,
-        n_days: int = None
+        n_days: int = None,
+        as_date: bool = False
     ):
         self.current_hospitalized = current_hospitalized
         self.doubling_time = doubling_time
@@ -38,6 +39,7 @@ def __init__(
         self.ventilated = ventilated
 
         self.max_y_axis = max_y_axis
+        self.as_date = as_date
 
         self.rates = tuple(each.rate for each in (hospitalized, icu, ventilated))
         self.lengths_of_stay = tuple(

src/penn_chime/presentation.py

@@ -94,6 +94,14 @@ def display_sidebar(st, d: Constants) -> Parameters:
     if d.known_infected < 1:
         raise ValueError("Known cases must be larger than one to enable predictions.")
 
+    n_days = st.sidebar.number_input(
+        "Number of days to project",
+        min_value=30,
+        value=d.n_days,
+        step=10,
+        format="%i",
+    )
+
     current_hospitalized = st.sidebar.number_input(
         "Currently Hospitalized COVID-19 Patients",
         min_value=0,
@@ -205,6 +213,8 @@ def display_sidebar(st, d: Constants) -> Parameters:
         format="%i",
     )
 
+    as_date = st.sidebar.checkbox(label="Present result as dates instead of days", value=False)
+
     max_y_axis_set = st.sidebar.checkbox("Set the Y-axis on graphs to a static value")
     max_y_axis = None
     if max_y_axis_set:
@@ -213,6 +223,7 @@ def display_sidebar(st, d: Constants) -> Parameters:
         )
 
     return Parameters(
+        n_days=n_days,
         current_hospitalized=current_hospitalized,
         doubling_time=doubling_time,
         known_infected=known_infected,
@@ -223,17 +234,7 @@ def display_sidebar(st, d: Constants) -> Parameters:
         icu=RateLos(icu_rate, icu_los),
         ventilated=RateLos(ventilated_rate, ventilated_los),
         max_y_axis=max_y_axis,
-    )
-
-
-def display_n_days_input(st, p: Parameters, d: Constants):
-    """Display n_days_input."""
-    p.n_days = st.number_input(
-        "Number of days to project",
-        min_value=30,
-        value=d.n_days,
-        step=10,
-        format="%i",
+        as_date=as_date,
     )
 
 
@@ -368,7 +369,7 @@ def write_footer(st):
 
 
 def show_additional_projections(
-    st, alt, charting_func, parameters, as_date: bool = False,
+    st, alt, charting_func, parameters
 ):
     st.subheader(
         "The number of infected and recovered individuals in the hospital catchment region at any given moment"
@@ -377,10 +378,7 @@ def show_additional_projections(
     st.altair_chart(
         charting_func(
             alt,
-            parameters.infected_v,
-            parameters.recovered_v,
-            as_date=as_date,
-            max_y_axis=parameters.max_y_axis,
+            parameters=parameters
         ),
         use_container_width=True,
     )