Update modules/utils/subtitle_manager.py

modules/utils/subtitle_manager.py

@@ -33,10 +33,14 @@ def get_srt(segments):
     return output
 
 def get_csv(segments,diarization: bool = False):
-    output = "Line;Start time;End time;Text;\n"
+    if diarization:
+        output = "Line;Start time;End time;Speaker;Text;\n"
+    else:
+        output = "Line;Start time;End time;Text;\n"
+        
     for i, segment in enumerate(segments):
         if segment['text'].startswith(' '):
-            segment['text'] = segment['text'][1:]
+            segment['text'] = segment['text'][1:]            
         output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{segment['text']};\n"
     return output