Adding chart language

Author: PJ Hoberman

src/app.py

@@ -19,7 +19,8 @@
 )
 from penn_chime.settings import DEFAULTS 
 from penn_chime.models import sim_sir_df, build_admissions_df, build_census_df
-from penn_chime.charts import additional_projections_chart, admitted_patients_chart, new_admissions_chart
+from penn_chime.charts import (additional_projections_chart, admitted_patients_chart,
+                               new_admissions_chart, chart_descriptions)
 # This is somewhat dangerous:
 # Hide the main menu with "Rerun", "run on Save", "clear cache", and "record a screencast"
 # This should not be hidden in prod, but removed
@@ -51,18 +52,25 @@
 
 st.subheader("New Admissions")
 st.markdown("Projected number of **daily** COVID-19 admissions at Penn hospitals")
+new_admit_chart = new_admissions_chart(alt, admissions_df, parameters=p, as_date=as_date)
 st.altair_chart(
-    new_admissions_chart(alt, admissions_df, parameters=p, as_date=as_date), use_container_width=True
+    new_admit_chart, use_container_width=True
 )
+
+
+st.markdown(chart_descriptions(new_admit_chart))
+
 if st.checkbox("Show Projected Admissions in tabular form"):
     draw_projected_admissions_table(st, admissions_df, as_date=as_date)
 st.subheader("Admitted Patients (Census)")
 st.markdown(
     "Projected **census** of COVID-19 patients, accounting for arrivals and discharges at Penn hospitals"
 )
+admit_chart = admitted_patients_chart(alt=alt, census=census_df, parameters=p, as_date=as_date)
 st.altair_chart(
-    admitted_patients_chart(alt=alt, census=census_df, parameters=p, as_date=as_date), use_container_width=True
+    admit_chart, use_container_width=True
 )
+st.markdown(chart_descriptions(admit_chart, True))
 if st.checkbox("Show Projected Census in tabular form"):
     draw_census_table(st, census_df, as_date=as_date)
 st.markdown(

src/penn_chime/charts.py

@@ -1,4 +1,4 @@
-
+from math import ceil
 from altair import Chart  # type: ignore
 import pandas as pd  # type: ignore
 import numpy as np  # type: ignore
@@ -123,3 +123,26 @@ def additional_projections_chart(
         )
         .interactive()
     )
+
+
+def chart_descriptions(chart, census=False):
+    messages = []
+    cols = ["Hospitalized", "ICU", "Ventilated"]
+    if census:
+        cols = [col + " Census" for col in cols]
+    asterisk = False
+    for col in cols:
+        if chart.data[col].idxmax() + 1 == len(chart.data):
+            asterisk = True
+        messages.append(
+            "{} peaks at {:,} on day {}{}".format(
+                col,
+                ceil(chart.data[col].max()),
+                chart.data[col].idxmax() + 1,
+                "*" if asterisk else "",
+            )
+        )
+
+    if asterisk:
+        messages.append("_* The max is at the upper bound of the data, and therefore may not be the actual max_")
+    return "\n\n".join(messages)

src/penn_chime/presentation.py

@@ -408,10 +408,10 @@ def draw_projected_admissions_table(
             drop_day_column=True,
             date_format=DATE_FORMAT
         )
-
     st.table(admits_table)
     return None
 
+
 def draw_census_table(st, census_df: pd.DataFrame, as_date: bool = False):
     census_table = census_df[np.mod(census_df.index, 7) == 0].copy()
     census_table.index = range(census_table.shape[0])