Spaces:
Sleeping
Sleeping
File size: 1,884 Bytes
6d6b8af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
"""
This module provides enhanced sentiment analysis capabilities.
"""
from typing import Dict, Any
try:
from transformers import pipeline
except Exception:
pipeline = None
class EnhancedSentimentAnalyzer:
"""Advanced sentiment analysis with additional techniques"""
def __init__(self):
if pipeline is not None:
try:
self.sentiment_pipeline = pipeline('sentiment-analysis')
except Exception:
self.sentiment_pipeline = None
else:
self.sentiment_pipeline = None
def analyze(self, text: str) -> Dict[str, Any]:
"""Analyze sentiment with advanced techniques"""
if self.sentiment_pipeline is None:
# Fallback simple heuristic
score = 0.5
label = 'NEUTRAL'
if any(w in text.lower() for w in ['good', 'great', 'excellent', 'happy']):
score = 0.9
label = 'POSITIVE'
elif any(w in text.lower() for w in ['bad', 'sad', 'terrible', 'awful']):
score = 0.1
label = 'NEGATIVE'
return {'label': label, 'score': score}
analysis = self.sentiment_pipeline(text)
return analysis[0]
def detailed_analysis(self, text: str) -> Dict[str, Any]:
"""Provide a more detailed sentiment analysis"""
if self.sentiment_pipeline is None:
base = self.analyze(text)
scores = base
else:
scores = self.sentiment_pipeline(text)[0]
if scores['label'] == 'POSITIVE':
sentiment = "Positive"
elif scores['label'] == 'NEGATIVE':
sentiment = "Negative"
else:
sentiment = "Neutral"
return {
"scores": scores,
"sentiment": sentiment
} |