import spacy
from commons.Configs import configs


class SpacyUtils:
    def __init__(self, debug=False):
        self.debug = debug

    # Receives a raw text and returns an array of sentences
    def splitSentences(self, text):
        """Split text into sentences"""
        nlp = self.spacyLoad()
        doc = nlp(text)
        return [str(sent.text).replace('"', '') for sent in doc.sents]

    # Returns a spacy.load() model
    def spacyLoad(self):
        """Load spacy model"""
        if not hasattr(self, 'spacyInstance'):
            self.spacyInstance = spacy.load(configs.spacyModel)
        return self.spacyInstance


spacyUtils = SpacyUtils()