Update app.py

app.py

@@ -88,23 +88,30 @@ with st.form(key='my_form'):
 if text_input:
     if selected_language == "German":
         doc_de = model(text_input)
-        entities = [(ent.text, ent.label_) for ent in doc_de.ents]  # Adjust this based on your needs
-
+        # 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]
     else:
-    
         entities = refined_model.process_text(text_input)
-    
-        entities_map = {}
-        entities_data = {}
-        for entity in entities:
+
+    # 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:
             single_entity_list = str(entity).strip('][').replace("\'", "").split(', ')
-            if len(single_entity_list) >= 2 and "wikidata" in single_entity_list[1]: 
+            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])
                 entity_data = get_entity_data(entities_map[single_entity_list[0].strip()]["link"])
                 if entity_data is not None:
                     entities_data[single_entity_list[0].strip()] = entity_data
-    
-        combined_entity_info_dictionary = dict([(k, [entities_map[k], entities_data[k] if k in entities_data else None]) for k in entities_map])
+
+    combined_entity_info_dictionary = dict([(k, [entities_map[k], entities_data[k] if k in entities_data else None]) for k in entities_map])
     
         if submit_button:
             # Prepare a list to hold the final output