Spaces:
Running
Running
File size: 2,544 Bytes
23a7dc1 f6b118b 23a7dc1 f6b118b 23a7dc1 f6b118b 23a7dc1 f6b118b 23a7dc1 f6b118b 23a7dc1 15f9ce8 f6b118b 15f9ce8 f6b118b 15f9ce8 62b81e7 f6b118b 7f1b106 23a7dc1 f6b118b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import gradio as gr
import os
from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
import ffmpeg # Make sure to install ffmpeg-python
def read_subtitle_file(subtitle_path):
with open(subtitle_path, 'r', encoding='utf-8') as file:
subtitle_content = file.read()
return os.path.basename(subtitle_path), subtitle_content
def add_subtitle_to_video(input_video, subtitle_file, subtitle_language, soft_subtitle):
video_input_stream = ffmpeg.input(input_video)
subtitle_input_stream = ffmpeg.input(subtitle_file)
input_video_name = os.path.splitext(os.path.basename(input_video))[0]
output_video = f"/tmp/output-{input_video_name}.mp4"
subtitle_track_title = os.path.splitext(os.path.basename(subtitle_file))[0]
if soft_subtitle:
stream = ffmpeg.output(
video_input_stream, subtitle_input_stream, output_video,
**{"c": "copy", "c:s": "mov_text"},
**{"metadata:s:s:0": f"language={subtitle_language}",
"metadata:s:s:0": f"title={subtitle_track_title}"}
)
else:
stream = ffmpeg.output(
video_input_stream, output_video,
vf=f"subtitles={subtitle_file}"
)
ffmpeg.run(stream, overwrite_output=True)
return output_video
def video_demo(video, subtitle, subtitle_type, subtitle_language):
if subtitle is not None:
soft_subtitle = subtitle_type == "Soft"
processed_video_path = add_subtitle_to_video(video, subtitle, subtitle_language, soft_subtitle)
return processed_video_path
else:
return video
# Interface improvements
description = """
Easily add subtitles to your videos with our tool. For web use, please visit [this space](https://huggingface.co/spaces/Lenylvt/SRT_to_Video).
**Subtitle Type Explained:**
- **Hard**: Subtitles are permanently embedded into the video. Ideal for ensuring compatibility across all players.
- **Soft**: Subtitles are added as a separate track. This allows viewers to toggle subtitles on or off as needed.
"""
demo = gr.Interface(
fn=video_demo,
title="Video Subtitler API",
description=description,
inputs=[
gr.Video(label="Video", interactive=True),
gr.File(label="Subtitle", file_types=[".srt", ".vtt"]),
gr.Radio(choices=["Hard", "Soft"], label="Subtitle Type"),
gr.Textbox(label="Subtitle Language (ISO 639-1 code, e.g., 'en' for English)"),
],
outputs=gr.Video(label="Processed Video"),
)
if __name__ == "__main__":
demo.launch()
|