Update app.py

app.py

@@ -84,7 +84,10 @@ with st.form(key='my_form'):
     text_input = st.text_area(label='Enter a sentence')
     submit_button = st.form_submit_button(label='Analyze')
 
-# When processing the text, check the language and adjust processing accordingly
+# Initialization
+entities_map = {}
+entities_data = {}
+
 if text_input:
     if selected_language == "German":
         doc_de = model(text_input)
@@ -98,27 +101,11 @@ if text_input:
                 entity_data = get_entity_data(formatted_wikidata_url)
                 if entity_data is not None:
                     entities_data[entity_string] = entity_data
-                
-        doc_de = model(text_input)
-        # Map entities to a format similar to English output
-        entities = [(ent.text, ent.label_, ent._.kb_qid, ent._.url_wikidata) for ent in doc_de.ents]
-        # Debug
-        #for ent in doc_de.ents:
-        #    st.write(f"Entity: {ent.text}, Label: {ent.label_}, QID: {ent._.kb_qid}, URL: {ent._.url_wikidata}")
+
     else:
         entities = model.process_text(text_input)
 
-    # Logic for English language processing
-    entities_map = {}
-    entities_data = {}
-    for entity in entities:
-        if selected_language == "German":
-            entity_string, entity_type, wikidata_id, wikidata_url = entity
-            entities_map[entity_string] = {"id": wikidata_id, "link": wikidata_url}
-            entity_data = get_entity_data(wikidata_url)
-            if entity_data is not None:
-                entities_data[entity_string] = entity_data
-        else:
+        for entity in entities:
             single_entity_list = str(entity).strip('][').replace("\'", "").split(', ')
             if len(single_entity_list) >= 2 and "wikidata" in single_entity_list[1]:
                 entities_map[single_entity_list[0].strip()] = get_wikidata_id(single_entity_list[1])