Ускорить процесс генерации видео moviepy

import os
import random
import moviepy.editor as mp
from moviepy.editor import TextClip, CompositeVideoClip
import whisper
import pysrt
from concurrent.futures import ThreadPoolExecutor
 
def split_audio_by_word(audio_file):
    word_timestamps = []
    model = whisper.load_model("small")
    result = model.transcribe(audio_file, language='ru')
 
    for segment in result['segments']:
        words = segment['text'].split()
        duration = (segment['end'] - segment['start']) / len(words)
        start_time = segment['start']
 
        for word in words:
            end_time = start_time + duration
            word_timestamps.append({
                'word': word,
                'start': start_time,
                'end': end_time
            })
            start_time = end_time
 
    return word_timestamps
 
def generate_word_subtitles(audio_file, output_srt):
    word_timestamps = split_audio_by_word(audio_file)
 
    with open(output_srt, 'w', encoding='utf-8') as f:
        for idx, word_timestamp in enumerate(word_timestamps):
            start = word_timestamp['start']
            end = word_timestamp['end']
            text = word_timestamp['word']
            f.write(f"{idx + 1}\n")
            f.write(f"{int(start//3600):02}:{int((start%3600)//60):02}:{start%60:.3f} --> {int(end//3600):02}:{int((end%3600)//60):02}:{end%60:.3f}\n")
            f.write(f"{text.strip()}\n\n")
 
def add_subtitles_from_file(video_clip, subtitle_file):
    subtitles = []
    subs = pysrt.open(subtitle_file)
 
    def to_seconds(t):
        return t.hours * 3600 + t.minutes * 60 + t.seconds + t.milliseconds / 1000
 
    for sub in subs:
        start_time = to_seconds(sub.start)
        end_time = to_seconds(sub.end)
        text = sub.text
 
        text_clip = TextClip(
            text,
            font='Arial-Bold',
            fontsize=90,
            color='white',
            stroke_color='black',
            stroke_width=3
        ).set_start(start_time).set_duration(end_time - start_time).set_position('center')
 
        subtitles.append(text_clip)
 
    return CompositeVideoClip([video_clip, *subtitles])
 
def ensure_audio_duration(audio_file, duration):
    audio_clip = mp.AudioFileClip(audio_file)
    if audio_clip.duration < duration:
        silence = mp.AudioClip(lambda t: [0], duration=duration - audio_clip.duration, fps=44100)
        audio_clip = mp.concatenate_audioclips([audio_clip, silence])
    return audio_clip
 
def create_final_video(folder1, folder2, folder3, folder4, output_dir, output_filename):
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
 
    output_video = os.path.join(output_dir, output_filename)
 
    video_files1 = [os.path.join(folder1, f) for f in os.listdir(folder1) if f.endswith(('.mp4', '.avi', '.mov'))]
    video_files2 = [os.path.join(folder2, f) for f in os.listdir(folder2) if f.endswith(('.mp4', '.avi', '.mov'))]
    audio_files3 = [os.path.join(folder3, f) for f in os.listdir(folder3) if f.endswith(('.mp3', '.wav'))]
    audio_files4 = [os.path.join(folder4, f) for f in os.listdir(folder4) if f.endswith(('.mp3', '.wav'))]
 
    random_videos = random.sample(video_files1, 4) + [random.choice(video_files2)]
    random_audio3 = random.choice(audio_files3)
    random_audio4 = random.choice(audio_files4)
 
    try:
        # Создание видео клипов
        video_clips = [mp.VideoFileClip(video).resize((1080, 1920)) for video in random_videos]
        final_video = mp.concatenate_videoclips(video_clips)
 
        # Генерация субтитров
        subtitles_srt = 'subtitles.srt'
        generate_word_subtitles(random_audio3, subtitles_srt)
        final_video = add_subtitles_from_file(final_video, subtitles_srt)
 
        # Создание текстовых клипов "бот в профиле" и "пиши + в комменты"
        fixed_text2 = "пиши софт в комменты"
        fixed_text_clip2 = TextClip(fixed_text2, fontsize=36, color='yellow', stroke_color='black', stroke_width=2, font='Arial-Black')
 
        # Устанавливаем текст с самого начала видео
        fixed_text_clip2 = fixed_text_clip2.set_duration(final_video.duration).set_position((200, 1300)).set_opacity(0.5)
 
        # Добавляем текстовые клипы в видео
        final_video = CompositeVideoClip([final_video, fixed_text_clip2])
 
        # Обработка аудиофайлов
        voiceover = ensure_audio_duration(random_audio3, final_video.duration)
        music = ensure_audio_duration(random_audio4, final_video.duration).volumex(0.35)
 
        # Комбинирование аудиодорожек
        final_audio = mp.CompositeAudioClip([voiceover.set_duration(final_video.duration), music.set_duration(final_video.duration)])
        final_video = final_video.set_audio(final_audio)
 
        # Запись финального видео
        final_video.write_videofile(output_video, codec='libx264', audio_codec='aac')
    except Exception as e:
        print(f"Error creating final video: {e}")
 
 
def create_videos_in_parallel(folder1, folder2, folder3, folder4, output_dir, num_videos, repeats):
    with ThreadPoolExecutor(max_workers=num_videos) as executor:
        futures = []
        for i in range(repeats):
            for j in range(num_videos):
                output_filename = f'final_output_{i*num_videos + j + 1}.mp4'
                futures.append(executor.submit(create_final_video, folder1, folder2, folder3, folder4, output_dir, output_filename))
        
        for future in futures:
            try:
                future.result()
            except Exception as e:
                print(f"Error: {e}")
 
folder1 = r'D:\Монтаж Видео\Генератор\НовоеКрео\ПапичМелстрой'
folder2 = r'D:\Монтаж Видео\Генератор\Бот'
folder3 = r'D:\Монтаж Видео\Генератор\НовоеКрео\Озвучка'
folder4 = r'D:\Монтаж Видео\Генератор\НовоеКрео\Треки'
output_dir = r'D:\Монтаж Видео\Генератор\НовоеКрео\ИтоговыеВидео'
num_videos = 2  # Number of videos to create in parallel
repeats = 1  # Number of times to repeat the process
 
create_videos_in_parallel(folder1, folder2, folder3, folder4, output_dir, num_videos, repeats)