diff --git a/server/concordia_preprocessor.py b/server/concordia_preprocessor.py index 38f6cf0..bb80bb1 100755 --- a/server/concordia_preprocessor.py +++ b/server/concordia_preprocessor.py @@ -10,11 +10,11 @@ api = Api(app) lemmatizer = spacy.load('en', disable=['parser', 'ner']) -app.logger.info("Lemmatizer initialized") +print("Lemmatizer initialized") pickle_in = open("dictionaries/dict.pickle","rb") dictionary = pickle.load(pickle_in) -app.logger.info("Dictionary initialized") +print("Dictionary initialized") diff --git a/server/lemmatizer_resource.py b/server/lemmatizer_resource.py index e074510..dc9a6d9 100644 --- a/server/lemmatizer_resource.py +++ b/server/lemmatizer_resource.py @@ -17,6 +17,9 @@ class LemmatizerResource(Resource): return {'error':'Missing parameter: sentences'}, 400 language = json_data['language'] + if language not in ['pl', 'en']: + return {'error':'Unsupported language: %s' % language}, 400 + sentences = [] for sentence in json_data['sentences']: @@ -34,7 +37,7 @@ class LemmatizerResource(Resource): if len(word) == 1: return word lemma = None - if language == 'pl': + if language == 'pl': lemma = self.dictionary.get(word, None) if lemma is None: lemma = word