Update app.py

app.py

@@ -99,12 +99,12 @@ def create_mask_dict(entities, additional_masks=None):
     
     for entity in entities:
         if entity['entity_group'] not in ['CARDINAL', 'EVENT', 'PERCENT', 'QUANTITY']:
-        if entity['word'] not in mask_dict:
-            if entity['entity_group'] not in entity_counters:
-                entity_counters[entity['entity_group']] = 1
-            else:
-                entity_counters[entity['entity_group']] += 1
-            mask_dict[entity['word']] = f"{entity['entity_group']}_{entity_counters[entity['entity_group']]}"
+            if entity['word'] not in mask_dict:  # Corrected indentation
+                if entity['entity_group'] not in entity_counters:
+                    entity_counters[entity['entity_group']] = 1
+                else:
+                    entity_counters[entity['entity_group']] += 1
+                mask_dict[entity['word']] = f"{entity['entity_group']}_{entity_counters[entity['entity_group']]}"
     
     if additional_masks:
         for word, replacement in additional_masks.items():
@@ -112,6 +112,7 @@ def create_mask_dict(entities, additional_masks=None):
 
     return mask_dict
 
+
 def replace_words_in_text(input_text, entities):
     replace_dict = create_mask_dict(entities)
     for word, replacement in replace_dict.items():