Khriis commited on
Commit
3a7a32b
verified
1 Parent(s): c875af6

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +7 -1
handler.py CHANGED
@@ -262,6 +262,12 @@ class EndpointHandler:
262
  return False
263
  if not any(ch.isalpha() for ch in a):
264
  return False
 
 
 
 
 
 
265
  return True
266
 
267
  raw_answers = [p.get("answer", "") for p in current_preds]
@@ -405,5 +411,5 @@ class EndpointHandler:
405
  break
406
  if best:
407
  return [_extract_from_text(target_text, best)]
408
- #
409
  return cleaned[:3]
 
262
  return False
263
  if not any(ch.isalpha() for ch in a):
264
  return False
265
+ # Filter out speaker labels and prompt artifacts
266
+ a_lower = a.lower()
267
+ if "patient:" in a_lower or "therapist:" in a_lower or "[target]" in a_lower:
268
+ return False
269
+ if a_lower in ["patient", "therapist"]:
270
+ return False
271
  return True
272
 
273
  raw_answers = [p.get("answer", "") for p in current_preds]
 
411
  break
412
  if best:
413
  return [_extract_from_text(target_text, best)]
414
+
415
  return cleaned[:3]