Update app.py

app.py

@@ -1,70 +1,53 @@
-import os
-import moviepy.video.io.VideoFileClip as mp
 import gradio as gr
-from concurrent.futures import ThreadPoolExecutor
+import moviepy.editor as mp
+from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 
-def split_video(video_file, parts, resolution, quality, audio_enabled, volume):
+def split_video(video_file, split_method, split_value):
     video = mp.VideoFileClip(video_file.name)
     duration = video.duration
-    part_duration = duration / parts
 
-    output_files = []
-    with ThreadPoolExecutor() as executor:
-        futures = []
-        for part in range(parts):
-            start_time = part * part_duration
-            end_time = (part + 1) * part_duration
-            futures.append(executor.submit(process_part, video, start_time, end_time, resolution, quality, audio_enabled, volume, part))
+    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
 
-        for future in futures:
-            output_filename = future.result()
-            output_files.append(output_filename)
+    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
 
-def process_part(video, start_time, end_time, resolution, quality, audio_enabled, volume, part):
-    part_clip = video.subclip(start_time, end_time)
-
-    # Настройка параметров кодирования
-    if quality == "Низкое":
-        bitrate = "1M"
-    elif quality == "Среднее":
-        bitrate = "2M"
-    else:
-        bitrate = "4M"
-
-    # Настройка параметров звука
-    if not audio_enabled:
-        part_clip.audio = None
-    else:
-        part_clip.audio = part_clip.audio.volumex(volume)
-
-    # Сохранение видео
-    output_filename = f"part_{part + 1}.mp4"
-    part_clip.write_videofile(
-        output_filename,
-        codec="libx264",
-        audio_codec="aac",
-        bitrate=bitrate,
-        preset="medium",
-        ffmpeg_params=["-vf", f"scale={resolution}"],
-    )
-
-    return output_filename
-
 iface = gr.Interface(
     fn=split_video,
     inputs=[
         gr.File(label="Upload Video"),
-        gr.Slider(minimum=2, maximum=10, value=2, label="Number of Parts"),
-        gr.Dropdown(choices=["720p", "1080p", "4K"], label="Resolution"),
-        gr.Dropdown(choices=["Низкое", "Среднее", "Высокое"], label="Качество"),
-        gr.Checkbox(label="Включить звук"),
-        gr.Slider(minimum=0, maximum=2, value=1, label="Громкость"),
+        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="Download Split Videos"),
     title="Video Splitter",
-    description="Upload your video and select how many parts you want to split it into."
+    description="Split your video file into multiple parts using different methods.",
 )
 
 if __name__ == "__main__":