Update app.py

app.py

@@ -114,27 +114,34 @@ 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..."):
-    st_analyze_results = analyze(
-        text=st_text,
-        entities=st_entities,
-        language="en",
-        score_threshold=st_threshold,
-        return_decision_process=st_return_decision_process,
-    )
-    annotated_tokens = annotate(st_text, st_analyze_results, st_entities)
-    # annotated_tokens
-    annotated_text(*annotated_tokens)
+    if button or st.session_state.first_load:
+        st_analyze_results = analyze(
+            text=st_text,
+            entities=st_entities,
+            language="en",
+            score_threshold=st_threshold,
+            return_decision_process=st_return_decision_process,
+        )
+        annotated_tokens = annotate(st_text, st_analyze_results, st_entities)
+        # annotated_tokens
+        annotated_text(*annotated_tokens)
 # vertical space
 st.text("")
     
 st.subheader("Anonymized")
 
 with st.spinner("Anonymizing..."):
-    st_anonymize_results = anonymize(st_text, st_analyze_results)
-    st_anonymize_results
+    if button or st.session_state.first_load:
+        st_anonymize_results = anonymize(st_text, st_analyze_results)
+        st_anonymize_results
 
 
 # table result
@@ -158,6 +165,7 @@ if st_analyze_results:
 else:
     st.text("No findings")
 
+st.session_state['first_load'] = True
 
 # json result
 class ToDictListEncoder(JSONEncoder):