feat: version stats

refactor: clear entire session on logout

Author: devxpy

routers/root.py

@@ -173,7 +173,7 @@ def authentication(request: Request, id_token: bytes = Depends(form_id_token)):
 
 @app.get("/logout/")
 async def logout(request: Request):
-    request.session.pop(FIREBASE_SESSION_COOKIE, None)
+    request.session.clear()
     return RedirectResponse(request.query_params.get("next", DEFAULT_LOGOUT_REDIRECT))
 
 

widgets/copilot_stats_plots.py

@@ -92,6 +92,8 @@ def render_copilot_stats_plots(
     plot_total_price(fig, bi, start_date, end_date, trunc_fn, pd_freq)
     plot_retention(fig, bi, start_date, end_date, trunc_fn, pd_freq, dt_index)
 
+    annotate_bot_versions(fig, bi, trunc_fn, pd_freq, dt_index, tz)
+
     gui.plotly_chart(fig, config=defaultPlotlyConfig)
 
 
@@ -432,6 +434,64 @@ def plot_retention(
     add_legend(fig, "Retention", color_idx=9, row=5)
 
 
+def annotate_bot_versions(
+    fig: go.Figure,
+    bi: BotIntegration,
+    trunc_fn: TruncBase,
+    pd_freq: str,
+    dt_index: pd.Index,
+    tz: pytz.timezone,
+    color="#d3d3d3",  # light gray
+):
+    if dt_index.empty:
+        return
+
+    versions = bi.published_run.versions.filter(
+        created_at__range=(dt_index.iloc[0], dt_index.iloc[-1]),
+    ).annotate(dt=trunc_fn("created_at"))
+    if not versions.exists():
+        return
+
+    dates = dt_index.to_list()
+    labels = [""] * len(dates)
+    idx = 0
+
+    for version in reversed(versions):
+        for idx in range(idx, len(dates)):
+            if dates[idx] > version.dt:
+                break
+
+        text = "✏️ "
+        if version.change_notes:
+            text += version.change_notes
+        elif version.title and bi.published_run.title != version.title:
+            text += f"Renamed to: {version.title}"
+        if version.changed_by:
+            text += f" by {version.changed_by.full_name()}"
+        text += f" ({version.dt.strftime('%b %d %Y')})"
+        labels[idx] = text
+
+        # add vertical line to all subplots
+        for row in range(1, fig.layout.grid.rows + 1):
+            fig.add_vline(
+                x=dates[idx],
+                line=dict(color=color, width=1, dash="dash"),
+                yref=f"y{row}",
+            )
+
+    # add a hidden scatter trace to show the hover labels
+    fig.add_trace(
+        go.Scatter(
+            x=dates,
+            y=[0] * len(dates),
+            customdata=labels,
+            hovertemplate="%{customdata}<extra></extra>",
+            showlegend=False,
+            line=dict(width=0, color=color),
+        )
+    )
+
+
 def get_line_marker(*, color_idx: int, gradient: bool = False):
     ret = dict(
         line=dict(color=COLOR_PALETTE[color_idx]),

widgets/plotly_theme.py

@@ -26,6 +26,7 @@ def apply_consistent_styling(fig: "go.Figure") -> "go.Figure":
         autosize=True,
         hovermode="x unified",
         hoverdistance=1000,
+        hoversubplots="axis",
     )
     fig.update_xaxes(
         spikemode="across",