LAP-DEV commited on
Commit
a00c871
·
verified ·
1 Parent(s): cef3668

Update modules/utils/subtitle_manager.py

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +2 -2
modules/utils/subtitle_manager.py CHANGED
@@ -43,8 +43,8 @@ def get_csv(segments):
43
  # Check if speakers are identified and get speaker id & text
44
  temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
45
  if temp != None:
46
- speaker_id = (temp.group()).replace(': ','')
47
- speaker_text = temp.replace(temp.group(),'')
48
  output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
49
  bDiarization = True
50
  else:
 
43
  # Check if speakers are identified and get speaker id & text
44
  temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
45
  if temp != None:
46
+ speaker_id = str(temp.group()).replace(': ','')
47
+ speaker_text = temp.replace(str(temp.group()),'')
48
  output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
49
  bDiarization = True
50
  else: