Update app.py

app.py

@@ -22,10 +22,10 @@ def get_result_text_es_pt (list_entity, text, lang):
         # check subwords
         if word[0] == "#": 
             subword = True
-            if punc_in != "":
-                word = result_words[-1].replace(punc_in, "") + text[start:end]
-            else: 
-                word = result_words[-1] + text[start:end]
+            p_s = list_entity[idx-1]["start"]
+            p_e = list_entity[idx-1]["end"]
+            # word = result_words[-1].replace(punc_in, "") + text[start:end]
+            word = text[p_s:p_e] + text[start:end]
         
         if tag == "l": 
             word = word 
@@ -67,10 +67,10 @@ def get_result_text_ca (list_entity, text):
         # check subwords
         if word[0] != "Ġ": 
             subword = True
-            if punc_in != "":
-                word = result_words[-1].replace(punc_in, "") + text[start:end]
-            else: 
-                word = result_words[-1] + text[start:end]
+            p_s = list_entity[idx-1]["start"]
+            p_e = list_entity[idx-1]["end"]
+            # word = result_words[-1].replace(punc_in, "") + text[start:end]
+            word = text[p_s:p_e] + text[start:end]
         else: 
             word = text[start:end]