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 tempfile
from typing import Callable, List, Optional, Union
from urllib.parse import unquote, urlparse
import PIL.Image
import PIL.ImageOps
@@ -80,12 +81,22 @@ def load_video(
)
if is_url:
video_data = requests.get(video, stream=True).raw
suffix = os.path.splitext(video)[1] or ".mp4"
response = requests.get(video, stream=True)
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
was_tempfile_created = True
video_data = response.iter_content(chunk_size=8192)
with open(video_path, "wb") as f:
f.write(video_data.read())
for chunk in video_data:
f.write(chunk)
video = video_path