Update app.py

app.py

@@ -66,13 +66,13 @@ model = load_model(selected_language, selected_model_name, selected_entity_set)
 def get_wikidata_id(entity_string):
     entity_list = entity_string.split("=")
     entity_id = str(entity_list[1])
-    entity_link = "http/www.wikidata.org/entity/" + entity_id
+    entity_link = "http://www.wikidata.org/entity/" + entity_id
     return {"id": entity_id, "link": entity_link}
     
 def get_entity_data(entity_link):
     try:
-        # Ensure that the URL format is correct
-        formatted_link = entity_link.replace("https://", "").replace("http://", "")
+        # Format the entity_link
+        formatted_link = entity_link.replace("http://", "http/")
         response = requests.get(f'https://api.wordlift.io/id/{formatted_link}')
         return response.json()
     except Exception as e:
@@ -87,6 +87,17 @@ with st.form(key='my_form'):
 # When processing the text, check the language and adjust processing accordingly
 if text_input:
     if selected_language == "German":
+        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
+            # Ensure correct format for the German model
+            formatted_wikidata_url = wikidata_url.replace("https://", "http://")
+            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
+        
         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]