Update app.py

app.py

@@ -19,11 +19,27 @@ st.set_page_config(
 
 # Sidebar
 st.sidebar.image("logo-wordlift.png")
-language_options = {"English", "German"}
+language_options = {"English", "English - spaCy", "German"}
 selected_language = st.sidebar.selectbox("Select the Language", list(language_options), index=0)
 
 # Based on selected language, configure model, entity set, and citation options
-if selected_language != "German":
+if selected_language == "German" or selected_language == "English - spaCy":
+    selected_model_name = None
+    selected_entity_set = None
+
+    entity_fishing_citation = """
+    @misc{entity-fishing,
+    title = {entity-fishing},
+    publisher = {GitHub},
+    year = {2016--2023},
+    archivePrefix = {swh},
+    eprint = {1:dir:cb0ba3379413db12b0018b7c3af8d0d2d864139c}
+    }
+    """
+
+    with st.sidebar.expander('Citations'):
+        st.markdown(entity_fishing_citation)
+else:
     model_options = ["aida_model", "wikipedia_model_with_numbers"]
     entity_set_options = ["wikidata", "wikipedia"]
     
@@ -41,22 +57,6 @@ if selected_language != "German":
 
     with st.sidebar.expander('Citations'):
         st.markdown(refined_citation)
-else:
-    selected_model_name = None
-    selected_entity_set = None
-
-    entity_fishing_citation = """
-    @misc{entity-fishing,
-    title = {entity-fishing},
-    publisher = {GitHub},
-    year = {2016--2023},
-    archivePrefix = {swh},
-    eprint = {1:dir:cb0ba3379413db12b0018b7c3af8d0d2d864139c}
-    }
-    """
-
-    with st.sidebar.expander('Citations'):
-        st.markdown(entity_fishing_citation)
 
 @st.cache_resource  # 👈 Add the caching decorator
 def load_model(selected_language, model_name=None, entity_set=None):
@@ -66,6 +66,12 @@ def load_model(selected_language, model_name=None, entity_set=None):
         nlp_model_de.add_pipe("entityfishing")
         
         return nlp_model_de
+    elif selected_language == "English":
+        # Load English-specific model
+        nlp_model_en = spacy.load("en_core_web_sm")
+        nlp_model_en.add_pipe("entityfishing")
+
+        return nlp_model_en    
     else:
         # Load the pretrained model for other languages
         refined_model = Refined.from_pretrained(model_name=model_name, entity_set=entity_set)
@@ -101,20 +107,19 @@ entities_map = {}
 entities_data = {}
 
 if text_input:
-    if selected_language == "German":
+    if selected_language in ["German", "English - spaCy"]::
         doc_de = model(text_input)
         entities = [(ent.text, ent.label_, ent._.kb_qid, ent._.url_wikidata) for ent in doc_de.ents]
         for entity in entities:
             entity_string, entity_type, wikidata_id, wikidata_url = entity
             if wikidata_url:
-                # Ensure correct format for the German model
+                # Ensure correct format for the German and English model
                 formatted_wikidata_url = wikidata_url.replace("https://www.wikidata.org/wiki/", "http://www.wikidata.org/entity/")
                 entities_map[entity_string] = {"id": wikidata_id, "link": formatted_wikidata_url}
                 entity_data = get_entity_data(formatted_wikidata_url)
                 
                 if entity_data is not None:
                     entities_data[entity_string] = entity_data
-
     else:
         entities = model.process_text(text_input)