Update app.py

app.py

@@ -114,11 +114,6 @@ st_text = st.text_area(
     height=200,
 )
 
-button = st.button("Detect PII")
-
-if 'first_load' not in st.session_state:
-    st.session_state['first_load'] = True
-
 # After
 st.subheader("Analyzed")
 with st.spinner("Analyzing..."):
@@ -142,8 +137,24 @@ with st.spinner("Anonymizing..."):
     st_anonymize_results = anonymize(st_text, st_analyze_results)
     st_anonymize_results
 
+# table result
+st.subheader("Detailed Findings")
+if st_analyze_results:
+    res_dicts = [r.to_dict() for r in st_analyze_results]
+    for d in res_dicts:
+        d['Value'] = st_text[d['start']:d['end']]
+    df = pd.DataFrame.from_records(res_dicts)
+    df = df[["entity_type", "Value", "score", "start", "end"]].rename(
+        {
+            "entity_type": "Entity type",
+            "start": "Start",
+            "end": "End",
+            "score": "Confidence",
+        },
+        axis=1,
+    )
 
-
+    st.dataframe(df, width=1000)
 # table result
 else:
     st.text("No findings")