Upload 2 files

src/models/fusion_engine.py

@@ -64,10 +64,16 @@ class LateDecisionFusion:
         
         # Determine primary stress origin
         t_cat = text_result.get("predicted_category", "Normal")
+        a_cat = audio_result.get("predicted_emotion", "Neutral")
         if combined_stress_score < 40.0:
             final_category = "Normal"
         else:
-            final_category = t_cat if t_cat != "Normal" else "Stress"
+            if t_cat != "Normal":
+                final_category = t_cat
+            elif a_cat not in ["Normal", "Neutral", "Stress"]:
+                final_category = a_cat
+            else:
+                final_category = "Stress"
             
         # Determine clinical risk tier
         if combined_stress_score < 40.0:
@@ -141,14 +147,16 @@ class LateDecisionFusion:
         else:
             cat = "Calm / Normal"
             
+        emotion_str = emotion if emotion not in ["Neutral", "Normal", "Stress"] else "Stress"
+        
         if score < 40.0:
             tier, color, summary = "Minimal / Normal", "green", "Vocal tone is calm and stable."
         elif score < 60.0:
-            tier, color, summary = "Mild Stress", "blue", f"Slight vocal tension ({emotion.lower()}) observed."
+            tier, color, summary = f"Mild {emotion_str}", "blue", f"Slight vocal tension ({emotion.lower()}) observed."
         elif score < 80.0:
-            tier, color, summary = "Moderate Stress", "orange", f"High vocal agitation ({emotion.lower()}) and spectral energy detected."
+            tier, color, summary = f"Moderate {emotion_str}", "orange", f"High vocal agitation ({emotion.lower()}) and spectral energy detected."
         else:
-            tier, color, summary = "Severe Stress", "red", f"Severe acoustic stress and emotional agitation ({emotion.lower()}) recorded."
+            tier, color, summary = f"Severe {emotion_str}", "red", f"Severe acoustic stress and emotional agitation ({emotion.lower()}) recorded."
             
         return {
             "modality_status": "Single-Modality (Speech Only)",