Update handler.py

handler.py

@@ -140,6 +140,7 @@ class EndpointHandler:
                     # Get prediction for this item
                     # Because top_k=20, 'current_preds' is a list of dicts: [{'answer': '...', 'score': ...}, ...]
                     current_preds = predictions[pred_idx]
+
                     
                     # Ensure it is a list
                     if isinstance(current_preds, dict):
@@ -149,6 +150,19 @@ class EndpointHandler:
                         "RECCON raw spans (answer, score): %s",
                         [(p.get("answer"), p.get("score", 0.0), 3) for p in current_preds[:5]]
                     )
+
+                    best_score = max((p.get("score", 0.0) for p in current_preds), default=0.0)
+
+                    MIN_BEST_SCORE = 0.01  # start here; tune later
+                    if best_score < MIN_BEST_SCORE:
+                        results.append({
+                            "utterance": utterance,
+                            "emotion": emotion,
+                            "triggers": [],  # no highlight
+                            "metadata": {"low_confidence": True, "best_score": best_score}
+                        })
+                        pred_idx += 1
+                        continue
                     
                     # Extract the answer strings
                     raw_answers = [p.get('answer', '') for p in current_preds]