diff --git a/main.py b/main.py index 5fc600d..c1da907 100644 --- a/main.py +++ b/main.py @@ -15,7 +15,12 @@ def clean_text(text: str): def extract_keywords(text: str, keywords_dict: dict): - return list + empty_list = [] + for k, v in keywords_dict.items(): + for i in v: + if i in text: + empty_list.append(k) + return (list(set(empty_list))) def calculate_sentiment_score(text: str, positive_words: set, negative_words: set): @@ -93,10 +98,10 @@ def main(): "золотое", "слишком дорого", "цена не соответствует качеству", "дороже чем у других", "наценка", "обдираловка", "цена выросла", "денег не стоит" ] - keywords_dict: { - "quality": [positive_words + negative_words], - "delivery": [delivery_words], - "price": [price_words] + keywords_dict = { + "quality": list(positive_words) + list(negative_words), + "delivery": delivery_words, + "price": price_words } print("\n======== Введите название файла с форматом .csv или используйте по умолчанию (Enter) ========") name_report = input()