Adding chart language

Author: PJ Hoberman

src/app.py

@@ -19,12 +19,10 @@
 )
 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,20 +49,24 @@
 
 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,3 +1,5 @@
+
+from math import ceil
 from altair import Chart  # type: ignore
 import pandas as pd  # type: ignore
 import numpy as np  # type: ignore
@@ -112,3 +114,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

@@ -404,7 +404,6 @@ def draw_projected_admissions_table(
         admits_table = add_date_column(
             admits_table, drop_day_column=True, date_format=DATE_FORMAT
         )
-
     st.table(admits_table)
     return None