Update app.py

app.py

@@ -71,12 +71,14 @@ def get_wikidata_id(entity_string):
     
 def get_entity_data(entity_link):
     try:
-        response = requests.get(f'https://api.wordlift.io/id/{entity_link}')
+        # Ensure that the URL format is correct
+        formatted_link = entity_link.replace("https://", "").replace("http://", "")
+        response = requests.get(f'https://api.wordlift.io/id/{formatted_link}')
         return response.json()
     except Exception as e:
         print(f"Exception when fetching data for entity: {entity_link}. Exception: {e}")
         return None
-    
+            
 # Create the form
 with st.form(key='my_form'):
     text_input = st.text_area(label='Enter a sentence')
@@ -127,6 +129,10 @@ if text_input:
     
        # Replace each entity in the text with its annotated version
         for entity_string, entity_info in entities_map.items():
+            # Check if the entity has a valid Wikidata link
+            if entity_info["link"] is None or entity_info["link"] == "None":
+                continue  # skip this entity
+            
             entity_data = entities_data.get(entity_string, None)
             entity_type = None
             if entity_data is not None:
@@ -152,7 +158,8 @@ if text_input:
                 
             # Add the entity to JSON-LD data
             entity_json_ld = combined_entity_info_dictionary[entity_string][1]
-            json_ld_data["mentions"].append(entity_json_ld)
+            if entity_json_ld and entity_json_ld.get("link") != "None":
+                json_ld_data["mentions"].append(entity_json_ld)
 
         # Split the modified text_input into a list
         text_list = text_input.split("{")