Update app.py

app.py

@@ -117,6 +117,12 @@ def create_masked_text(input_text, entities):
                 masked_text[entity['end']:]
             )
     return masked_text
+    
+def replace_words_in_text(input_text, entities):
+     replace_dict = create_mask_dict(entities)
+    for word, replacement in replace_dict.items():
+        text = text.replace(word, replacement)
+    return text
 
 Run_Button = st.button("Run")
 
@@ -144,7 +150,7 @@ if Run_Button and input_text:
     output_comb = entity_comb(all_outputs)
     
     # Create masked text and masking dictionary
-    masked_text = create_masked_text(input_text, output_comb)
+    masked_text = replace_words_in_text(input_text, output_comb)#create_masked_text(input_text, output_comb)
     mask_dict = create_mask_dict(output_comb)
 
     # Display the masked text and masking dictionary