from rake_nltk import Rake
import nltk
nltk.download('stopwords')

# Uses stopwords for english from NLTK, and all puntuation characters by
# default
class KeyExtractor:
    def __init__(self):
        self.model = Rake()
    def get_keywords(self, text):
        # Extraction given the text.
        self.model.extract_keywords_from_text(text)
        # To get keyword phrases ranked highest to lowest.
        imp_keywords = self.model.get_ranked_phrases()[0:4]

        result = []
        for answer in imp_keywords:
            result.append(answer)
        return result