Update modules/utils/subtitle_manager.py
Browse files
modules/utils/subtitle_manager.py
CHANGED
|
@@ -55,16 +55,16 @@ def get_csv(segments):
|
|
| 55 |
if bDiarization:
|
| 56 |
speaker_id = ((segment['text']).split(":", 1)[0]).strip()
|
| 57 |
speaker_text = ((segment['text']).split(":", 1)[1]).strip()
|
| 58 |
-
output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text}
|
| 59 |
else:
|
| 60 |
speaker_text = (segment['text']).strip()
|
| 61 |
-
output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_text}
|
| 62 |
|
| 63 |
# Add titles to csv file
|
| 64 |
if bDiarization:
|
| 65 |
-
output = "Line;Start time;End time;Speaker;Text
|
| 66 |
else:
|
| 67 |
-
output = "Line;Start time;End time;Text
|
| 68 |
|
| 69 |
return output.rstrip("\n")
|
| 70 |
|
|
|
|
| 55 |
if bDiarization:
|
| 56 |
speaker_id = ((segment['text']).split(":", 1)[0]).strip()
|
| 57 |
speaker_text = ((segment['text']).split(":", 1)[1]).strip()
|
| 58 |
+
output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text}\n"
|
| 59 |
else:
|
| 60 |
speaker_text = (segment['text']).strip()
|
| 61 |
+
output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_text}\n"
|
| 62 |
|
| 63 |
# Add titles to csv file
|
| 64 |
if bDiarization:
|
| 65 |
+
output = "Line;Start time;End time;Speaker;Text\n" + output
|
| 66 |
else:
|
| 67 |
+
output = "Line;Start time;End time;Text\n" + output
|
| 68 |
|
| 69 |
return output.rstrip("\n")
|
| 70 |
|