Commit
·
4a21148
1
Parent(s):
c749ace
define temporary audio file path to fix write permission issue in space (2)
Browse files- utils/process_video.py +3 -2
- utils/subtitler.py +2 -2
utils/process_video.py
CHANGED
|
@@ -10,11 +10,12 @@ def process_video(invideo_file: str,
|
|
| 10 |
text_color:str,
|
| 11 |
highlight_mode: bool,
|
| 12 |
highlight_color: str,
|
| 13 |
-
caption_mode:str
|
|
|
|
| 14 |
):
|
| 15 |
invideo_path_parts = os.path.normpath(invideo_file).split(os.path.sep)
|
| 16 |
VIDEO_NAME = os.path.basename(invideo_file)
|
| 17 |
OUTVIDEO_PATH = os.path.join(os.path.normpath('/'.join(invideo_path_parts[:-1])), f"result_{VIDEO_NAME}")
|
| 18 |
logging.info("Subtitling...")
|
| 19 |
-
subtitler(invideo_file, srt_string, srt_json, OUTVIDEO_PATH, fontsize, font, bg_color, text_color, highlight_mode, highlight_color, caption_mode)
|
| 20 |
return OUTVIDEO_PATH
|
|
|
|
| 10 |
text_color:str,
|
| 11 |
highlight_mode: bool,
|
| 12 |
highlight_color: str,
|
| 13 |
+
caption_mode:str,
|
| 14 |
+
temp_dir: str
|
| 15 |
):
|
| 16 |
invideo_path_parts = os.path.normpath(invideo_file).split(os.path.sep)
|
| 17 |
VIDEO_NAME = os.path.basename(invideo_file)
|
| 18 |
OUTVIDEO_PATH = os.path.join(os.path.normpath('/'.join(invideo_path_parts[:-1])), f"result_{VIDEO_NAME}")
|
| 19 |
logging.info("Subtitling...")
|
| 20 |
+
subtitler(invideo_file, srt_string, srt_json, OUTVIDEO_PATH, fontsize, font, bg_color, text_color, highlight_mode, highlight_color, caption_mode, temp_dir)
|
| 21 |
return OUTVIDEO_PATH
|
utils/subtitler.py
CHANGED
|
@@ -85,7 +85,7 @@ def subtitler(video_file: str,
|
|
| 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')
|
| 89 |
return
|
| 90 |
# Normal mode
|
| 91 |
subtitles = parse_srt(srt_string)
|
|
@@ -105,4 +105,4 @@ def subtitler(video_file: str,
|
|
| 105 |
subtitle_clips.append(txt_clip)
|
| 106 |
video = CompositeVideoClip(size=None, clips=[clip] + subtitle_clips)
|
| 107 |
video.set_audio(temp_audiofile)
|
| 108 |
-
video.write_videofile(output_file, codec='libx264', audio_codec='aac')
|
|
|
|
| 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)
|
| 89 |
return
|
| 90 |
# Normal mode
|
| 91 |
subtitles = parse_srt(srt_string)
|
|
|
|
| 105 |
subtitle_clips.append(txt_clip)
|
| 106 |
video = CompositeVideoClip(size=None, clips=[clip] + subtitle_clips)
|
| 107 |
video.set_audio(temp_audiofile)
|
| 108 |
+
video.write_videofile(output_file, codec='libx264', audio_codec='aac', temp_audiofile = temp_audiofile)
|