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