Ashlee Kupor commited on
Commit
28763a3
·
1 Parent(s): ab4bb95

Sort by timing

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -272,7 +272,7 @@ class EndpointHandler():
272
  formatted = []
273
  for utterance in utterances:
274
  formatted.append({'speaker': utterance.speaker, 'data': utterance.text, 'time': utterance.starttime})
275
- return formatted
276
 
277
  def talk_moves_list_to_json(self, utterances: List[Utterance]) -> List[str]:
278
  formatted = []
@@ -282,7 +282,7 @@ class EndpointHandler():
282
  {'speaker': utterance.prev_prev_utterance.speaker, 'data': utterance.prev_prev_utterance.text, 'time': utterance.prev_prev_utterance.starttime},
283
  {'speaker': utterance.prev_utterance.speaker, 'data': utterance.prev_utterance.text, 'time': utterance.prev_utterance.starttime},
284
  {'speaker': utterance.speaker, 'data': utterance.text, 'time': utterance.starttime}]})
285
- return formatted
286
 
287
  def get_utterances_list(self, full_transcript, utterances_list, utterances_indexes, model_id):
288
  for i in range(len(full_transcript)):
 
272
  formatted = []
273
  for utterance in utterances:
274
  formatted.append({'speaker': utterance.speaker, 'data': utterance.text, 'time': utterance.starttime})
275
+ return sorted(formatted, key=lambda d: d['time'])
276
 
277
  def talk_moves_list_to_json(self, utterances: List[Utterance]) -> List[str]:
278
  formatted = []
 
282
  {'speaker': utterance.prev_prev_utterance.speaker, 'data': utterance.prev_prev_utterance.text, 'time': utterance.prev_prev_utterance.starttime},
283
  {'speaker': utterance.prev_utterance.speaker, 'data': utterance.prev_utterance.text, 'time': utterance.prev_utterance.starttime},
284
  {'speaker': utterance.speaker, 'data': utterance.text, 'time': utterance.starttime}]})
285
+ return sorted(formatted, key=lambda d: d['timing'])
286
 
287
  def get_utterances_list(self, full_transcript, utterances_list, utterances_indexes, model_id):
288
  for i in range(len(full_transcript)):