Update app.py

app.py

@@ -129,20 +129,43 @@ def analyze_emotion(text):
         if isinstance(emotions, list) and len(emotions) > 0:
             emotion_scores = emotions[0]
             
-            # Define urgent emotions
-            urgent_emotions = ['anger', 'fear', 'annoyance', 'disapproval', 'nervousness']
-            high_priority_emotions = ['desire', 'excitement', 'surprise']
+            # Define emotion categories with their respective thresholds
+            urgent_emotions = {
+                'anger': 0.15,
+                'fear': 0.40,
+                'annoyance': 0.10,
+                'disapproval': 0.30,
+                'nervousness': 0.25,
+                'disgust': 0.20,
+                'disappointment': 0.40
+            }
             
-            # Calculate urgency based on emotional content
-            urgent_score = sum(score for emotion, score in emotion_scores.items() 
-                             if emotion in urgent_emotions)
-            high_priority_score = sum(score for emotion, score in emotion_scores.items() 
-                                    if emotion in high_priority_emotions)
+            high_priority_emotions = {
+                'desire': 0.25,
+                'excitement': 0.35,
+                'surprise': 0.15,
+                'curiosity': 0.25,
+                'optimism': 0.20,
+                'pride': 0.10
+            }
             
-            # Determine urgency level based on emotion scores
-            if urgent_score > 0.5:
+            # Calculate weighted urgency scores
+            urgent_score = 0
+            high_priority_score = 0
+            
+            for item in emotion_scores:
+                emotion = item['label']
+                score = item['score']
+                
+                if emotion in urgent_emotions and score > urgent_emotions[emotion]:
+                    urgent_score += score
+                elif emotion in high_priority_emotions and score > high_priority_emotions[emotion]:
+                    high_priority_score += score
+            
+            # Determine urgency level based on weighted scores
+            if urgent_score > 0.4:  # Adjusted threshold
                 return "urgent"
-            elif high_priority_score > 0.4 or urgent_score > 0.3:
+            elif high_priority_score > 0.3 or urgent_score > 0.2:  # Adjusted thresholds
                 return "high"
             return "normal"