o Gi@s ddlZddlZddlmZmZddlmZmZddlZ ddl Z ddl Z ddl m Z ddlmZmZ ddee jjBdee jjge jjfdBd e jjfd d Z dd edeee jjgee jjfdBd ee jjfd dZded eeeffddZdefddZdS)N)AnyCallable)unquoteurlparse)DIFFUSERS_REQUEST_TIMEOUT)BACKENDS_MAPPINGis_imageio_availableimageconvert_methodreturncCst|tr2|ds|drtjtj|dtdj }n#t j |r*tj|}nt d|dt|tjjr<|}nt dtj|}|durP||}|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://T)streamtimeoutIIncorrect 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 startswithPILImageopenrequestsgetrrawospathisfile ValueErrorImageOpsexif_transposeconvert)r r r$Q/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/utils/loading_utils.py load_images&     r&videocCs|dp |d}tj|}d}|s|std|d|r|tj|dd}|jdkr4td |jt|}tj t |j}tj |d pJd }t j |dd j} d}|jd d} t| d} | D]} | | qcWdn1suwY| }g} |drtj|}z | |||d qtyYnKwtrddl}n ttdd dz|jj !Wn t"yt"dw|#|}|D] }| tj$|qWdn1swY|rt%| |dur|| } | 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 rFrrT)rz'Failed to download video. Status code: rz.mp4)suffixdeletei ) chunk_sizewbNz.gifrimageio load_videozf`Unable to find an ffmpeg installation on your machine. Please install via `pip install imageio-ffmpeg)&rrrrr rr status_coderbasenamersplitexttempfileNamedTemporaryFilename iter_contentrwriteendswithrrappendcopyseektellEOFErrorr r- ImportErrorrformatpluginsffmpegget_exeAttributeError get_reader fromarrayremove)r'r is_urlis_filewas_tempfile_createdresponse parsed_url file_namer) video_path video_datafchunk pil_imagesgifr-readerframer$r$r%r.9sl             r. tensor_namecCs^d|vr+|d}|ddD]}t||}|dur$t|d|d|}q|d}||fS)N.z has no attribute )splitgetattrr )modulerTsplitsrW new_moduler$r$r%get_module_from_names  r\ module_pathcCs@|}|d}|D]}|rt|}||}q t||}q |S)NrU)rWisdigitintrX) root_moduler]currentpartspartidxr$r$r%get_submodule_by_names   re)N)rr2typingrr urllib.parserr PIL.Imager PIL.ImageOpsr constantsr import_utilsrr rrr&listr.tupler\rer$r$r$r%s6   -   U