getting started on filters

Author: keegangeorge

assets/javascripts/discourse/components/admin-report-sentiment-analysis.gjs

@@ -12,10 +12,6 @@ import DoughnutChart from "./doughnut-chart";
 export default class AdminReportSentimentAnalysis extends Component {
   @tracked selectedChart = null;
 
-  get labels() {
-    return ["Positive", "Neutral", "Negative"];
-  }
-
   get colors() {
     return ["#2ecc71", "#95a5a6", "#e74c3c"];
   }
@@ -71,6 +67,14 @@ export default class AdminReportSentimentAnalysis extends Component {
     }
   }
 
+  doughnutTitle(data) {
+    if (data.posts?.length > 0) {
+      return `${data.category_name} (${data.posts.length})`;
+    } else {
+      return data.category_name;
+    }
+  }
+
   <template>
     {{! TODO add more details about posts on click + tag data }}
     <div class="admin-report-sentiment-analysis">
@@ -87,10 +91,10 @@ export default class AdminReportSentimentAnalysis extends Component {
           }}
         >
           <DoughnutChart
-            @labels={{this.labels}}
+            @labels={{@model.labels}}
             @colors={{this.colors}}
             @data={{data.scores}}
-            @doughnutTitle={{data.category_name}}
+            @doughnutTitle={{this.doughnutTitle data}}
           />
         </div>
       {{/each}}

config/locales/client.en.yml

@@ -15,6 +15,11 @@ en:
         emotion:
           title: "Emotion"
           description: "The table lists a count of posts classified with a determined emotion. Classified with the model 'SamLowe/roberta-base-go_emotions'."
+        reports:
+          filters:
+            filter_by:
+              label: "Filter by"
+
   js:
     discourse_automation:
       scriptables:

config/locales/server.en.yml

@@ -413,6 +413,10 @@ en:
           anger: "Anger 😡"
           joy: "Joy 😀"
           disgust: "Disgust 🤢"
+        sentiment_analysis:
+          positive: "Positive"
+          negative: "Negative"
+          neutral: "Neutral"
 
     llm:
       configuration:

lib/sentiment/sentiment_analysis_report.rb

@@ -46,6 +46,23 @@ def self.register!(plugin)
                   end,
               }
             end
+
+          # TODO: connect filter to make the report data change.
+          filter_type = report.filters.dig(:filter_type) || "Category"
+          report.add_filter(
+            "filter_by",
+            type: "list",
+            default: filter_type,
+            choices: [{ id: "category", name: "Category" }, { id: "tag", name: "Tag" }],
+            allow_any: false,
+            auto_insert_none_item: false,
+          )
+
+          report.labels = [
+            I18n.t("discourse_ai.sentiment.reports.sentiment_analysis.positive"),
+            I18n.t("discourse_ai.sentiment.reports.sentiment_analysis.neutral"),
+            I18n.t("discourse_ai.sentiment.reports.sentiment_analysis.negative"),
+          ]
         end
       end