Update modules/utils/subtitle_manager.py
Browse files
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:
|