g"dhddlmZddlmZmZmZmZddlmZddl Z ddl m Z ddl m Z mZmZddlmZmZmZmZmZmZdd lmZmZmZmZdd lmZdd lmZm Z dd l!m"Z"m#Z#dd l$m%Z%ddl&m'Z'ddl(m)Z)ddgZ*GddeZ+Gdde Z,GddZ-GddZ.GddZ/GddZ0y)) annotations)UnionMappingOptionalcast)LiteralN)_legacy_response)image_edit_paramsimage_generate_paramsimage_create_variation_params) NOT_GIVENBodyQueryHeadersNotGiven FileTypes) extract_filesmaybe_transformdeepcopy_minimalasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)make_request_options) ImageModel)ImagesResponseImages AsyncImagesc eZdZed dZed dZeeeeeddded d dZeeeeeeddded d dZeeeeeeeddded dd Z y)r ct|Sa This property can be used as a prefix for any HTTP method call to return the 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 )ImagesWithRawResponseselfs L/var/www/openai/venv/lib/python3.12/site-packages/openai/resources/images.pywith_raw_responsezImages.with_raw_responses%T**ct|Sz 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 )ImagesWithStreamingResponser&s r(with_streaming_responsezImages.with_streaming_response(s+400r*N modelnresponse_formatsizeuser extra_headers extra_query extra_bodytimeoutc  t||||||d} ttttt f| dgg} ddi|xsi}|j dt| tj| t||| | tS)  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 imager0r1r2r3r4r<paths Content-Typemultipart/form-data/images/variationsr5r6r7r8bodyfilesoptionscast_to) rrrrstrobject_postrr ImageCreateVariationParamsrr r'r<r0r1r2r3r4r5r6r7r8rDrEs r(create_variationzImages.create_variation1s` #2    d73;#7>ykR()>X=CVTVX zz  'D'_'_`(+Q[el#  r* maskr0r1r2r3r4r5r6r7r8c t||||||||d} ttttt f| dgdgg}ddi| xsi} |j dt| tj|t| | | | t 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 r<promptrOr0r1r2r3r4r<rOr=r?r@ /images/editsrBrC) rrrrrHrIrJrr ImageEditParamsrrr'r<rSrOr0r1r2r3r4r5r6r7r8rDrEs r(editz Images.editzsp  #2   d73;#7>ySYRZF[\()>X=CVTVX zz  '8'H'HI(+Q[el#  r* r0r1qualityr2r3styler4r5r6r7r8c |jdt||||||||dtjt | | | | t 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/generationsrSr0r1rYr2r3rZr4rBrDrFrG)rJrr ImageGenerateParamsrr r'rSr0r1rYr2r3rZr4r5r6r7r8s r(generatezImages.generateshvzz ! $"&'6 "  &99 )+Q[el#%  r*)returnr%)rcr-r<rr0'Union[str, ImageModel, None] | NotGivenr1Optional[int] | NotGivenr2/Optional[Literal['url', 'b64_json']] | NotGivenr3?Optional[Literal['256x256', '512x512', '1024x1024']] | NotGivenr4str | NotGivenr5Headers | Noner6 Query | Noner7 Body | Noner8'float | httpx.Timeout | None | NotGivenrcrr<rrSrHrOzFileTypes | NotGivenr0rer1rfr2rgr3rhr4rir5rjr6rkr7rlr8rmrcrrSrHr0rer1rfrYz$Literal['standard', 'hd'] | NotGivenr2rgr3zYOptional[Literal['256x256', '512x512', '1024x1024', '1792x1024', '1024x1792']] | NotGivenrZz0Optional[Literal['vivid', 'natural']] | NotGivenr4rir5rjr6rkr7rlr8rmrcr __name__ __module__ __qualname__rr)r.rrMrWrbr*r(r r sB++11:C&/KTPY()-$("&;DG G 7 G $ G I G NG G &G "G  G 9G  G \&/9B&/KTPY()-$("&;D!Q Q  Q # Q 7 Q  $Q IQ NQ Q &Q "Q  Q 9!Q " #Q n:C&/8AKTjsBK()-$("&;D!N N 7 N $ N 6 N IN hN @N N &N "N  N 9!N " #N r*c eZdZed dZed dZeeeeeddded d dZeeeeeeddded d dZeeeeeeeddded dd Z y)r!ct|Sr$)AsyncImagesWithRawResponser&s r(r)zAsyncImages.with_raw_responses*$//r*ct|Sr,) AsyncImagesWithStreamingResponser&s r(r.z#AsyncImages.with_streaming_response)s055r*Nr/c <Kt||||||d} ttttt f| dgg} ddi|xsi}|j dt| tjd{| t||| | t d{S7%7w) r:r;r<r=r?r@rANrBrC) rrrrrHrIrJrr rKrrrLs r(rMzAsyncImages.create_variation2s` #2    d73;#7>ykR()>X=CVTVX ZZ ,T3P3k3kll(+Q[el#    l s$A0B2B 3 BBBBrNc DKt||||||||d} ttttt f| dgdgg}ddi| xsi} |j dt| tjd{|t| | | | t d{S7%7w) rQrRr<rOr=r?r@rTNrBrC) rrrrrHrIrJrr rUrrrVs r(rWzAsyncImages.edit{sp  #2   d73;#7>ySYRZF[\()>X=CVTVX ZZ ,T3D3T3TUU(+Q[el#    U s$A4B 6B 7 B BB B rXc K|jdt||||||||dtjd{t | | | | t d{S7$7w)r\r]r^NrBr_)rJrr r`rrras r(rbzAsyncImages.generatesvZZ !,$"&'6 "  &99  )+Q[el#%      s!3AA AAAA)rcrw)rcryrdrnrorprtr*r(r!r!sB0066:C&/KTPY()-$("&;DG G 7 G $ G I G NG G &G "G  G 9G  G \&/9B&/KTPY()-$("&;D!Q Q  Q # Q 7 Q  $Q IQ NQ Q &Q "Q  Q 9!Q " #Q n:C&/8AKTjsBK()-$("&;D!N N 7 N $ N 6 N IN hN @N N &N "N  N 9!N " #N r*ceZdZddZy)r%c||_tj|j|_tj|j|_tj|j |_yN)_imagesr to_raw_response_wrapperrMrWrbr'imagess r(__init__zImagesWithRawResponse.__init__ s[ 0 H H  # #! %<< KK  )@@ OO  r*Nrr rcNonerqrrrsrrtr*r(r%r%  r*r%ceZdZddZy)rwc||_tj|j|_tj|j|_tj|j |_yr)rr async_to_raw_response_wrapperrMrWrbrs r(rz#AsyncImagesWithRawResponse.__init__/s[ 0 N N  # #! %BB KK  )FF OO  r*Nrr!rcrrrtr*r(rwrw.rr*rwceZdZddZy)r-c||_t|j|_t|j|_t|j|_yr)rrrMrWrbrs r(rz$ImagesWithStreamingResponse.__init__>sI <  # #! 1 KK  5 OO  r*Nrrrtr*r(r-r-=rr*r-ceZdZddZy)ryc||_t|j|_t|j|_t|j|_yr)rrrMrWrbrs r(rz)AsyncImagesWithStreamingResponse.__init__MsI B  # #! 7 KK  ; OO  r*Nrrrtr*r(ryryLrr*ry)1 __future__rtypingrrrrtyping_extensionsrhttpxr typesr r r _typesrrrrrr_utilsrrrr_compatr _resourcerr _responserr _base_clientrtypes.image_modelrtypes.images_responser__all__r r!r%rwr-ryrtr*r(rs#11% [[II &9X/*2 ] #~ _~ B~ "~ B            r*