Upload classifier.py

classifier.py

@@ -0,0 +1,24 @@
+from textblob import TextBlob
+import pandas as pd
+
+def classify_comment(comment):
+    text = comment.lower()
+    if '?' in text:
+        return 'question'
+    elif any(word in text for word in ['bad', 'worst', 'hate', 'not good', 'terrible', 'disagree']):
+        return 'criticism'
+    elif any(word in text for word in ['good', 'love', 'nice', 'amazing', 'agree', 'great', 'awesome']):
+        return 'affirmative'
+    else:
+        polarity = TextBlob(comment).sentiment.polarity
+        if polarity > 0.2:
+            return 'affirmative'
+        elif polarity < -0.2:
+            return 'criticism'
+        else:
+            return 'neutral'
+
+def classify_comments(comments):
+    df = pd.DataFrame(comments)
+    df["category"] = df["text"].astype(str).apply(classify_comment)
+    return df