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