Added table & plot translation capabilities

Author: ckoerber

src/chime_dash/app/services/callbacks.py

@@ -38,10 +38,14 @@ def handle_model_change(i, sidebar_data):
         if sidebar_data:
             pars = parameters_deserializer(sidebar_data["parameters"])
             model = SimSirModel(pars)
+            vis = i.components.get("visualizations", None) if i else None
+            vis_content = vis.content if vis else None
+
             viz_kwargs = dict(
                 labels=pars.labels,
                 table_mod=7,
                 max_y_axis=pars.max_y_axis,
+                content=vis_content
             )
         result.extend(i.components["intro"].build(model, pars))
         for df_key in ["admits_df", "census_df", "sim_sir_w_date_df"]:

src/chime_dash/app/services/plotting.py

@@ -7,9 +7,17 @@
 from pandas import DataFrame
 
 
-def plot_dataframe(dataframe: DataFrame, max_y_axis: int = None,) -> Dict[str, Any]:
-    """
+def plot_dataframe(
+    dataframe: DataFrame,
+    max_y_axis: int = None,
+) -> Dict[str, Any]:
+    """Returns dictionary used for plotly graphs
+
+    Arguments:
+        dataframe: The dataframe to plot. Plots all columns as y, index is x.
+        max_y_axis: Maximal value on y-axis.
     """
+
     if max_y_axis is None:
         yaxis = {}
     else:

src/chime_dash/app/templates/en/visualizations.yml

@@ -16,5 +16,20 @@ hosp: Hospitalized
 icu: ICU
 vent: Ventalized
 
+# Admits df column translations
+admits_hospitalized: Hospitalized
+admits_icu: ICU
+admits_ventilated: Ventilated
+
+# Admits df column translations
+census_hospitalized: Hospitalized
+census_icu: ICU
+census_ventilated: Ventilated
+
+# SIR df column translations
+susceptible: Susceptible
+infected: Infected
+recovered: Recovered
+
 # Date Localization
 date-format: "%m%d%Y"

src/chime_dash/app/utils/__init__.py

@@ -116,18 +116,49 @@ def get_n_switch_values(input_value, elements_to_update) -> List[bool]:
 
 def prepare_visualization_group(df: DataFrame = None, **kwargs) -> List[Any]:
     """Creates plot, table and download link for data frame.
+
+    Arguments:
+        df: The Dataframe to plot
+        content: Dict[str, str]
+            Mapping for translating columns and index.
+        max_y_axis:  int
+            Maximal value on y-axis
+        labels: List[str]
+            Columns to display
+        table_mod: int
+            Displays only each `table_mod` row in table
+
     """
     result = [{}, None, None]
     if df is not None and isinstance(df, DataFrame):
+
+        date_column = "date"
+        day_column = "day"
+
+        # Translate column and index if specified
+        content = kwargs.get("content", None)
+        if content:
+            columns = {col: content[col] for col in df.columns if col in content}
+            index = (
+                {df.index.name: content[df.index.name]}
+                if df.index.name and df.index.name in content
+                else None
+            )
+            df = df.rename(columns=columns, index=index)
+            date_column = content.get(date_column, date_column)
+            day_column = content.get(day_column, day_column)
+
         plot_data = plot_dataframe(
-            df.dropna().set_index("date").drop(columns=["day"]),
+            df.dropna().set_index(date_column).drop(columns=[day_column]),
             max_y_axis=kwargs.get("max_y_axis", None),
         )
 
         table = (
             df_to_html_table(
                 build_table(
-                    df=df,
+                    df=df.rename( # translate back for backwards compability
+                        columns={day_column: "day", date_column: "date"}
+                    ),
                     labels=kwargs.get("labels", df.columns),
                     modulo=kwargs.get("table_mod", 7),
                 ),
@@ -153,4 +184,5 @@ def get_instance(*args, **kwargs):
         if class_ not in instances:
             instances[class_] = class_(*args, **kwargs)
         return instances[class_]
+
     return get_instance