diff --git a/main.py b/main.py index f3f0fd7..d9b209b 100644 --- a/main.py +++ b/main.py @@ -40,7 +40,7 @@ def validate_rating(rating: int): return res -def is_spam(text: str, spam_indicators: list): +def validate_rating(text: str, spam_indicators: list): res = False caps_count = 0 for i in spam_indicators: @@ -58,7 +58,15 @@ def is_spam(text: str, spam_indicators: list): def enrich_review(review: dict, keywords_dict: dict, positive_words: set, negative_words: set): - return dict + upgrade_review = { + "id": review.get("id"), + "rating": review.get("rating"), + "text": review.get("text"), + "clean_txt": clean_text(review.get("text")), + "sentiment": calculate_sentiment_score(review.get("text"), positive_words, negative_words), + "topics": extract_keywords(review.get("text"), keywords_dict) + } + return upgrade_review def filter_valid_reviews(reviews: list):