fix

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import sys
 import os
 from datasets import load_from_disk
-from st_aggrid import AgGrid, GridOptionsBuilder, GridUpdateMode
+# from st_aggrid import AgGrid, GridOptionsBuilder, GridUpdateMode
 from sklearn.metrics.pairwise import cosine_similarity
 import numpy as np
 import time
@@ -73,15 +73,15 @@ def get_annotated_text(text, word, pos):
 	end = start+len(word)
 	return (text[:start], (text[start:end ], 'SELECTED'), text[end:]), end
 
-class AgGridBuilder:
-	__static_key = 0
-	def build_ag_grid(table, display_columns):
-		AgGridBuilder.__static_key += 1
-		options_builder = GridOptionsBuilder.from_dataframe(table[display_columns])
-		options_builder.configure_pagination(paginationAutoPageSize=False, paginationPageSize=10)
-		options_builder.configure_selection(selection_mode= 'single', pre_selected_rows = [0])
-		options = options_builder.build()
-		return AgGrid(table, gridOptions = options, fit_columns_on_grid_load=True, key = AgGridBuilder.__static_key, reload_data = True, update_mode = GridUpdateMode.SELECTION_CHANGED | GridUpdateMode.VALUE_CHANGED)
+# class AgGridBuilder:
+# 	__static_key = 0
+# 	def build_ag_grid(table, display_columns):
+# 		AgGridBuilder.__static_key += 1
+# 		options_builder = GridOptionsBuilder.from_dataframe(table[display_columns])
+# 		options_builder.configure_pagination(paginationAutoPageSize=False, paginationPageSize=10)
+# 		options_builder.configure_selection(selection_mode= 'single', pre_selected_rows = [0])
+# 		options = options_builder.build()
+# 		return AgGrid(table, gridOptions = options, fit_columns_on_grid_load=True, key = AgGridBuilder.__static_key, reload_data = True, update_mode = GridUpdateMode.SELECTION_CHANGED | GridUpdateMode.VALUE_CHANGED)
 
 if __name__ == '__main__':
 	st.set_page_config(layout="wide")