Update

app.py

@@ -6,7 +6,7 @@ import roner
 
 example_list = [
     "Ana merge în București.",
-    """Ana merge în București. Ana merge în București. Ana merge în București. Ana merge în București. Ana merge în București. Ana merge în București."""
+    """Președintele Statelor Unite, Joe Biden, a spus, vineri, că va trimite un număr de militari americani în Europa de Est „în curând”, ca urmare a situației tot mai tensionate din Ucraina. Președintele american a spus că „nu va trimite foarte mult” și a exclus din nou posibilitatea desfășurării de trupe în Ucraina, care nu face parte din Alianța Nord-Atlantică. """
 ]
 
 st.set_page_config(layout="wide")
@@ -62,7 +62,8 @@ if Run_Button == True:
                 "word": word["text"],
                 "entity": word["tag"],
                 "start_char": word["start_char"],
-                "end_char": word["end_char"]
+                "end_char": word["end_char"],
+                "multi-word-entity": word["multi_word_entity"]
             })
     df = pd.DataFrame.from_dict(data)
     st.subheader("Recognized Entities")
@@ -75,9 +76,23 @@ if Run_Button == True:
     spacy_display["text"] = output["text"]
     spacy_display["title"] = None
 
-    for word in output["words"]:
-        if word["tag"]!="O":
-            spacy_display["ents"].append({"start": word["start_char"], "end": word["end_char"], "label": word["tag"]})
+    i = 0
+    words = output["words"]
+    while i < len(words):
+        if words[i]["tag"]!="O":
+            start = words[i]["start_char"]
+            end = words[i]["end_char"]
+            label = words[i]["tag"]
+
+            for j in range(i+1,len(words)):
+                if words[j]["tag"] == words[i]["tag"] and words[j]["multi_word_entity"] is True:
+                    end = words[j]["end_char"]
+                    i = j
+
+            spacy_display["ents"].append({"start": start, "end": end, "label": label})
+        i += 1
+
+
     
     entity_list = ['PERSON', 'ORG', 'GPE', 'LOC', 'NAT_REL_POL',
      'EVENT', 'LANGUAGE', 'WORK_OF_ART', 'DATETIME',