LAP-DEV commited on
Commit
8e1fa57
·
verified ·
1 Parent(s): e5689dc

Update modules/utils/subtitle_manager.py

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +5 -3
modules/utils/subtitle_manager.py CHANGED
@@ -53,9 +53,11 @@ def get_csv(segments):
53
  temp = re.search(r'_[0-9][0-9]: ',segment['text'])
54
 
55
  if temp != None:
56
- temp_string = str(temp.group())
57
- speaker_id = temp_string.replace(': ','')
58
- speaker_text = (segment['text']).replace(temp_string,'')
 
 
59
  output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
60
  bDiarization = True
61
  else:
 
53
  temp = re.search(r'_[0-9][0-9]: ',segment['text'])
54
 
55
  if temp != None:
56
+ #temp_string = str(temp.group())
57
+ #speaker_id = temp_string.replace(': ','')
58
+ #speaker_text = (segment['text']).replace(temp_string,'')
59
+ speaker_id = ((segment['text']).split(":", 1)[0]).strip()
60
+ speaker_text = ((segment['text']).split(":", 1)[1]).strip()
61
  output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
62
  bDiarization = True
63
  else: