Update app.py

app.py

@@ -9,13 +9,9 @@ import streamlit.components.v1 as components
 from annotated_text import annotated_text
 from streamlit_tags import st_tags
 from PyPDF2 import PdfReader, PdfWriter
-from presidio_helpers import (
-    get_supported_entities,
-    analyze,
-    anonymize,
-    annotate,
-    analyzer_engine,
-)
+from presidio_analyzer import AnalyzerEngine, RecognizerRegistry
+from presidio_anonymizer import AnonymizerEngine
+from presidio_helpers import get_supported_entities, analyze, anonymize
 
 st.set_page_config(
     page_title="Presidio PHI De-identification",
@@ -53,7 +49,7 @@ st.sidebar.markdown(f"[View model on HuggingFace]({selected_model_url})")
 st_model_package = st_model.split("/")[0]
 st_model = st_model if st_model_package.lower() not in ("spacy", "huggingface") else "/".join(st_model.split("/")[1:])
 
-analyzer_params = (st_model_package, st_model, "", "")
+analyzer_params = (st_model_package, st_model)
 st.sidebar.warning("Note: Models might take some time to download.")
 
 st_operator = st.sidebar.selectbox(
@@ -97,7 +93,7 @@ with col1:
         # Analyze
         analyzer = analyzer_engine(*analyzer_params)
         st_analyze_results = analyze(
-            *analyzer_params,
+            analyzer,
             text=text,
             entities=get_supported_entities(*analyzer_params),
             language="en",
@@ -115,10 +111,12 @@ with col1:
             st.info("No PHI detected")
 
         # Anonymize
+        anonymizer = AnonymizerEngine()
         anonymized_result = anonymize(
             text=text,
             operator=st_operator,
             analyze_results=st_analyze_results,
+            anonymizer=anonymizer,
         )
 
         # Create new PDF