Update app.py

app.py

@@ -87,9 +87,10 @@ ocr_reader = easyocr.Reader(["en"], gpu=False)
 # ——— 4) Emotional-Tone Tagging —————————————————————————————————————————————
 def get_emotional_tone_tag(emotion_profile, patterns, text_lower):
     """
-    Assigns one of 18 nuanced tone categories based on emotion scores, patterns, and text.
+    Assigns one of 18 nuanced tone categories based on
+    model scores, NRC-EmoLex counts, detected patterns, and text.
     """
-    # unpack all emotion scores before any rules
+    # unpack model emotion scores
     sadness  = emotion_profile.get("sadness",  0)
     joy      = emotion_profile.get("joy",      0)
     neutral  = emotion_profile.get("neutral",  0)
@@ -98,76 +99,96 @@ def get_emotional_tone_tag(emotion_profile, patterns, text_lower):
     fear     = emotion_profile.get("fear",     0)
     surprise = emotion_profile.get("surprise", 0)
 
-    # 0. Support override
-    if any(k in text_lower for k in ["support", "hope", "grace"]):
+    # count lexicon hits for the big five
+    words = text_lower.split()
+    lex_counts = {
+        emo: sum(EMOLEX.get(w, {}).get(emo, 0) for w in words)
+        for emo in ["anger","joy","sadness","fear","disgust"]
+    }
+
+    # 0. Support override: explicit keywords or detected joy in lexicon
+    if any(k in text_lower for k in ["support", "hope", "grace"]) or lex_counts["joy"] > 0:
         return "supportive"
 
     # 1. Performative Regret
-    if sadness > 0.4 and any(p in patterns for p in ["blame shifting", "guilt tripping", "recovery phase"]):
+    if (sadness > 0.4 or lex_counts["sadness"] > 1) \
+       and any(p in patterns for p in ["blame shifting","guilt tripping","recovery phase"]):
         return "performative regret"
 
     # 2. Coercive Warmth
-    if (joy > 0.3 or sadness > 0.4) and any(p in patterns for p in ["control", "gaslighting"]):
+    if ((joy > 0.3 or lex_counts["joy"] > 1) or (sadness > 0.4 or lex_counts["sadness"] > 1)) \
+       and any(p in patterns for p in ["control","gaslighting"]):
         return "coercive warmth"
 
     # 3. Cold Invalidation
-    if (neutral + disgust) > 0.5 and any(p in patterns for p in ["dismissiveness", "projection", "obscure language"]):
+    if ((neutral + disgust) > 0.5 or lex_counts["disgust"] > 1) \
+       and any(p in patterns for p in ["dismissiveness","projection","obscure language"]):
         return "cold invalidation"
 
     # 4. Genuine Vulnerability
-    if (sadness + fear) > 0.5 and all(p == "recovery phase" for p in patterns):
+    if ((sadness + fear) > 0.5 or (lex_counts["sadness"] + lex_counts["fear"]) > 1) \
+       and all(p == "recovery phase" for p in patterns):
         return "genuine vulnerability"
 
     # 5. Emotional Threat
-    if (anger + disgust) > 0.5 and any(p in patterns for p in ["control", "threat", "insults", "dismissiveness"]):
+    if ((anger + disgust) > 0.5 or (lex_counts["anger"] + lex_counts["disgust"]) > 1) \
+       and any(p in patterns for p in ["control","threat","insults","dismissiveness"]):
         return "emotional threat"
 
     # 6. Weaponized Sadness
-    if sadness > 0.6 and any(p in patterns for p in ["guilt tripping", "projection"]):
+    if (sadness > 0.6 or lex_counts["sadness"] > 2) \
+       and any(p in patterns for p in ["guilt tripping","projection"]):
         return "weaponized sadness"
 
     # 7. Toxic Resignation
-    if neutral > 0.5 and any(p in patterns for p in ["dismissiveness", "obscure language"]):
+    if (neutral > 0.5) and any(p in patterns for p in ["dismissiveness","obscure language"]):
         return "toxic resignation"
 
     # 8. Indignant Reproach
-    if anger > 0.5 and any(p in patterns for p in ["guilt tripping", "contradictory statements"]):
+    if (anger > 0.5 or lex_counts["anger"] > 1) \
+       and any(p in patterns for p in ["guilt tripping","contradictory statements"]):
         return "indignant reproach"
 
     # 9. Confrontational
-    if anger > 0.6 and patterns:
+    if (anger > 0.6 or lex_counts["anger"] > 1) and patterns:
         return "confrontational"
 
     # 10. Passive Aggression
-    if neutral > 0.6 and any(p in patterns for p in ["dismissiveness", "projection"]):
+    if (neutral > 0.6) and any(p in patterns for p in ["dismissiveness","projection"]):
         return "passive aggression"
 
     # 11. Sarcastic Mockery
-    if joy > 0.3 and "insults" in patterns:
+    if (joy > 0.3 or lex_counts["joy"] > 1) and "insults" in patterns:
         return "sarcastic mockery"
 
     # 12. Menacing Threat
-    if fear > 0.3 and "threat" in patterns:
+    if (fear > 0.3 or lex_counts["fear"] > 0) and "threat" in patterns:
         return "menacing threat"
 
     # 13. Pleading Concern
-    if sadness > 0.3 and any(k in text_lower for k in APOLOGY_KEYWORDS) and not patterns:
+    if (sadness > 0.3 or lex_counts["sadness"] > 0) \
+       and any(k in text_lower for k in APOLOGY_KEYWORDS) \
+       and not patterns:
         return "pleading concern"
 
     # 14. Fear-mongering
-    if (fear + disgust) > 0.5 and "projection" in patterns:
+    if ((fear + disgust) > 0.5 or (lex_counts["fear"] + lex_counts["disgust"]) > 1) \
+       and "projection" in patterns:
         return "fear-mongering"
 
     # 15. Disbelieving Accusation
-    if surprise > 0.3 and "blame shifting" in patterns:
+    if (surprise > 0.3 or lex_counts.get("surprise",0) > 0) \
+       and "blame shifting" in patterns:
         return "disbelieving accusation"
 
     # 16. Empathetic Solidarity
-    if joy > 0.2 and sadness > 0.2 and not patterns:
+    if (joy > 0.2 and sadness > 0.2) \
+       and (lex_counts["joy"] > 0 and lex_counts["sadness"] > 0) \
+       and not patterns:
         return "empathetic solidarity"
 
     # 17. Assertive Boundary
-    if anger > 0.4 and "control" in patterns:
+    if (anger > 0.4 or lex_counts["anger"] > 0) and "control" in patterns:
         return "assertive boundary"
 
     # 18. Stonewalling