o
    +if`                     @  s8  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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Z.d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)ImagesResponseImagesAsyncImagesc                   @     e Zd Zed1ddZed2ddZeeeeeddded		d3dd Zeeeeeeddded!
d4d&d'Zeeeeeeeddded(d5d/d0Z	dS )6r   returnImagesWithRawResponsec                 C     t | S N)r"   self r'   P/var/www/html/corbot_env/lib/python3.10/site-packages/openai/resources/images.pywith_raw_response      zImages.with_raw_responseImagesWithStreamingResponsec                 C  r#   r$   )r+   r%   r'   r'   r(   with_streaming_response#   r*   zImages.with_streaming_responseN	modelnresponse_formatsizeuserextra_headersextra_query
extra_bodytimeoutimager   r.   0Union[str, Literal['dall-e-2'], None] | NotGivenr/   Optional[int] | NotGivenr0   /Optional[Literal['url', 'b64_json']] | NotGivenr1   ?Optional[Literal['256x256', '512x512', '1024x1024']] | NotGivenr2   str | NotGivenr3   Headers | Noner4   Query | Noner5   Body | Noner6   'float | httpx.Timeout | None | NotGivenr   c       
      
   C  sp   t ||||||d}tttttf |dggd}|r$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
        r7   r.   r/   r0   r1   r2   r7   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&   r7   r.   r/   r0   r1   r2   r3   r4   r5   r6   rJ   rK   r'   r'   r(   create_variation'   s*   0

zImages.create_variation
maskr.   r/   r0   r1   r2   r3   r4   r5   r6   promptrN   rU   FileTypes | NotGivenc             
   C  sx   t ||||||||d}tttttf |dgdggd}|r(ddi|	p&i }	| 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
        r7   rV   rU   r.   r/   r0   r1   r2   r7   rU   rC   rE   rF   /images/editsrH   rI   )r   r   r   r   rN   rO   rP   r   r
   ImageEditParamsr   r   r&   r7   rV   rU   r.   r/   r0   r1   r2   r3   r4   r5   r6   rJ   rK   r'   r'   r(   editq   s.   8"
zImages.editr.   r/   qualityr0   r1   styler2   r3   r4   r5   r6   <Union[str, Literal['dall-e-2', 'dall-e-3'], None] | NotGivenr_   $Literal['standard', 'hd'] | NotGivenYOptional[Literal['256x256', '512x512', '1024x1024', '1792x1024', '1024x1792']] | NotGivenr`   0Optional[Literal['vivid', 'natural']] | NotGivenc                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rV   r.   r/   r_   r0   r1   r`   r2   rH   rJ   rL   rM   )rP   r   r   ImageGenerateParamsr   r   r&   rV   r.   r/   r_   r0   r1   r`   r2   r3   r4   r5   r6   r'   r'   r(   generate   s&   ;
zImages.generate)r!   r"   )r!   r+   r7   r   r.   r8   r/   r9   r0   r:   r1   r;   r2   r<   r3   r=   r4   r>   r5   r?   r6   r@   r!   r   r7   r   rV   rN   rU   rW   r.   r8   r/   r9   r0   r:   r1   r;   r2   r<   r3   r=   r4   r>   r5   r?   r6   r@   r!   r   rV   rN   r.   ra   r/   r9   r_   rb   r0   r:   r1   rc   r`   rd   r2   r<   r3   r=   r4   r>   r5   r?   r6   r@   r!   r   
__name__
__module____qualname__r   r)   r,   r   rS   r]   rk   r'   r'   r'   r(   r      L    OXc                   @  r    )6r   r!   AsyncImagesWithRawResponsec                 C  r#   r$   )rt   r%   r'   r'   r(   r)     r*   zAsyncImages.with_raw_response AsyncImagesWithStreamingResponsec                 C  r#   r$   )ru   r%   r'   r'   r(   r,     r*   z#AsyncImages.with_streaming_responseNr-   r7   r   r.   r8   r/   r9   r0   r:   r1   r;   r2   r<   r3   r=   r4   r>   r5   r?   r6   r@   r   c       
      
     s~   t ||||||d}tttttf |dggd}|r%ddi|p#i }| jdt|tj	I dH |t
|||	|
dtd	I dH S )
rA   rB   r7   rC   rE   rF   rG   NrH   rI   )r   r   r   r   rN   rO   rP   r   r   rQ   r   r   rR   r'   r'   r(   rS     s,   0
zAsyncImages.create_variationrT   rV   rN   rU   rW   c             
     s   t ||||||||d}tttttf |dgdggd}|r)ddi|	p'i }	| jdt|tj	I dH |t
|	|
||d	td
I dH S )rX   rY   r7   rU   rC   rE   rF   rZ   NrH   rI   )r   r   r   r   rN   rO   rP   r   r
   r[   r   r   r\   r'   r'   r(   r]   i  s0   8"zAsyncImages.editr^   ra   r_   rb   rc   r`   rd   c                  sF   | j dt||||||||dtjI dH t|	|
||dtdI dH S )re   rf   rg   NrH   rh   )rP   r   r   ri   r   r   rj   r'   r'   r(   rk     s(   ;
zAsyncImages.generate)r!   rt   )r!   ru   rl   rm   rn   ro   r'   r'   r'   r(   r     rs   c                   @     e Zd ZdddZdS )	r"   imagesr   r!   Nonec                 C  4   || _ t|j| _t|j| _t|j| _d S r$   )_imagesr	   to_raw_response_wrapperrS   r]   rk   r&   rw   r'   r'   r(   __init__     
zImagesWithRawResponse.__init__Nrw   r   r!   rx   rp   rq   rr   r}   r'   r'   r'   r(   r"         r"   c                   @  rv   )	rt   rw   r   r!   rx   c                 C  ry   r$   )rz   r	   async_to_raw_response_wrapperrS   r]   rk   r|   r'   r'   r(   r}     r~   z#AsyncImagesWithRawResponse.__init__Nrw   r   r!   rx   r   r'   r'   r'   r(   rt     r   rt   c                   @  rv   )	r+   rw   r   r!   rx   c                 C  .   || _ t|j| _t|j| _t|j| _d S r$   )rz   r   rS   r]   rk   r|   r'   r'   r(   r}   -     
z$ImagesWithStreamingResponse.__init__Nr   r   r'   r'   r'   r(   r+   ,  r   r+   c                   @  rv   )	ru   rw   r   r!   rx   c                 C  r   r$   )rz   r   rS   r]   rk   r|   r'   r'   r(   r}   <  r   z)AsyncImagesWithStreamingResponse.__init__Nr   r   r'   r'   r'   r(   ru   ;  r   ru   )/
__future__r   typingr   r   r   r   typing_extensionsr   httpx r	   typesr
   r   r   _typesr   r   r   r   r   r   _utilsr   r   r   r   _compatr   	_resourcer   r   	_responser   r   _base_clientr   types.images_responser   __all__r   r   r"   rt   r+   ru   r'   r'   r'   r(   <module>   s,     y y