From 7c2217e7a31ff7c91411154e1f5520ebc143d611 Mon Sep 17 00:00:00 2001
From: David Gutman <dagutman@gmail.com>
Date: Thu, 24 Aug 2023 15:20:06 -0400
Subject: [PATCH] added virtualRowData

---
 src/components/dataset_view.py | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/src/components/dataset_view.py b/src/components/dataset_view.py
index 07fb558..e617416 100644
--- a/src/components/dataset_view.py
+++ b/src/components/dataset_view.py
@@ -3,6 +3,7 @@
 import dash_bootstrap_components as dbc
 from typing import List
 import pandas as pd
+import dash_ag_grid as dag
 
 from ..utils.database import getProjectDataset
 from ..utils.helpers import generate_generic_DataTable
@@ -25,8 +26,8 @@
                 dmc.TabsPanel(
                     [
                         html.Div(
-                            html.Div(),
                             id="project-itemSet-div",
+                            children=[dag.AgGrid(id="project-itemSet-table")],
                         ),
                     ],
                     value="table",
@@ -98,9 +99,16 @@ def updateProjectItemStore(projectId: str, projectData: List[dict]) -> List[dict
 
 @callback(
     Output("filteredItem_store", "data"),
-    [Input("projectItem_store", "data"), Input("tasks-dropdown", "value")],
+    [
+        Input("projectItem_store", "data"),
+        Input("tasks-dropdown", "value"),
+        Input("project-itemSet-table", "filterModel"),
+        Input("project-itemSet-table", "virtualRowData"),
+    ],
 )
-def updateFilteredItemStore(projectItemSet, selectedTask):
+def updateFilteredItemStore(
+    projectItemSet, selectedTask, tableFilterModel, virtualRowData
+):
     ### Update the filteredItemStore based on selected task...
     print(
         len(projectItemSet),
@@ -108,6 +116,13 @@ def updateFilteredItemStore(projectItemSet, selectedTask):
         selectedTask,
     )
 
+    if tableFilterModel:
+        print("JC Gets to figure out how to use this too to the dataframe!")
+        print(tableFilterModel)
+
+    if virtualRowData:
+        print(len(virtualRowData), "rows in the virtual data table..")
+
     if projectItemSet:
         df = pd.json_normalize(projectItemSet, sep="-")