a
    _gd                     @  sD  d dl mZ d dlmZmZmZmZ d dlmZ d dl	Z	ddl
mZ ddlmZmZmZ ddlmZmZmZmZmZmZ dd	lmZmZmZmZ dd
lmZ ddlmZm Z  ddl!m"Z"m#Z# ddl$m%Z% ddl&m'Z' ddl(m)Z) ddgZ*G dd deZ+G dd de Z,G dd dZ-G dd dZ.G dd dZ/G dd dZ0dS )    )annotations)UnionMappingOptionalcast)LiteralN   )_legacy_response)image_edit_paramsimage_generate_paramsimage_create_variation_params)	NOT_GIVENBodyQueryHeadersNotGiven	FileTypes)extract_filesmaybe_transformdeepcopy_minimalasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)make_request_options)
ImageModel)ImagesResponseImagesAsyncImagesc                   @  s   e Zd ZeddddZeddddZeeeeeddded		d
dddddddddddddZeeeeeeddded
d
dddddddddddddddZeeeeeeedddeddddddd d!ddddddd"d#d$Z	dS )%r   ImagesWithRawResponsereturnc                 C  s   t | S a  
        This property can be used as a prefix for any HTTP method call to return
        the raw response object instead of the parsed content.

        For more information, see https://www.github.com/openai/openai-python#accessing-raw-response-data-eg-headers
        )r!   self r'   e/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/openai/resources/images.pywith_raw_response   s    zImages.with_raw_responseImagesWithStreamingResponsec                 C  s   t | S z
        An alternative to `.with_raw_response` that doesn't eagerly read the response body.

        For more information, see https://www.github.com/openai/openai-python#with_streaming_response
        )r*   r%   r'   r'   r(   with_streaming_response(   s    zImages.with_streaming_responseN	modelnresponse_formatsizeuserextra_headersextra_query
extra_bodytimeoutr   'Union[str, ImageModel, None] | NotGivenOptional[int] | NotGiven/Optional[Literal['url', 'b64_json']] | NotGiven?Optional[Literal['256x256', '512x512', '1024x1024']] | NotGivenstr | NotGivenHeaders | NoneQuery | NoneBody | None'float | httpx.Timeout | None | NotGivenr   imager.   r/   r0   r1   r2   r3   r4   r5   r6   r#   c       
      
   C  sl   t ||||||d}tttttf |dggd}ddi|p@i }| jdt|tj	|t
|||	|
dtdS )	  
        Creates a variation of a given image.

        Args:
          image: The image to use as the basis for the variation(s). Must be a valid PNG file,
              less than 4MB, and square.

          model: The model to use for image generation. Only `dall-e-2` is supported at this
              time.

          n: The number of images to generate. Must be between 1 and 10. For `dall-e-3`, only
              `n=1` is supported.

          response_format: The format in which the generated images are returned. Must be one of `url` or
              `b64_json`. URLs are only valid for 60 minutes after the image has been
              generated.

          size: The size of the generated images. Must be one of `256x256`, `512x512`, or
              `1024x1024`.

          user: A unique identifier representing your end-user, which can help OpenAI to monitor
              and detect abuse.
              [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids).

          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        rA   r.   r/   r0   r1   r2   rA   pathsContent-Typemultipart/form-data/images/variationsr3   r4   r5   r6   bodyfilesoptionscast_to)r   r   r   r   strobject_postr   r   ImageCreateVariationParamsr   r   r&   rA   r.   r/   r0   r1   r2   r3   r4   r5   r6   rK   rL   r'   r'   r(   create_variation1   s(    0

zImages.create_variation
maskr.   r/   r0   r1   r2   r3   r4   r5   r6   rO   FileTypes | NotGivenrA   promptrV   r.   r/   r0   r1   r2   r3   r4   r5   r6   r#   c             
   C  st   t ||||||||d}tttttf |dgdggd}ddi|	pHi }	| jdt|tj	|t
|	|
||dtd	S )
  
        Creates an edited or extended image given an original image and a prompt.

        Args:
          image: The image to edit. Must be a valid PNG file, less than 4MB, and square. If mask
              is not provided, image must have transparency, which will be used as the mask.

          prompt: A text description of the desired image(s). The maximum length is 1000
              characters.

          mask: An additional image whose fully transparent areas (e.g. where alpha is zero)
              indicate where `image` should be edited. Must be a valid PNG file, less than
              4MB, and have the same dimensions as `image`.

          model: The model to use for image generation. Only `dall-e-2` is supported at this
              time.

          n: The number of images to generate. Must be between 1 and 10.

          response_format: The format in which the generated images are returned. Must be one of `url` or
              `b64_json`. URLs are only valid for 60 minutes after the image has been
              generated.

          size: The size of the generated images. Must be one of `256x256`, `512x512`, or
              `1024x1024`.

          user: A unique identifier representing your end-user, which can help OpenAI to monitor
              and detect abuse.
              [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids).

          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        rA   rY   rV   r.   r/   r0   r1   r2   rA   rV   rD   rF   rG   /images/editsrI   rJ   )r   r   r   r   rO   rP   rQ   r   r
   ImageEditParamsr   r   r&   rA   rY   rV   r.   r/   r0   r1   r2   r3   r4   r5   r6   rK   rL   r'   r'   r(   editz   s,    8"
zImages.editr.   r/   qualityr0   r1   styler2   r3   r4   r5   r6   $Literal['standard', 'hd'] | NotGivenYOptional[Literal['256x256', '512x512', '1024x1024', '1792x1024', '1024x1792']] | NotGiven0Optional[Literal['vivid', 'natural']] | NotGivenrY   r.   r/   ra   r0   r1   rb   r2   r3   r4   r5   r6   r#   c                C  s8   | j dt||||||||dtjt|	|
||dtdS )  
        Creates an image given a prompt.

        Args:
          prompt: A text description of the desired image(s). The maximum length is 1000
              characters for `dall-e-2` and 4000 characters for `dall-e-3`.

          model: The model to use for image generation.

          n: The number of images to generate. Must be between 1 and 10. For `dall-e-3`, only
              `n=1` is supported.

          quality: The quality of the image that will be generated. `hd` creates images with finer
              details and greater consistency across the image. This param is only supported
              for `dall-e-3`.

          response_format: The format in which the generated images are returned. Must be one of `url` or
              `b64_json`. URLs are only valid for 60 minutes after the image has been
              generated.

          size: The size of the generated images. Must be one of `256x256`, `512x512`, or
              `1024x1024` for `dall-e-2`. Must be one of `1024x1024`, `1792x1024`, or
              `1024x1792` for `dall-e-3` models.

          style: The style of the generated images. Must be one of `vivid` or `natural`. Vivid
              causes the model to lean towards generating hyper-real and dramatic images.
              Natural causes the model to produce more natural, less hyper-real looking
              images. This param is only supported for `dall-e-3`.

          user: A unique identifier representing your end-user, which can help OpenAI to monitor
              and detect abuse.
              [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids).

          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        /images/generationsrY   r.   r/   ra   r0   r1   rb   r2   rI   rK   rM   rN   )rQ   r   r   ImageGenerateParamsr   r   r&   rY   r.   r/   ra   r0   r1   rb   r2   r3   r4   r5   r6   r'   r'   r(   generate   s&    ;
zImages.generate
__name__
__module____qualname__r   r)   r,   r   rT   r_   rm   r'   r'   r'   r(   r      sJ   	&N*Wc                   @  s   e Zd ZeddddZeddddZeeeeeddded		d
dddddddddddddZeeeeeeddded
d
dddddddddddddddZeeeeeeedddeddddddd d!ddddddd"d#d$Z	dS )%r    AsyncImagesWithRawResponser"   c                 C  s   t | S r$   )rr   r%   r'   r'   r(   r)     s    zAsyncImages.with_raw_response AsyncImagesWithStreamingResponsec                 C  s   t | S r+   )rs   r%   r'   r'   r(   r,   )  s    z#AsyncImages.with_streaming_responseNr-   r   r7   r8   r9   r:   r;   r<   r=   r>   r?   r   r@   c       
      
     sx   t ||||||d}tttttf |dggd}ddi|p@i }| jdt|tj	I dH |t
|||	|
dtd	I dH S )
rB   rC   rA   rD   rF   rG   rH   NrI   rJ   )r   r   r   r   rO   rP   rQ   r   r   rR   r   r   rS   r'   r'   r(   rT   2  s(    0
zAsyncImages.create_variationrU   rO   rW   rX   c             
     s   t ||||||||d}tttttf |dgdggd}ddi|	pHi }	| jdt|tj	I dH |t
|	|
||d	td
I dH S )rZ   r[   rA   rV   rD   rF   rG   r\   NrI   rJ   )r   r   r   r   rO   rP   rQ   r   r
   r]   r   r   r^   r'   r'   r(   r_   {  s,    8"zAsyncImages.editr`   rc   rd   re   rf   c                  sD   | j dt||||||||dtjI dH t|	|
||dtdI dH S )rg   rh   ri   NrI   rj   )rQ   r   r   rk   r   r   rl   r'   r'   r(   rm     s&    ;
zAsyncImages.generatern   r'   r'   r'   r(   r      sJ   	&N*Wc                   @  s   e Zd ZdddddZdS )r!   r   Noneimagesr#   c                 C  s4   || _ t|j| _t|j| _t|j| _d S N)_imagesr	   Zto_raw_response_wrapperrT   r_   rm   r&   rv   r'   r'   r(   __init__   s    zImagesWithRawResponse.__init__Nro   rp   rq   rz   r'   r'   r'   r(   r!     s   r!   c                   @  s   e Zd ZdddddZdS )rr   r    rt   ru   c                 C  s4   || _ t|j| _t|j| _t|j| _d S rw   )rx   r	   Zasync_to_raw_response_wrapperrT   r_   rm   ry   r'   r'   r(   rz   /  s    z#AsyncImagesWithRawResponse.__init__Nr{   r'   r'   r'   r(   rr   .  s   rr   c                   @  s   e Zd ZdddddZdS )r*   r   rt   ru   c                 C  s.   || _ t|j| _t|j| _t|j| _d S rw   )rx   r   rT   r_   rm   ry   r'   r'   r(   rz   >  s    z$ImagesWithStreamingResponse.__init__Nr{   r'   r'   r'   r(   r*   =  s   r*   c                   @  s   e Zd ZdddddZdS )rs   r    rt   ru   c                 C  s.   || _ t|j| _t|j| _t|j| _d S rw   )rx   r   rT   r_   rm   ry   r'   r'   r(   rz   M  s    z)AsyncImagesWithStreamingResponse.__init__Nr{   r'   r'   r'   r(   rs   L  s   rs   )1
__future__r   typingr   r   r   r   Ztyping_extensionsr   Zhttpx r	   typesr
   r   r   _typesr   r   r   r   r   r   _utilsr   r   r   r   Z_compatr   Z	_resourcer   r   	_responser   r   Z_base_clientr   Ztypes.image_modelr   Ztypes.images_responser   __all__r   r    r!   rr   r*   rs   r'   r'   r'   r(   <module>   s0        