utils/get_ner_results.py

from keys import endpoint, key
# endpoint = "https://eastus.api.cognitive.microsoft.com/"

from azure.ai.textanalytics import TextAnalyticsClient
from azure.core.credentials import AzureKeyCredential

# Authenticate the client using your key and endpoint 
def authenticate_client():
 ta_credential = AzureKeyCredential(key)
 text_analytics_client = TextAnalyticsClient(
 endpoint=endpoint, 
 credential=ta_credential)
 return text_analytics_client

client = authenticate_client()


def replace_original_text_with_entities(original_text:str):
 try:
 result = client.recognize_entities(documents = [original_text])[0]

 for entity in result.entities:
 # print("\tText: \t", entity.text, "\tCategory: \t", entity.category, "\tSubCategory: \t", entity.subcategory,
 # "\n\tConfidence Score: \t", round(entity.confidence_score, 2), "\tLength: \t", entity.length, "\tOffset: \t", entity.offset, "\n")
 original_text= original_text.replace(
 entity.text, 
 entity.text+ f' ({entity.category}) '
 )
 return original_text

 except Exception as err:
 
 print("Encountered exception. {}".format(err))
 return original_text