Update presidio_streamlit.py

presidio_streamlit.py

@@ -49,7 +49,7 @@ model_help_text = """
 st_ta_key = st_ta_endpoint = ""
 
 model_list = [
-    "spaCy/cs_core_news_sm",
+    "spacy/en_core_web_sm",
     "iiiorg/piiranha-v1-detect-personal-information",
     "FacebookAI/xlm-roberta-large-finetuned-conll03-english",
 ]
@@ -69,13 +69,13 @@ st_model_package = st_model.split("/")[0]
 # Odstranění prefixu balíčku (pokud je potřeba)
 st_model = (
     st_model
-    if st_model_package.lower() not in ("spacy", "piiiranha")
+    if st_model_package.lower() not in ("spacy", "iiiorg")
     else "/".join(st_model.split("/")[1:])
 )
 
 if st_model == "Other":
     st_model_package = st.sidebar.selectbox(
-        "NER model OSS balíček", options=["spacy", "piiiranha"]
+        "NER model OSS balíček", options=["spacy", "iiiorg"]
     )
     st_model = st.sidebar.text_input(f"Název NER modelu", value="")
 
@@ -338,12 +338,7 @@ try:
             },
             axis=1,
         )
-        df_subset["Text"] = [st_text[res.start : res.end] for res in st_analyze_results]
-        if st_return_decision_process:
-            analysis_explanation_df = pd.DataFrame.from_records(
-                [r.analysis_explanation.to_dict() for r in st_analyze_results]
-            )
-            df_subset = pd.concat([df_subset, analysis_explanation_df], axis=1)
+       df_subset = pd.concat([df_subset, analysis_explanation_df], axis=1)
         st.dataframe(df_subset.reset_index(drop=True), use_container_width=True)
     else:
         st.text("Žádné nálezy")