in src/ui/ui.py [0:0]
def _button_helper_record_name(self, b):
"""Helper method defining the button's action to manually detect a name (highlight).
Args:
b (-): Required argument on action functions for buttons.
Returns:
None
"""
name = self.add_name.value
category = self.add_category.value
p = re.compile(f"(?i)\\b{name}\\b")
# current entities set
current_names = set([d["Text"] for d in self.entities[category]])
if name not in current_names:
for match in re.finditer(p, self.nlq):
detected_entity = {
"BeginOffset": match.start(),
"EndOffset": match.end(),
"Text": name,
}
# add to raw entities
self.entities[category].append(deepcopy(detected_entity))
# process & add to proc entities
placeholder_idx_strat = {category: len(self.proc_entities[category])}
proc_detected_entity = self.tool.process_entities(
{category: [deepcopy(detected_entity)]},
start_indices=placeholder_idx_strat,
)
self.proc_entities[category].append(proc_detected_entity[category][0])
self._display_main()
self._update_options()