LAP-DEV commited on
Commit
92dfb7f
·
verified ·
1 Parent(s): e7dd32b

Update modules/utils/subtitle_manager.py

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +4 -4
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};\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
 
 
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