Update app.py

app.py

@@ -1,38 +1,58 @@
 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(input_video, duration):
-    if not input_video:
-        raise ValueError("No video file provided.")
+def split_video(video_file, split_method, split_value):
+    video = mp.VideoFileClip(video_file.name)
+    duration = video.duration
 
-    video_info = ffmpeg.probe(input_video)
-    video_duration = float(video_info['format']['duration'])
-    num_segments = ceil(video_duration / duration)
+    # Free Split: Пользователь может указать список временных меток через запятую для разбиения видео на сегменты.
+    if split_method == "Free Split":
+        split_times = [float(t) for t in split_value.split(",")]
+        split_times = [t for t in split_times if 0 <= 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:])]
     
-    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)
+    # Average Split: Пользователь указывает количество частей, на которые нужно разделить видео, и каждый сегмент будет иметь одинаковую продолжительность.
+    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)]
     
-    return segments
+    # Time Split: Пользователь указывает список временных меток через запятую для разбиения видео. Каждый сегмент будет начинаться с указанного времени.
+    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])]
+    
+    # Size Split: Пользователь указывает целевой размер сегмента в мегабайтах. Видео будет разделено на сегменты, каждый из которых будет иметь размер, близкий к указанному значению.
+    elif split_method == "Size Split":
+        target_size = int(split_value) * 1024 * 1024
+        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
+
+    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
 
-interface = gr.Interface(
-    split_video,
+iface = gr.Interface(
+    fn=split_video,
     inputs=[
         gr.File(label="Upload Video"),
-        gr.Number(label="Segment Duration (seconds)"),
+        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)"),
     ],
-    outputs=gr.Files(label="Video Segments"),
+    outputs=gr.Files(label="Download Split Videos"),
     title="Video Splitter",
     description="""
     Split your video file into multiple parts using different methods:
@@ -43,4 +63,5 @@ interface = gr.Interface(
     """,
 )
 
-interface.launch()
+if __name__ == "__main__":
+    iface.launch()