marquesafonso commited on
Commit
d8510c0
·
1 Parent(s): 4a21148

improve wordlevel subtitle padding

Browse files
Files changed (1) hide show
  1. utils/subtitler.py +2 -3
utils/subtitler.py CHANGED
@@ -70,7 +70,7 @@ def subtitler(video_file: str,
70
  # Calculate word-level highlight positions
71
  current_x = x_center
72
  for word_info in line["words"]:
73
- word = word_info["word"]
74
  word_start = float(word_info["start"])
75
  word_end = float(word_info["end"])
76
 
@@ -80,9 +80,8 @@ def subtitler(video_file: str,
80
  word_clip = word_clip.set_start(word_start).set_end(word_end)
81
  word_clip = word_clip.set_position((current_x - 7.5, subtitle_y_position))
82
  subtitle_clips.append(word_clip)
 
83
 
84
- space_width = TextClip(" ", fontsize=fontsize, font=font, method='label').w
85
- current_x += word_clip.w + space_width
86
  video = CompositeVideoClip(size=None, clips=[clip] + subtitle_clips)
87
  video.set_audio(temp_audiofile)
88
  video.write_videofile(output_file, codec='libx264', audio_codec='aac', temp_audiofile = temp_audiofile)
 
70
  # Calculate word-level highlight positions
71
  current_x = x_center
72
  for word_info in line["words"]:
73
+ word = word_info["word"] + " "
74
  word_start = float(word_info["start"])
75
  word_end = float(word_info["end"])
76
 
 
80
  word_clip = word_clip.set_start(word_start).set_end(word_end)
81
  word_clip = word_clip.set_position((current_x - 7.5, subtitle_y_position))
82
  subtitle_clips.append(word_clip)
83
+ current_x += word_clip.w
84
 
 
 
85
  video = CompositeVideoClip(size=None, clips=[clip] + subtitle_clips)
86
  video.set_audio(temp_audiofile)
87
  video.write_videofile(output_file, codec='libx264', audio_codec='aac', temp_audiofile = temp_audiofile)