diff --git a/video_creation/background.py b/video_creation/background.py index 81f2ff6..6c9018a 100644 --- a/video_creation/background.py +++ b/video_creation/background.py @@ -18,15 +18,15 @@ def download_background(): Path("./assets/backgrounds/").mkdir(parents=True, exist_ok=True) background_options = [ # uri , filename , credit ("https://www.youtube.com/watch?v=n_Dv4JMiwK8", "parkour.mp4", "bbswitzer"), - ("https://www.youtube.com/watch?v=hf3r9TNSsyY", "sailing.mp4", "loopylongplays"), - ("https://www.youtube.com/watch?v=p4jroQX6Hwk", "witcher.mp4", "loopylongplays"), - ("https://www.youtube.com/watch?v=c03fwcx1fEQ", "valheim.mp4", "loopylongplays"), - ("https://www.youtube.com/watch?v=GTaXbH6iSFA", "parkour2.mp4", "siswizz"), - ("https://www.youtube.com/watch?v=-pj9hvEQLA0", "rollercoaster.mp4", "SlowTV RelaxBackground"), - ("https://www.youtube.com/watch?v=8L0HHaxfRmM", "diving.mp4", "SlowTV RelaxBackground"), - ("https://www.youtube.com/watch?v=dQiYnFQX3Qg", "driving.mp4", "SUJANSHETTY"), - ("https://www.youtube.com/watch?v=OhBo1A8atuA", "space.mp4", "SUJANSHETTY"), - ("https://www.youtube.com/watch?v=ps20zN1ZqUo", "parkour3.mp4", "plop"), + # ("https://www.youtube.com/watch?v=hf3r9TNSsyY", "sailing.mp4", "loopylongplays"), + # ("https://www.youtube.com/watch?v=p4jroQX6Hwk", "witcher.mp4", "loopylongplays"), + # ("https://www.youtube.com/watch?v=c03fwcx1fEQ", "valheim.mp4", "loopylongplays"), + # ("https://www.youtube.com/watch?v=GTaXbH6iSFA", "parkour2.mp4", "siswizz"), + # ("https://www.youtube.com/watch?v=-pj9hvEQLA0", "rollercoaster.mp4", "SlowTV RelaxBackground"), + # ("https://www.youtube.com/watch?v=8L0HHaxfRmM", "diving.mp4", "SlowTV RelaxBackground"), + # ("https://www.youtube.com/watch?v=dQiYnFQX3Qg", "driving.mp4", "SUJANSHETTY"), + # ("https://www.youtube.com/watch?v=OhBo1A8atuA", "space.mp4", "SUJANSHETTY"), + # ("https://www.youtube.com/watch?v=ps20zN1ZqUo", "parkour3.mp4", "plop"), # ( # "https://www.youtube.com/watch?v=2X9QGY__0II", # "rocket_league.mp4", @@ -59,6 +59,7 @@ def chop_background_video(video_length): background = VideoFileClip(f"assets/backgrounds/{choice}") + # adding 1 second to video length to prevent hard cut start_time, end_time = get_start_and_end_times(video_length + 1, background.duration) ffmpeg_extract_subclip( f"assets/backgrounds/{choice}", diff --git a/video_creation/screenshot_downloader.py b/video_creation/screenshot_downloader.py index 6147dff..ea54f60 100644 --- a/video_creation/screenshot_downloader.py +++ b/video_creation/screenshot_downloader.py @@ -2,13 +2,12 @@ import json from os import getenv from pathlib import Path -from playwright.async_api import async_playwright from playwright.sync_api import sync_playwright, ViewportSize + from rich.progress import track +from rich.console import Console from utils.console import print_step, print_substep -import json -from rich.console import Console console = Console() diff --git a/video_creation/video_uploader.py b/video_creation/video_uploader.py index 42e2328..de84817 100644 --- a/video_creation/video_uploader.py +++ b/video_creation/video_uploader.py @@ -1,7 +1,4 @@ -from asyncio import sleep import asyncio -import imp -from os import getenv import os from pathlib import Path import pickle @@ -12,10 +9,10 @@ from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.common.keys import Keys from playwright.async_api import async_playwright -import undetected_chromedriver.v2 as uc +import undetected_chromedriver as uc -from utils.console import print_step, print_substep from rich.console import Console +from utils.console import print_step, print_substep console = Console()