Update app.py

app.py

@@ -1,53 +1,40 @@
 import gradio as gr
+import ffmpeg
+from math import ceil
 import moviepy.editor as mp
-from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 
-def split_video(video_file, split_method, split_value):
-    video = mp.VideoFileClip(video_file.name)
-    duration = video.duration
+def split_video(input_video, duration):
+    if not input_video:
+        raise ValueError("No video file provided.")
 
-    if split_method == "Free Split":
-        split_times = [float(t) for t in split_value.split(",")]
-        split_times = [t for t in split_times if t >= 0 and t <= duration]
-        split_times.sort()
-        split_times = [0] + split_times + [duration]
-        clips = [video.subclip(start, end) for start, end in zip(split_times[:-1], split_times[1:])]
-    elif split_method == "Average Split":
-        num_parts = int(split_value)
-        part_duration = duration / num_parts
-        clips = [video.subclip(i * part_duration, (i + 1) * part_duration) for i in range(num_parts)]
-    elif split_method == "Time Split":
-        split_times = [float(t) for t in split_value.split(",")]
-        clips = [ffmpeg_extract_subclip(video_file.name, start, end) for start, end in zip([0] + split_times, split_times + [duration])]
-    elif split_method == "Size Split":
-        target_size = int(split_value) * 1024 * 1024  # Convert from MB to bytes
-        clips = []
-        start = 0
-        while start < duration:
-            end = start + 1
-            while end <= duration and video.subclip(start, end).size < target_size:
-                end += 1
-            clips.append(video.subclip(start, end))
-            start = end
+    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
 
-    output_files = []
-    for i, clip in enumerate(clips):
-        output_filename = f"split_{i + 1}.mp4"
-        clip.write_videofile(output_filename)
-        output_files.append(output_filename)
-
-    return output_files
-
-iface = gr.Interface(
-    fn=split_video,
+interface = gr.Interface(
+    split_video,
     inputs=[
         gr.File(label="Upload Video"),
-        gr.Dropdown(choices=["Free Split", "Average Split", "Time Split", "Size Split"], label="Split Method"),
-        gr.Textbox(label="Split Value (comma-separated for Free Split and Time Split, number for Average Split and Size Split)"),
+        gr.Number(label="Segment Duration (seconds)"),
     ],
-    outputs=gr.Files(label="Download Split Videos"),
+    outputs=gr.Files(label="Video Segments"),
     title="Video Splitter",
-   description="""
+    description="""
     Split your video file into multiple parts using different methods:
     1. Free Split: Specify a list of comma-separated time stamps to split the video into segments.
     2. Average Split: Enter the number of parts to divide the video into, with each segment having equal duration.
@@ -56,5 +43,4 @@ iface = gr.Interface(
     """,
 )
 
-if __name__ == "__main__":
-    iface.launch()
+interface.launch()