Update modules/utils/subtitle_manager.py
Browse files
modules/utils/subtitle_manager.py
CHANGED
|
@@ -51,11 +51,11 @@ def get_csv(segments):
|
|
| 51 |
else:
|
| 52 |
output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{segment['text']};\n"
|
| 53 |
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
|
| 60 |
return output
|
| 61 |
|
|
|
|
| 51 |
else:
|
| 52 |
output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{segment['text']};\n"
|
| 53 |
|
| 54 |
+
# Add titles to csv file
|
| 55 |
+
if bDiarization:
|
| 56 |
+
output = "Line;Start time;End time;Speaker;Text;\n" + output
|
| 57 |
+
else:
|
| 58 |
+
output = "Line;Start time;End time;Text;\n" + output
|
| 59 |
|
| 60 |
return output
|
| 61 |
|