update app

app.py

@@ -7,7 +7,7 @@ from presidio_analyzer import AnalyzerEngine, PatternRecognizer, RecognizerResul
 
 from annotated_text import annotated_text
 from flair_recognizer import FlairRecognizer
-
+from detoxify import Detoxify
 
 ###############################
 #### Render Streamlit page ####
@@ -15,13 +15,12 @@ from flair_recognizer import FlairRecognizer
 
 st.title("Anonymise your text!")
 st.markdown(
-    "This mini-app anonymises text using Flair. You can find the code in the Files and versions tab above."
+    "This mini-app anonymises text using Flair and Presidio. You can find the code in the Files and versions tab above."
 )
 
 # Configure logger
 logging.basicConfig(format="\n%(asctime)s\n%(message)s", level=logging.INFO, force=True)
 
-
 ##############################
 ###### Define functions ######
 ##############################
@@ -92,6 +91,20 @@ def analyze_text():
     if not st.session_state.text:
         st.session_state.text_error = "Please enter your text"
         return
+    
+    toxicity_results = Detoxify('original').predict(st.session_state.text)
+    is_toxic=False
+    for k in toxicity_results.keys():
+        for k in toxicity_results.keys():
+            if k!='toxicity':
+                if toxicity_results[k]>0.5:
+                    is_toxic=True
+            else:
+                if toxicity_results[k]>0.65:
+                    is_toxic=True
+    if is_toxic:
+        st.session_state.text_error = "Your text entry was detected as toxic, please re-write it."
+        return
 
     with text_spinner_placeholder:
         with st.spinner("Please wait while your text is being analysed..."):