Updating topic_word in AgGrid

app/pages/02_Document_Explorer_📖.py

@@ -90,7 +90,7 @@ def main():
             builder = GridOptionsBuilder.from_dataframe(data[ordered_cols])
             builder.configure_pagination()
             go = builder.build()
-            AgGrid(st.session_state.selected_data[cols], theme='streamlit', gridOptions=go,
+            AgGrid(data[ordered_cols], theme='streamlit', gridOptions=go,
                    columns_auto_size_mode=ColumnsAutoSizeMode.FIT_CONTENTS)
         else:
             st.markdown('Select points in the graph with the `lasso` or `box` select tools to populate this table.')

app/pages/03_Semantic_Search_🔍.py

@@ -91,12 +91,13 @@ def main():
         cols = ['id', 'document_scores', 'topic_id', 'documents']
         data = st.session_state.data_to_model_without_point.loc[:, cols]
         data['topic_word'] = data.topic_id.replace(st.session_state.topic_str_to_word)
-        builder = GridOptionsBuilder.from_dataframe(data)
+        ordered_cols = ['id', 'document_scores', 'topic_id', 'topic_word', 'documents']
+        builder = GridOptionsBuilder.from_dataframe(data[ordered_cols])
         builder.configure_pagination()
         builder.configure_column('document_scores', type=["numericColumn", "numberColumnFilter", "customNumericFormat"],
                                  precision=2)
         go = builder.build()
-        AgGrid(st.session_state.data_to_model_without_point.loc[:, cols], theme='streamlit', gridOptions=go,
+        AgGrid(data[ordered_cols], theme='streamlit', gridOptions=go,
                columns_auto_size_mode=ColumnsAutoSizeMode.FIT_CONTENTS)
 
     with tab2: