Update app.py

app.py

@@ -1,36 +1,34 @@
 import gradio as gr
-import ffmpeg
+import moviepy.editor as mp
 from math import ceil
 
 def split_video(input_video, duration):
-    video_info = ffmpeg.probe(input_video)
-    video_duration = float(video_info['format']['duration'])
-    num_segments = ceil(video_duration / duration)
-    
-    segments = []
-    for i in range(num_segments):
-        start_time = i * duration
-        end_time = min((i + 1) * duration, video_duration)
-        output_video = f"segment_{i+1}.mp4"
-        
-        stream = ffmpeg.input(input_video)
-        stream = ffmpeg.trim(stream, start=start_time, end=end_time)
-        stream = ffmpeg.output(stream, output_video)
-        ffmpeg.run(stream, overwrite_output=True)
-        
-        segments.append(output_video)
-    
-    return segments
+video_clip = mp.VideoFileClip(input_video)
+video_duration = video_clip.duration
+num_segments = ceil(video_duration / duration)
+
+segments = []
+for i in range(num_segments):
+start_time = i * duration
+end_time = min((i + 1) * duration, video_duration)
+output_video = f"segment_{(i+1).zfill(2)}.mp4"
+
+video_segment = video_clip.subclip(start_time, end_time)
+video_segment.write_videofile(output_video, codec='libx264', threads=4)
+
+segments.append(output_video)
+
+return segments
 
 interface = gr.Interface(
-    split_video,
-    inputs=[
-        gr.File(label="Upload Video"),
-        gr.Number(label="Segment Duration (seconds)"),
-    ],
-    outputs=gr.Files(label="Video Segments"),
-    title="Video Splitter",
-    description="Split a video into equal-duration segments.",
+split_video,
+inputs=[
+gr.File(label="Upload Video"),
+gr.Number(label="Segment Duration (seconds)"),
+],
+outputs=gr.Files(label="Video Segments"),
+title="Video Splitter",
+description="Split a video into equal-duration segments. The segments will be named 'segment_01.mp4', 'segment_02.mp4', etc.",
 )
 
 interface.launch()