Upload app_plot_utils.py

app_plot_utils.py

@@ -0,0 +1,41 @@
+import pandas as pd
+import altair as alt
+"""
+app_plot_utils.py
+
+Description: This file contains utility functions for generating interactive plots
+using the Altair library. These functions are designed for visualizing fish count data
+obtained from processed videos and historical records.
+
+Author: Austin Powell
+"""
+
+def plot_count_date(dataframe):
+    """Plots counts vs relative time for uploaded video."""
+    dataframe["seconds"] = dataframe["timestamps"] / 1000
+    dataframe["class"] = "Herring"  # TBD: Hard-coded for now
+    return (
+        alt.Chart(dataframe, title="Processed video detected fish")
+        .mark_line()
+        .encode(x="seconds", y="fish_count", color="class")
+        .interactive()
+    )
+
+def plot_historical_data(dataframe):
+    """Returns altair plot of historical counts to be rendered on main dashboard."""
+    dataframe["Date"] = pd.to_datetime(dataframe["Date"])
+    s = (
+        dataframe.resample(rule="D", on="Date")["Count"].sum().reset_index()
+    )  # Resample on day
+    return (
+        alt.Chart(s, title="Historical Video Counts of Herring")
+        .mark_bar()
+        .transform_window(
+            # The field to average
+            rolling_mean="mean(Count)",
+            # The number of values before and after the current value to include.
+            frame=[-9, 0],
+        )
+        .encode(x="Date", y="Count", tooltip=["Count", "Date"])
+        .interactive()
+    )