Upload 10 files

presidio_streamlit.py

@@ -1,11 +1,12 @@
 """Streamlit app for Presidio."""
 import logging
 import os
+import traceback
 
+import dotenv
 import pandas as pd
 import streamlit as st
 import streamlit.components.v1 as components
-import dotenv
 from annotated_text import annotated_text
 from streamlit_tags import st_tags
 
@@ -17,7 +18,6 @@ from presidio_helpers import (
     annotate,
     create_fake_data,
     analyzer_engine,
-    nlp_engine_and_registry,
 )
 
 st.set_page_config(
@@ -35,7 +35,6 @@ logger = logging.getLogger("presidio-streamlit")
 
 allow_other_models = os.getenv("ALLOW_OTHER_MODELS", False)
 
-can_present_results = True
 
 # Sidebar
 st.sidebar.header(
@@ -172,7 +171,7 @@ elif st_operator == "synthesize":
         api_type=openai_api_type,
     )
 
-    can_present_results = True if st_openai_key else False
+    input_is_valid = True if st_openai_key else False
 
 st_threshold = st.sidebar.slider(
     label="Acceptance threshold",
@@ -241,27 +240,9 @@ with st.expander("About this demo", expanded=False):
     )
 
 analyzer_load_state = st.info("Starting Presidio analyzer...")
-nlp_engine, registry = nlp_engine_and_registry(*analyzer_params)
 
 analyzer_load_state.empty()
 
-# Choose entities
-st_entities_expander = st.sidebar.expander("Choose entities to look for")
-st_entities = st_entities_expander.multiselect(
-    label="Which entities to look for?",
-    options=get_supported_entities(*analyzer_params),
-    default=list(get_supported_entities(*analyzer_params)),
-    help="Limit the list of PII entities detected. "
-    "This list is dynamic and based on the NER model and registered recognizers. "
-    "More information can be found here: https://microsoft.github.io/presidio/analyzer/adding_recognizers/",
-)
-
-
-analyzer_load_state = st.info("Starting Presidio analyzer...")
-analyzer = analyzer_engine(*analyzer_params)
-analyzer_load_state.empty()
-
-
 # Read default text
 with open("demo_text.txt") as f:
     demo_text = f.readlines()
@@ -275,20 +256,35 @@ st_text = col1.text_area(
     label="Enter text", value="".join(demo_text), height=400, key="text_input"
 )
 
+try:
+    # Choose entities
+    st_entities_expander = st.sidebar.expander("Choose entities to look for")
+    st_entities = st_entities_expander.multiselect(
+        label="Which entities to look for?",
+        options=get_supported_entities(*analyzer_params),
+        default=list(get_supported_entities(*analyzer_params)),
+        help="Limit the list of PII entities detected. "
+        "This list is dynamic and based on the NER model and registered recognizers. "
+        "More information can be found here: https://microsoft.github.io/presidio/analyzer/adding_recognizers/",
+    )
 
-st_analyze_results = analyze(
-    *analyzer_params,
-    text=st_text,
-    entities=st_entities,
-    language="en",
-    score_threshold=st_threshold,
-    return_decision_process=st_return_decision_process,
-    allow_list=st_allow_list,
-    deny_list=st_deny_list,
-)
+    # Before
+    analyzer_load_state = st.info("Starting Presidio analyzer...")
+    analyzer = analyzer_engine(*analyzer_params)
+    analyzer_load_state.empty()
+
+    st_analyze_results = analyze(
+        *analyzer_params,
+        text=st_text,
+        entities=st_entities,
+        language="en",
+        score_threshold=st_threshold,
+        return_decision_process=st_return_decision_process,
+        allow_list=st_allow_list,
+        deny_list=st_deny_list,
+    )
 
-# After
-if can_present_results:
+    # After
     if st_operator not in ("highlight", "synthesize"):
         with col2:
             st.subheader(f"Output")
@@ -348,6 +344,11 @@ if can_present_results:
     else:
         st.text("No findings")
 
+except Exception as e:
+    print(e)
+    traceback.print_exc()
+    st.error(e)
+
 components.html(
     """
     <script type="text/javascript">