Hariharan Vijayachandran commited on
Commit
9a753c2
·
1 Parent(s): eae62cc
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -3,7 +3,7 @@ import pandas as pd
3
  import sys
4
  import os
5
  from datasets import load_from_disk
6
- from st_aggrid import AgGrid, GridOptionsBuilder, GridUpdateMode
7
  from sklearn.metrics.pairwise import cosine_similarity
8
  import numpy as np
9
  import time
@@ -73,15 +73,15 @@ def get_annotated_text(text, word, pos):
73
  end = start+len(word)
74
  return (text[:start], (text[start:end ], 'SELECTED'), text[end:]), end
75
 
76
- class AgGridBuilder:
77
- __static_key = 0
78
- def build_ag_grid(table, display_columns):
79
- AgGridBuilder.__static_key += 1
80
- options_builder = GridOptionsBuilder.from_dataframe(table[display_columns])
81
- options_builder.configure_pagination(paginationAutoPageSize=False, paginationPageSize=10)
82
- options_builder.configure_selection(selection_mode= 'single', pre_selected_rows = [0])
83
- options = options_builder.build()
84
- 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)
85
 
86
  if __name__ == '__main__':
87
  st.set_page_config(layout="wide")
 
3
  import sys
4
  import os
5
  from datasets import load_from_disk
6
+ # from st_aggrid import AgGrid, GridOptionsBuilder, GridUpdateMode
7
  from sklearn.metrics.pairwise import cosine_similarity
8
  import numpy as np
9
  import time
 
73
  end = start+len(word)
74
  return (text[:start], (text[start:end ], 'SELECTED'), text[end:]), end
75
 
76
+ # class AgGridBuilder:
77
+ # __static_key = 0
78
+ # def build_ag_grid(table, display_columns):
79
+ # AgGridBuilder.__static_key += 1
80
+ # options_builder = GridOptionsBuilder.from_dataframe(table[display_columns])
81
+ # options_builder.configure_pagination(paginationAutoPageSize=False, paginationPageSize=10)
82
+ # options_builder.configure_selection(selection_mode= 'single', pre_selected_rows = [0])
83
+ # options = options_builder.build()
84
+ # 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)
85
 
86
  if __name__ == '__main__':
87
  st.set_page_config(layout="wide")