Update app.py

app.py

@@ -37,7 +37,7 @@ def getcolor(texts, labels):
     colors = {'LOC': '#38419D', 'PER': '#BF3131', 'ORG': '#597E52', 'MISC':'#F1C232'}
     return [(t,l,colors[l]) for t, l in zip(texts, labels)]
 
-def color_annotation(to_print,transcript) :  
+def color_annotation(to_print,text) :  
     text_ner = []
     label_ner = []
     for i in range(len(to_print)) :
@@ -49,22 +49,18 @@ def color_annotation(to_print,transcript) :
     text_ner = list(set(text_ner))
     text_ner = list(sorted(text_ner, key = len))
 
-    display = transcript.replace(" ",' ","')
     for i in range(len(anns)):
         for j in range(len(text_ner)):
             if text_ner[j] == anns[i][0]:
-                 display = display.replace(text_ner[j].replace(" ",' ","'),str(anns[i]))
-    display = display.replace(', ","' , ',')
-    
-    for i in re.findall(r"\((.*?)\)", display) : # pour gérer les cas de mots inclus dans des n_grams
+                 text = text.replace(text_ner[j],str(anns[i]))
+
+    for i in re.findall(r"\((.*?)\)", text) : # pour gérer les cas de mots inclus dans des n_grams
         if "(" in i:
-            display = display.replace(i+")",i.split(",")[0].replace("('","")).replace("''","'")
+            text = text.replace(i+")",i.split(', ')[0][2:-1])
             
-    # sanity check
-    display = display.replace(")",')","').replace(')","","',')","').replace("(",'","(').replace('","","(','","(').replace("'-","-")
-
-    return display
+    text = text.replace(")",')","').replace(')","","',')","').replace("(",'","(').replace('","","(','","(').replace("'-","-")
 
+    return text