Compare commits

...

2 Commits

Author SHA1 Message Date
Dhruv Nair
6ac0d7764a update 2024-08-15 05:00:10 +00:00
Dhruv Nair
0082fe9477 update 2024-08-15 04:52:55 +00:00

View File

@@ -1,6 +1,7 @@
import os import os
import tempfile import tempfile
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
from urllib.parse import unquote, urlparse
import PIL.Image import PIL.Image
import PIL.ImageOps import PIL.ImageOps
@@ -80,12 +81,22 @@ def load_video(
) )
if is_url: if is_url:
video_data = requests.get(video, stream=True).raw response = requests.get(video, stream=True)
suffix = os.path.splitext(video)[1] or ".mp4" if response.status_code != 200:
raise ValueError(f"Failed to download video. Status code: {response.status_code}")
parsed_url = urlparse(video)
file_name = os.path.basename(unquote(parsed_url.path))
suffix = os.path.splitext(file_name)[1] or ".mp4"
video_path = tempfile.NamedTemporaryFile(suffix=suffix, delete=False).name video_path = tempfile.NamedTemporaryFile(suffix=suffix, delete=False).name
was_tempfile_created = True was_tempfile_created = True
video_data = response.iter_content(chunk_size=8192)
with open(video_path, "wb") as f: with open(video_path, "wb") as f:
f.write(video_data.read()) for chunk in video_data:
f.write(chunk)
video = video_path video = video_path