o pi @sddlZddlZddlmZmZmZmZddlZddl Zddl Z ddl m Z m Z  d deeejjfdeeejjgejjfdejjfdd Z d d edeeeejjgeejjfdeejjfd d ZdS)N)CallableListOptionalUnion)BACKENDS_MAPPINGis_opencv_availableimageconvert_methodreturncCst|tr1|ds|drtjtj|ddj}n#t j |r)tj|}nt d|dt|tjjr;|}nt dtj |}|durO||}|S|d }|S) a Loads `image` to a PIL Image. Args: image (`str` or `PIL.Image.Image`): The image to convert to the PIL Image format. convert_method (Callable[[PIL.Image.Image], PIL.Image.Image], *optional*): A conversion method to apply to the image after loading it. When set to `None` the image will be converted "RGB". Returns: `PIL.Image.Image`: A PIL Image. http://https://TstreamIIncorrect path or URL. URLs must start with `http://` or `https://`, and  is not a valid path.zgIncorrect format used for the image. Should be a URL linking to an image, a local path, or a PIL image.NRGB) isinstancestr startswithPILImageopenrequestsgetrawospathisfile ValueErrorImageOpsexif_transposeconvert)r r r#[/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/utils/loading_utils.py load_image s&     r%videocCs|dp |d}tj|}d}|s|std|d|rVtj|ddj}tj tj |ddd j }d}t |d }| |Wd n1sOwY|}g}|d rtj |} z || | | dqety~Yn=wtrd d l} n ttddd| |} | \} } | r| | | j} |tj| | \} } | s| |rt!||d ur||}|S)a Loads `video` to a list of PIL Image. Args: video (`str`): A URL or Path to a video to convert to a list of PIL Image format. convert_method (Callable[[List[PIL.Image.Image]], List[PIL.Image.Image]], *optional*): A conversion method to apply to the video after loading it. When set to `None` the images will be converted to "RGB". Returns: `List[PIL.Image.Image]`: The video as a list of PIL images. r r FrrTrr)suffixdeletewbNz.gifropencv load_video)"rrrrrrrrtempfileNamedTemporaryFilesplitextnamerwritereadendswithrrappendcopyseektellEOFErrorrcv2 ImportErrorrformat VideoCapturecvtColor COLOR_BGR2RGB fromarrayreleaseremove)r&r is_urlis_filewas_tempfile_created video_data video_pathf pil_imagesgifr8 video_capturesuccessframer#r#r$r+7sT           r+)N)rr,typingrrrr PIL.Imager PIL.ImageOpsr import_utilsrrrrr%r+r#r#r#r$s. -