From c657cdcf29f93b149d3fe5d663bf9c9667e4f9a6 Mon Sep 17 00:00:00 2001
From: da03 <dengyuntian@gmail.com>
Date: Mon, 23 Jun 2025 15:19:44 +0000
Subject: [PATCH] .

---
 main.py                       | 2 +-
 static/js/views/embeddings.js | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 16d76df..1307a57 100644
--- a/main.py
+++ b/main.py
@@ -44,7 +44,7 @@ def build_query_for_index(index_name, filters, contains, from_, size_):
             "nested": {
                 "path": "openai_moderation",
                 "query": {
-                    "term": {"openai_moderation.flagged": True}
+                    "term": {"openai_moderation.flagged": filters['toxic'] == 'true'}
                 }
             }
         })
diff --git a/static/js/views/embeddings.js b/static/js/views/embeddings.js
index 8cba09d..a62bda0 100644
--- a/static/js/views/embeddings.js
+++ b/static/js/views/embeddings.js
@@ -459,8 +459,13 @@ $(document).ready(function () {
             acc.set(d.i, d);
             return acc;
         }, new Map());
-        const uniqueDataArray = Array.from(mergedData.values());
 
+        // Filter by dataset if the dataset filter is set
+        let uniqueDataArray = Array.from(mergedData.values());
+        const datasetFilter = $('#filter-dataset').val();
+        if (datasetFilter) {
+            uniqueDataArray = uniqueDataArray.filter(d => d.dataset === datasetFilter);
+        }
 
         updateLayer(uniqueDataArray, highlightIds);
         hideLoading();