Update app.py

app.py

@@ -3,32 +3,33 @@ import moviepy.editor as mp
 from math import ceil
 
 def split_video(input_video, duration):
-video_clip = mp.VideoFileClip(input_video)
-video_duration = video_clip.duration
-num_segments = ceil(video_duration / duration)
+    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"
+    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"
 
-video_segment = video_clip.subclip(start_time, end_time)
-video_segment.write_videofile(output_video, codec='libx264', threads=4)
+        stream = mp.VideoFileClip(input_video)
+        stream = stream.subclip(start_time, end_time)
+        stream.write_videofile(output_video, codec='libx264', threads=4)
 
-segments.append(output_video)
+        segments.append(output_video)
 
-return segments
+    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. The segments will be named 'segment_01.mp4', 'segment_02.mp4', etc.",
+    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.",
 )
 
-interface.launch()
+interface.launch()