selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-22 05:09:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

server: add tests on segmentation and fix issue with speaker

Browse Source
This commit is contained in:
Mathieu Virbel
2023-10-20 16:14:30 +02:00
committed by Mathieu Virbel
parent 01d7add6cc
commit f4cffc0e66
2 changed files with 34 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
server/reflector/processors/types.py
View File

@@ -125,16 +125,30 @@ class Transcript(BaseModel):
speaker=word.speaker,
)
continue
# If the word is attach to another speaker, push the current segment
# and start a new one
if word.speaker != current_segment.speaker:
segments.append(current_segment)
current_segment = TranscriptSegment(
text=word.text,
start=word.start,
speaker=word.speaker,
)
continue
# if the word is the end of a sentence, and we have enough content,
# add the word to the current segment and push it
current_segment.text += word.text
have_punc = PUNC_RE.search(word.text)
if word.speaker != current_segment.speaker or (
have_punc and (len(current_segment.text) > MAX_SEGMENT_LENGTH)
):
if have_punc and (len(current_segment.text) > MAX_SEGMENT_LENGTH):
segments.append(current_segment)
current_segment = None
if current_segment:
segments.append(current_segment)
return segments