LAP-DEV commited on
Commit
d0fe6ef
·
verified ·
1 Parent(s): ebfa7b3

Update modules/utils/subtitle_manager.py

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +2 -1
modules/utils/subtitle_manager.py CHANGED
@@ -58,7 +58,8 @@ def get_csv(segments):
58
  output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
59
  bDiarization = True
60
  else:
61
- output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{segment['text']};\n"
 
62
 
63
  # Add titles to csv file
64
  if bDiarization:
 
58
  output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
59
  bDiarization = True
60
  else:
61
+ speaker_text = (segment['text']).strip()
62
+ output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_text};\n"
63
 
64
  # Add titles to csv file
65
  if bDiarization: