From 2ce10dddc926a70130cc356e87e6a5b1f933d838 Mon Sep 17 00:00:00 2001
From: "Abdulkadir, Ahmed" <ahmed.abdulkadir@pennmedicine.upenn.edu>
Date: Tue, 30 Nov 2021 16:04:39 -0500
Subject: [PATCH] Add data exploration tool

---
 QtBrainChartGUI/plugins/data/data.py | 10 +++++++++-
 QtBrainChartGUI/plugins/data/data.ui |  7 +++++++
 setup.cfg                            |  1 +
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/QtBrainChartGUI/plugins/data/data.py b/QtBrainChartGUI/plugins/data/data.py
index fa3ec6ae9..2714bb85c 100644
--- a/QtBrainChartGUI/plugins/data/data.py
+++ b/QtBrainChartGUI/plugins/data/data.py
@@ -4,7 +4,7 @@
 import sys, os
 import pandas as pd
 from QtBrainChartGUI.plugins.data.dataio import DataIO
-
+import dtale
 
 class PandasModel(QtCore.QAbstractTableModel):
     def __init__(self, data, parent=None):
@@ -47,6 +47,14 @@ def SetupConnections(self):
         self.ui.open_data_file_Btn.clicked.connect(lambda: self.OnOpenDataFileBtnClicked())
         self.datamodel.data_changed.connect(lambda: self.OnDataChanged())
         self.ui.save_data_Btn.clicked.connect(lambda: self.OnSaveDataBtClicked())
+        self.ui.dtale_Btn.clicked.connect(lambda: self.OnDtaleBtnClicked())
+
+
+    def OnDtaleBtnClicked(self):
+        if ('level_0' in self.datamodel.data.keys()):
+            self.datamodel.data.drop('level_0', axis=1, inplace=True)
+        d = dtale.show(self.datamodel.data.reset_index(drop=True))
+        d.open_browser()
 
 
     def OnSaveDataBtClicked(self):
diff --git a/QtBrainChartGUI/plugins/data/data.ui b/QtBrainChartGUI/plugins/data/data.ui
index 8c0d8ad02..7d2ef04c2 100644
--- a/QtBrainChartGUI/plugins/data/data.ui
+++ b/QtBrainChartGUI/plugins/data/data.ui
@@ -37,6 +37,13 @@
        </property>
       </widget>
      </item>
+     <item>
+      <widget class="QPushButton" name="dtale_Btn">
+       <property name="text">
+        <string>Explore...</string>
+       </property>
+      </widget>
+     </item>
     </layout>
    </item>
    <item>
diff --git a/setup.cfg b/setup.cfg
index 1db46fd87..942fb382f 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -19,6 +19,7 @@ python_requires = >=3.8
 include_package_data = True
 install_requires = 
                 cycler==0.10.0
+                dtale==1.61.1
                 joblib==1.0.1
                 kiwisolver==1.3.1
                 matplotlib==3.4.2