LAP-DEV commited on
Commit
824d2a1
·
verified ·
1 Parent(s): c2f69b0

Update modules/utils/subtitle_manager.py

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +7 -0
modules/utils/subtitle_manager.py CHANGED
@@ -32,6 +32,13 @@ def get_srt(segments):
32
  output += f"{segment['text']}\n\n"
33
  return output
34
 
 
 
 
 
 
 
 
35
 
36
  def get_vtt(segments):
37
  output = "WebVTT\n\n"
 
32
  output += f"{segment['text']}\n\n"
33
  return output
34
 
35
+ def get_csv(segments):
36
+ output = "Line;Start time;End time;Text"
37
+ for i, segment in enumerate(segments):
38
+ if segment['text'].startswith(' '):
39
+ segment['text'] = segment['text'][1:]
40
+ output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{segment['text']};\n"
41
+ return output
42
 
43
  def get_vtt(segments):
44
  output = "WebVTT\n\n"