Update app.py

app.py

@@ -1,36 +1,30 @@
+import os
+import moviepy.editor as mp
 import gradio as gr
-import ffmpeg
-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
+def split_video(video_file, parts):
+    video = mp.VideoFileClip(video_file.name)
+    duration = video.duration
+    part_duration = duration / parts
 
-interface = gr.Interface(
-    split_video,
-    inputs=[
-        gr.File(label="Upload Video"),
-        gr.Number(label="Segment Duration (seconds)"),
-    ],
-    outputs=gr.Files(label="Video Segments"),
+    output_files = []
+    for part in range(parts):
+        start_time = part * part_duration
+        end_time = (part + 1) * part_duration
+        part_clip = video.subclip(start_time, end_time)
+        output_filename = f"part_{part + 1}.mp4"
+        part_clip.write_videofile(output_filename, codec="libx264", audio_codec="aac")
+        output_files.append(output_filename)
+
+    return output_files
+
+iface = gr.Interface(
+    fn=split_video,
+    inputs=[gr.inputs.File(type="file", label="Upload Video"), gr.inputs.Slider(minimum=2, maximum=10, default=2, label="Number of Parts")],
+    outputs=gr.outputs.Files(label="Download Split Videos"),
     title="Video Splitter",
-    description="Split a video into equal-duration segments.",
+    description="Upload your video and select how many parts you want to split it into."
 )
 
-interface.launch()
+if __name__ == "__main__":
+    iface.launch()