Update app.py

app.py

@@ -33,10 +33,13 @@ else:
 
 # --- 3. LOGIC ---
 def analyze(text):
+    # 1. Simple Check: Is the model variable loaded?
+    if model is None: 
+        return {}, "⚠️ Model failed to load. Check logs."
+
+    # 2. Check for Empty Text
     if not text or not text.strip():
         return {}, "⚠️ Please type something first!"
-    if model is None: 
-        return {}, "⚠️ Model files missing in this folder."
 
     inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
     model.eval()
@@ -47,10 +50,19 @@ def analyze(text):
     results = {}
     detected_emotions = []
     
+    # Penalties for "Sleep" bias
+    penalties = {"sleep": {"fear": 0.50}, "tired": {"fear": 0.40}, "hungry": {"fear": 0.40}}
+    text_lower = text.lower()
+
     for i, label in enumerate(LABELS):
         score = float(probs[i])
+        for word, penalty_map in penalties.items():
+            if word in text_lower and label in penalty_map:
+                score = max(0.0, score - penalty_map[label])
+        
         results[label] = score
-        if score > THRESHOLDS.get(label, 0.5):
+        # Use simple 0.5 threshold
+        if score > 0.5:
             detected_emotions.append(label.upper())
             
     if not detected_emotions: