diff --git a/src/webui/data/analysis.py b/src/webui/data/analysis.py
index f78f672..155ece0 100644
--- a/src/webui/data/analysis.py
+++ b/src/webui/data/analysis.py
@@ -13,6 +13,10 @@ from sklearn.preprocessing import normalize as sk_normalize
 
 from ietf_analyzer.config import Config
 from ietf_analyzer.db import Database
+
+SAFETY_CATEGORIES = {"AI safety/alignment", "Agent identity/auth", "Policy/governance"}
+CAPABILITY_CATEGORIES = {"A2A protocols", "Agent discovery/reg", "Autonomous netops",
+                         "Data formats/interop", "Human-agent interaction", "Model serving/inference"}
 from webui.data._shared import _cached, _extract_month
 from webui.data.drafts import get_draft_detail