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:
|