Update sentiment_analysis.py

sentiment_analysis.py

@@ -1,14 +1,17 @@
-from transformers import pipeline
-
-# Load Pre-trained Sentiment Model
-sentiment_analyzer = pipeline("sentiment-analysis")
-
-def analyze_sentiment(text):
-    result = sentiment_analyzer(text)
-    return result[0]
-
-# Test
-if __name__ == "__main__":
-    text = "I am really happy with the service!"
-    sentiment = analyze_sentiment(text)
-    print("Sentiment:", sentiment)
+from transformers import pipeline
+
+class SentimentAnalyzer:
+    def __init__(self):
+        self.analyzer = pipeline(
+            "sentiment-analysis",
+            model="distilbert-base-uncased-finetuned-sst-2-english",
+            return_all_scores=True
+        )
+    
+    def analyze(self, text):
+        try:
+            results = self.analyzer(text)
+            return results[0]
+        except Exception as e:
+            print(f"Error in sentiment analysis: {str(e)}")
+            return [{"label": "ERROR", "score": 0.0}]