Spaces:
Sleeping
Sleeping
import os | |
import moviepy.editor as mp | |
import gradio as gr | |
from concurrent.futures import ThreadPoolExecutor | |
def split_video(video_file, parts, resolution, quality, audio_enabled, volume): | |
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)) | |
for future in futures: | |
output_filename = future.result() | |
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 | |
# Остальной код остается без изменений |