JRE/main.py at main · amrnth/JRE · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import asyncio
from csv_utils import CSVUtils
from make_shorts import Subtitles, VideoEditor
from clip_new import get_intervals, clip_video
from llm import LLM
from utils import merge_intervals
from yt_utils import VideoTools

async def perform_work(video_urls: list[str]):
    for url in video_urls:
        try:
            slug = VideoTools.get_slug_from_yt_video_url(url)
            data_folder = f"processed_data"
            video_dir = data_folder+"/"+slug

            results = await asyncio.gather(
                VideoTools.download_yt_video(url, video_dir),
                VideoTools.download_yt_subtitles(url, video_dir)
            )

            [video_path, subtitles_path] = results

            reduced_subtitles_path = LLM.generate_script_gemini(video_dir, subtitles_path)

            output_video_path = video_dir + "/final_video.mp4"
            intervals = merge_intervals(get_intervals(reduced_subtitles_path))
            clip_video(video_path, intervals, output_video_path)
            offsetted_subtitles = CSVUtils.offset_csv_file_timestamps(reduced_subtitles_path)
            final_video_path = video_dir+"/final_video_subbed.mp4"

            editor = VideoEditor(video_dir, output_video_path, output_path=final_video_path, subtitles=Subtitles(offsetted_subtitles))
            editor.process_video()
        except Exception as e:
            print(f"error processing {url}: ", e)
            continue

if __name__ == "__main__":
    yt_video_urls = [
        # "https://youtube.com/watch?v=37KieyXOYG4",
        # "https://youtube.com/watch?v=8kJacGNg_9w",
        # "https://youtube.com/watch?v=af_BsEH5qFk",
        # "https://youtube.com/watch?v=bklrNzdtU3Q",
        # "https://youtube.com/watch?v=CO5E-VoR3sA",
        # "https://youtube.com/watch?v=DEsCPOdPTM4",
        # "https://youtube.com/watch?v=FozCkl1xj-w",
        # "https://www.youtube.com/watch?v=Zy71pfPreuU",
        # "https://www.youtube.com/watch?v=kMD7K-kTKvg",
        # "https://www.youtube.com/watch?v=kJjgr0So7ek",
        # "https://www.youtube.com/watch?v=18vrb7i8dX0&t=6s",
        # "https://www.youtube.com/watch?v=GfWrvaibYnM",
        # "https://www.youtube.com/watch?v=GNG6ZzDh9C8",
        "https://www.youtube.com/watch?v=i2RCpvil-fs&pp=ygUHanJlIDMzOA%3D%3D",
        ]
    asyncio.run(perform_work(yt_video_urls))
    # download_save_video_info(yt_video_urls)