Update app.py

app.py

@@ -106,7 +106,50 @@ label_to_ethnicity = {
 ethnicity_to_label = {v: k for k, v in label_to_ethnicity.items()}
 gender_to_label = {v: k for k, v in label_to_gender.items()}
 
+def make_table():
+    if 'table_data' in streamlit.session_state:
+        df = streamlit.session_state['table_data']
+    else:
+        refs = References(streamlit.session_state.bib)
+        refs.infer_gender()
+        refs.infer_ethnicity()
 
+        df = refs.raw_results
+        df = df.replace({"Most Likely Ethnicity": label_to_ethnicity})
+        df = df.replace({"Most Likely Gender": label_to_gender})
+        df = df.sort_values(["Last Name", "First Name"])
+        df = df.reset_index(drop=True)
+
+    gb = st_aggrid.GridOptionsBuilder.from_dataframe(df)
+    gb.configure_default_column(editable=True)
+
+    gb.configure_column('Most Likely Ethnicity',
+                        cellEditor='agRichSelectCellEditor',
+                        cellEditorParams={'values': list(label_to_ethnicity.values())},
+                        cellEditorPopup=True
+                        )
+
+    gb.configure_column('Most Likely Gender',
+                        cellEditor='agRichSelectCellEditor',
+                        cellEditorParams={'values': list(label_to_gender.values())},
+                        cellEditorPopup=True
+                        )
+
+    gb.configure_column('Title',
+                        editable=False
+                        )
+
+    response = st_aggrid.AgGrid(
+        data=df,
+        gridOptions=gb.build(),
+        fit_columns_on_grid_load=True,
+        update_mode=st_aggrid.GridUpdateMode.VALUE_CHANGED,
+        height=400,
+    )
+
+    streamlit.session_state['table_data'] = response.data
+    if response.column_state:
+        streamlit.experimental_rerun()
 
 
 # Define a function for addition
@@ -217,5 +260,6 @@ if time_to_analyze or 'already_analyzed' in streamlit.session_state:
     streamlit.session_state['already_analyzed'] = True
     placeholder.empty()
     with streamlit.spinner("Analyzing..."):
+        make_table()
         streamlit.markdown("These tabs summarize your results with a variety of visualizations and statistics.")
         make_results()