Subtitled Video

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import whisper
 import os
+from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
 
 # Load the Whisper model
 model = whisper.load_model("base")  # Choose 'tiny', 'base', 'small', 'medium', or 'large'
@@ -28,16 +29,51 @@ def transcribe_video(video_file):
     
     # Write the transcription as subtitles
     write_srt(result, srt_file)
+
+    # Create subtitled video
+    subtitled_video_file = create_subtitled_video(video_file, srt_file)
+
+    return subtitled_video_file
+
+def create_subtitled_video(video_file, srt_file):
+    # Load video
+    video_clip = VideoFileClip(video_file)
+
+    # Read SRT file and create text clips
+    subs = []
+    with open(srt_file, 'r') as f:
+        lines = f.readlines()
+        for i in range(0, len(lines), 4):
+            if i + 2 < len(lines):
+                start_time = convert_to_seconds(lines[i + 1].split(" --> ")[0])
+                end_time = convert_to_seconds(lines[i + 1].split(" --> ")[1])
+                text = lines[i + 2].strip()
+                text_clip = TextClip(text, fontsize=24, color='white', bg_color='black', size=video_clip.size, print_cmd=True)
+                text_clip = text_clip.set_start(start_time).set_end(end_time).set_position('bottom')
+                subs.append(text_clip)
+
+    # Combine video and subtitles
+    final_video = CompositeVideoClip([video_clip] + subs)
     
-    return srt_file
+    # Save the final video with subtitles
+    subtitled_video_file = "subtitled_video.mp4"
+    final_video.write_videofile(subtitled_video_file, codec="libx264", audio_codec="aac")
+
+    return subtitled_video_file
+
+def convert_to_seconds(timestamp):
+    """Convert timestamp in SRT format (HH:MM:SS,mmm) to seconds."""
+    h, m, s = timestamp.split(':')
+    seconds = int(h) * 3600 + int(m) * 60 + float(s.replace(',', '.'))
+    return seconds
 
 # Gradio interface
 iface = gr.Interface(
     fn=transcribe_video,
     inputs=gr.File(label="Upload Video"),
-    outputs=gr.File(label="Download Subtitles"),
+    outputs=gr.File(label="Download Subtitled Video"),
     title="Video Subtitle Generator",
-    description="Upload a video file to generate subtitles using Whisper."
+    description="Upload a video file to generate subtitles and download a subtitled video."
 )
 
 if __name__ == "__main__":