Update app.py

app.py

@@ -47,29 +47,27 @@ if text_input:
     def get_entity_description(entity_link, combined_entity_info_dictionary):
         return combined_entity_info_dictionary[entity_link][1]
 
-    annotations = []
-    for wikidata_link, entity in entities_map.items():
-        description = get_entity_description(wikidata_link, combined_entity_info_dictionary)
-        annotations.append((entity, description, "#8ef"))
-
-    # Annotate text with entities
     if submit_button:
-        # Split the input text into words
-        words = nltk.word_tokenize(text_input)
-
         # Prepare a list to hold the final output
         final_text = []
 
-        for word in words:
-            # If the word is an entity, annotate it
-            if word in entities_map.keys():
-                final_text.append((word, get_entity_description(word, combined_entity_info_dictionary), "#8ef"))
-            # If the word is not an entity, keep it as it is
+        # Replace each entity in the text with its annotated version
+        for wikidata_link, entity in entities_map.items():
+            description = get_entity_description(wikidata_link, combined_entity_info_dictionary)
+            entity_annotation = (entity, description, "#8ef")
+            text_input = text_input.replace(entity, f'{{{str(entity_annotation)}}}', 1)
+
+        # Split the modified text_input into a list
+        text_list = text_input.split("{")
+
+        for item in text_list:
+            if "}" in item:
+                item_list = item.split("}")
+                final_text.append(eval(item_list[0]))
+                if len(item_list[1]) > 0:
+                    final_text.append(item_list[1])
             else:
-                final_text.append(word + " ")
+                final_text.append(item)
 
         # Pass the final_text to the annotated_text function
-        annotated_text(final_text)
-
-        with st.expander("See annotations"):
-            st.write(final_text)
+        annotated_text(*final_text)