From 54d98c598f21439c9f1c8c285eb8284336dc1a14 Mon Sep 17 00:00:00 2001
From: Aaron Diamond-Reivich <aarondr77@gmail.com>
Date: Tue, 7 Nov 2023 16:47:36 -0500
Subject: [PATCH] using the new Spreadsheet API

---
 app-mito.py      | 15 +++++++--------
 requirements.txt |  3 ++-
 utils.py         |  3 +--
 3 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/app-mito.py b/app-mito.py
index 58718b7..32956e9 100644
--- a/app-mito.py
+++ b/app-mito.py
@@ -2,13 +2,11 @@
 from dash import Dash, html, callback, Input, Output, dcc, dash_table, State
 from dash.exceptions import PreventUpdate
 import base64
-from mitosheet.mito_dash.v1 import Spreadsheet, mito_callback
-
-import io
-import plotly.express as px
+from mitosheet.mito_dash.v1 import Spreadsheet, mito_callback, activate_mito
 from utils import get_correlation_df, get_graph_group
 
 app = Dash(__name__)
+activate_mito(app)
 
 app.layout = dmc.MantineProvider(
     [
@@ -107,7 +105,7 @@
         
         html.Div(
             [
-                Spreadsheet(id='spreadsheet', import_folder='data'),
+                Spreadsheet(id={'type': 'spreadsheet', 'id': 'sheet'}, import_folder='data'),
             ],
             style={"height": "80%", "maxWidth": "80%", "margin": "auto", "padding": "10px"},
         ),
@@ -154,9 +152,10 @@
 )
 
 @callback(
-    Output("spreadsheet", "data"), 
+    Output({'type': 'spreadsheet', 'id': 'sheet'}, "data"), 
     [Input("upload-data", "contents")], 
-    [State("spreadsheet", "data")])
+    [State({'type': 'spreadsheet', 'id': 'sheet'}, "data")]
+)
 def update_spreadsheet_data(uploaded_contents, data):
     if uploaded_contents is None:
         raise PreventUpdate
@@ -178,7 +177,7 @@ def update_spreadsheet_data(uploaded_contents, data):
 @mito_callback(
     Output("graph-output", "children"),
     Output("correlation-table", "children"),
-    Input("spreadsheet", "spreadsheet_result"),
+    Input({'type': 'spreadsheet', 'id': 'sheet'}, "spreadsheet_result"),
 )
 def update_outputs(spreadsheet_result):
     if spreadsheet_result is None or len(spreadsheet_result.dfs()) == 0:
diff --git a/requirements.txt b/requirements.txt
index c86d26d..8ebe117 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,5 @@ dash-mantine-components
 dash
 dash-pivottable
 pandas
-plotly
\ No newline at end of file
+plotly
+mitosheet
\ No newline at end of file
diff --git a/utils.py b/utils.py
index 57e787d..7a59995 100644
--- a/utils.py
+++ b/utils.py
@@ -99,5 +99,4 @@ def get_graph_group(df, date_columns):
             side_by_side_graphs,
             moving_average_graph
         ]
-    )
-
+    )
\ No newline at end of file