Update app.py
Browse files
app.py
CHANGED
|
@@ -127,7 +127,7 @@ def analyze_emotion(text):
|
|
| 127 |
|
| 128 |
# Extract emotion scores
|
| 129 |
if isinstance(emotions, list) and len(emotions) > 0:
|
| 130 |
-
emotion_scores = emotions[0]
|
| 131 |
|
| 132 |
# Define emotion categories with their respective thresholds
|
| 133 |
urgent_emotions = {
|
|
@@ -137,7 +137,10 @@ def analyze_emotion(text):
|
|
| 137 |
'disapproval': 0.30,
|
| 138 |
'nervousness': 0.25,
|
| 139 |
'disgust': 0.20,
|
| 140 |
-
'disappointment': 0.40
|
|
|
|
|
|
|
|
|
|
| 141 |
}
|
| 142 |
|
| 143 |
high_priority_emotions = {
|
|
@@ -146,7 +149,11 @@ def analyze_emotion(text):
|
|
| 146 |
'surprise': 0.15,
|
| 147 |
'curiosity': 0.25,
|
| 148 |
'optimism': 0.20,
|
| 149 |
-
'pride': 0.10
|
|
|
|
|
|
|
|
|
|
|
|
|
| 150 |
}
|
| 151 |
|
| 152 |
# Calculate weighted urgency scores
|
|
|
|
| 127 |
|
| 128 |
# Extract emotion scores
|
| 129 |
if isinstance(emotions, list) and len(emotions) > 0:
|
| 130 |
+
emotion_scores = [item for item in emotions[0] if item['label'] != 'neutral']
|
| 131 |
|
| 132 |
# Define emotion categories with their respective thresholds
|
| 133 |
urgent_emotions = {
|
|
|
|
| 137 |
'disapproval': 0.30,
|
| 138 |
'nervousness': 0.25,
|
| 139 |
'disgust': 0.20,
|
| 140 |
+
'disappointment': 0.40,
|
| 141 |
+
'grief': 0.05, # Added more emotional states
|
| 142 |
+
'remorse': 0.10,
|
| 143 |
+
'sadness': 0.40
|
| 144 |
}
|
| 145 |
|
| 146 |
high_priority_emotions = {
|
|
|
|
| 149 |
'surprise': 0.15,
|
| 150 |
'curiosity': 0.25,
|
| 151 |
'optimism': 0.20,
|
| 152 |
+
'pride': 0.10,
|
| 153 |
+
'joy': 0.40, # Added more emotional states
|
| 154 |
+
'love': 0.25,
|
| 155 |
+
'admiration': 0.25,
|
| 156 |
+
'gratitude': 0.45
|
| 157 |
}
|
| 158 |
|
| 159 |
# Calculate weighted urgency scores
|