From c74cf2f95a67a07a44d00a77ac2023cb48f82dbc Mon Sep 17 00:00:00 2001 From: Rohit Date: Tue, 15 Oct 2024 10:51:15 -0700 Subject: [PATCH] Add sentiment class --- utils/sentiment.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 utils/sentiment.py diff --git a/utils/sentiment.py b/utils/sentiment.py new file mode 100644 index 0000000000..746420e1e1 --- /dev/null +++ b/utils/sentiment.py @@ -0,0 +1,18 @@ +from textblob import TextBlob + +def detect_sentiment(text): + sentences = text.split('. ') + sentiment_analysis = [] + + for sentence in sentences: + blob = TextBlob(sentence) + polarity = blob.sentiment.polarity + if polarity > 0: + sentiment = 'Positive' + elif polarity < 0: + sentiment = 'Negative' + else: + sentiment = 'Neutral' + sentiment_analysis.append((sentence, sentiment)) + + return sentiment_analysis