Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ def create_subtitle_clips(subtitles, videosize, fontsize, font, color, debug):
|
|
| 28 |
#color_clips.append(myclip.with_position(text_position))
|
| 29 |
return subtitle_clips
|
| 30 |
|
| 31 |
-
def video_edit(srt, input_video, color, font, input_audio):
|
| 32 |
print(input_video)
|
| 33 |
input_video_name = input_video.split(".mp4")[0]
|
| 34 |
video = VideoFileClip(input_video)
|
|
@@ -37,7 +37,7 @@ def video_edit(srt, input_video, color, font, input_audio):
|
|
| 37 |
print(video)
|
| 38 |
output_video_file = input_video_name + '_subtitled' + ".mp4"
|
| 39 |
subtitles = pysrt.open(srt, encoding="utf-8")
|
| 40 |
-
subtitle_clips = create_subtitle_clips(subtitles, video.size,
|
| 41 |
final_video = CompositeVideoClip([video]+ subtitle_clips)
|
| 42 |
final_video.write_videofile(output_video_file, codec="libx264", audio_codec="aac", logger=None, preset = "faster", fps=24)
|
| 43 |
print('final')
|
|
@@ -57,7 +57,7 @@ with gr.Blocks() as demo:
|
|
| 57 |
output_video = gr.Video()
|
| 58 |
btn.click(
|
| 59 |
fn=video_edit,
|
| 60 |
-
inputs=[srt_file, video_in, color, font, audio_in],
|
| 61 |
outputs=output_video
|
| 62 |
)
|
| 63 |
|
|
|
|
| 28 |
#color_clips.append(myclip.with_position(text_position))
|
| 29 |
return subtitle_clips
|
| 30 |
|
| 31 |
+
def video_edit(srt, input_video, color, font, font_size, input_audio):
|
| 32 |
print(input_video)
|
| 33 |
input_video_name = input_video.split(".mp4")[0]
|
| 34 |
video = VideoFileClip(input_video)
|
|
|
|
| 37 |
print(video)
|
| 38 |
output_video_file = input_video_name + '_subtitled' + ".mp4"
|
| 39 |
subtitles = pysrt.open(srt, encoding="utf-8")
|
| 40 |
+
subtitle_clips = create_subtitle_clips(subtitles, video.size, int(font_size), f'{font}.ttf', color, False)
|
| 41 |
final_video = CompositeVideoClip([video]+ subtitle_clips)
|
| 42 |
final_video.write_videofile(output_video_file, codec="libx264", audio_codec="aac", logger=None, preset = "faster", fps=24)
|
| 43 |
print('final')
|
|
|
|
| 57 |
output_video = gr.Video()
|
| 58 |
btn.click(
|
| 59 |
fn=video_edit,
|
| 60 |
+
inputs=[srt_file, video_in, color, font, font_size, audio_in],
|
| 61 |
outputs=output_video
|
| 62 |
)
|
| 63 |
|