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

Update modules/utils/subtitle_manager.py

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +3 -2
modules/utils/subtitle_manager.py CHANGED
@@ -43,8 +43,9 @@ 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 = 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:
 
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
+ temp_string = str(temp.group())
47
+ speaker_id = temp_string.replace(': ','')
48
+ speaker_text = temp_string.replace(temp_string,'')
49
  output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
50
  bDiarization = True
51
  else: