From 97e897c2409ff96e946d74b89795c547126ddae4 Mon Sep 17 00:00:00 2001 From: Dylan De Faoite Date: Sun, 22 Mar 2026 14:34:05 +0000 Subject: [PATCH] fix(analysis): broken entity handling in cultural endpoint --- server/analysis/cultural.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/server/analysis/cultural.py b/server/analysis/cultural.py index fc4a93a..6aef58a 100644 --- a/server/analysis/cultural.py +++ b/server/analysis/cultural.py @@ -84,14 +84,14 @@ class CulturalAnalysis: } def get_avg_emotions_per_entity(self, df: pd.DataFrame, top_n: int = 25, min_posts: int = 10) -> dict[str, Any]: - if "entities" not in df.columns: + if "ner_entities" not in df.columns: return {"entity_emotion_avg": {}} emotion_cols = [c for c in df.columns if c.startswith("emotion_")] - entity_df = df[["entities"] + emotion_cols].explode("entities") + entity_df = df[["ner_entities"] + emotion_cols].explode("ner_entities") - entity_df["entity_text"] = entity_df["entities"].apply( + entity_df["entity_text"] = entity_df["ner_entities"].apply( lambda e: e.get("text").strip() if isinstance(e, dict) and isinstance(e.get("text"), str) and len(e.get("text")) >= 3 else None