Update edite_video.py

edite_video.py

@@ -27,7 +27,7 @@ def create_subtitle_clips(subtitles, videosize, fontsize, font, color, debug):
         color_clips.append(myclip.with_position(text_position))
     return subtitle_clips, color_clips
 
-def video_edit(srt, input_video, input_audio= "audio.mp3"):
+def video_edit(srt, input_video, color, font, input_audio= "audio.mp3"):
     print(input_video)
     input_video_name = input_video.split(".mp4")[0]
     video = VideoFileClip(input_video)
@@ -36,7 +36,7 @@ def video_edit(srt, input_video, input_audio= "audio.mp3"):
     print(video)
     output_video_file = input_video_name + '_subtitled' + ".mp4"
     subtitles = pysrt.open(srt, encoding="utf-8")
-    subtitle_clips, color_clips = create_subtitle_clips(subtitles, video.size, 24, 'arial.ttf', 'white', False)
+    subtitle_clips, color_clips = create_subtitle_clips(subtitles, video.size, 24, f'{font}.ttf', color, False)
     final_video = CompositeVideoClip([video]+color_clips + subtitle_clips)
     final_video.write_videofile(output_video_file, codec="libx264", audio_codec="aac", logger=None, preset = "faster", fps=24)
     print('final')