feat: Pull dataset values from DB (#167)

navapbc · Jan 6, 2025 · d61791c · d61791c
1 parent d3db181
commit d61791c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/app/notebooks/metrics/q_a_generation.ipynb b/app/notebooks/metrics/q_a_generation.ipynb
@@ -113,9 +113,17 @@
     "    disabled=False\n",
     ")\n",
     "\n",
+    "with app_config.db_session() as db_session:\n",
+    "    unique_datasets = (\n",
+    "        db_session.query(Document.dataset)\n",
+    "        .distinct()\n",
+    "        .all()\n",
+    "    )\n",
+    "    unique_dataset_list = [row[0] for row in unique_datasets]  # Extract values from the query result\n",
+    "\n",
     "question_dataset = widgets.SelectMultiple(\n",
     "    options=['Imagine LA', 'CA EDD', 'BEM'],\n",
-    "    value=['Imagine LA', 'CA EDD' ],\n",
+    "    value=[],\n",
     "    description='Dataset source:',\n",
     "    disabled=False\n",
     ")\n",