
    -`i(                         d dl mZ d dlmZ d dlmZ d dlZd dlmZ ddl	m
Z
 dZed	         Z ed
           G d d                      ZdS )    )	dataclass)Path)LiteralN)Image   )get_vllm_public_assetsvision_model_images)	stop_signcherry_blossomhatoz12560px-Gfp-wisconsin-madison-the-nature-boardwalk$Grayscale_8bits_palette_sample_imagez1280px-Venn_diagram_rgb	RGBA_compz237-400x300z231-200x300z
27-500x500z
17-150x600zhandelsblatt-previewzpaper-11T)frozenc                       e Zd ZU eed<   dedefdZede	j	        fd            Z
dede	j	        fdZedej        fd            ZdedefdZd	S )

ImageAssetnameextreturnc                 B    t          | j         d| t                    S )z1
        Return s3 path for given image.
        .)filename	s3_prefix)r   r   VLM_IMAGES_DIR)selfr   s     e/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/assets/image.pyget_pathzImageAsset.get_path$   s0     &	))C))^
 
 
 	
    c                 .    |                      d          S )Njpgr   )pil_image_ext)r   s    r   	pil_imagezImageAsset.pil_image,   s    !!e!,,,r   c                 V    |                      |          }t          j        |          S )Nr    )r   r   open)r   r   
image_paths      r   r!   zImageAsset.pil_image_ext0   s%    ]]s]++
z*%%%r   c                 Z    |                      d          }t          j        |dd          S )zR
        Image embeddings, only used for testing purposes with llava 1.5.
        ptcpuT)map_locationweights_only)r   torchload)r   r%   s     r   image_embedszImageAsset.image_embeds4   s,    
 ]]4((
z*5tLLLLr   c                 n    t          |                     |                    }|                                S )N)r   r   
read_bytes)r   r   ps      r   r/   zImageAsset.read_bytes<   s)    s##$$||~~r   N)__name__
__module____qualname__ImageAssetName__annotations__strr   r   propertyr   r"   r!   r+   Tensorr-   bytesr/    r   r   r   r       s         

C 
D 
 
 
 
 -5; - - - X-& & & & & & Mel M M M XMc e      r   r   )dataclassesr   pathlibr   typingr   r+   PILr   baser   r   r4   r   r:   r   r   <module>r@      s    " ! ! ! ! !                    ( ( ( ( ( (&" $         r   