Extract entities from text using the AlchemyAPI

alchemy_api_entity_extraction_text.py

import requests

entities_endpoint = "http://gateway-a.watsonplatform.net/calls/text/TextGetRankedNamedEntities"
def get_entities(text):
    data = {
        'apikey': API_KEY,
        'outputMode': 'json',
        'text': text
    }
    headers = {'content-type': 'application/x-www-form-urlencoded'}
    r = requests.post(entities_endpoint, data=data, headers=headers)
    if r.status_code == 200:
        resp = r.json()
        # Print entities, type and relevance score.
        # Entities are already ranked by relevance, high -> low
        entities = [e['text'] for e in resp.get('entities')]
    
        # Print top 3 most relevant
        return entities