U ~fhq@sHddlmZddlmZmZmZmZmZddlm Z ddl Z ddl m Z ddl mZmZmZddlmZmZmZmZmZmZdd lmZmZmZmZdd lmZdd lm Z m!Z!dd l"m#Z#m$Z$dd l%m&Z&ddl'm(Z(ddl)m*Z*ddgZ+Gddde Z,Gddde!Z-GdddZ.GdddZ/GdddZ0GdddZ1dS)) annotations)ListUnionMappingOptionalcast)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@seZdZeddddZeddddZeeeeeddded d d d d dddddddd ddZeeeeeeeeeedddedddddd d d ddd dddddddd d!d"Zeeeeeeeeeeeddded#ddd d$d d dd%d d&d'ddddddd(d)d*Z dS)+r ImagesWithRawResponsereturncCst|Sa 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"selfr(;/tmp/pip-unpacked-wheel-rjiviquo/openai/resources/images.pywith_raw_responseszImages.with_raw_responseImagesWithStreamingResponsecCst|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 )r+r&r(r(r)with_streaming_response#szImages.with_streaming_responseN modelnresponse_formatsizeuser extra_headers extra_query extra_bodytimeoutr'Union[str, ImageModel, None] | NotGivenOptional[int] | NotGiven1Optional[Literal[('url', 'b64_json')]] | NotGivenAOptional[Literal[('256x256', '512x512', '1024x1024')]] | NotGivenstr | NotGivenHeaders | None Query | None Body | None'float | httpx.Timeout | None | NotGivenr imager/r0r1r2r3r4r5r6r7r$c Cslt||||||d} tttttf| dggd} ddi|p@i}|jdt| tj | t ||| | dt dS) Creates a variation of a given image. This endpoint only supports `dall-e-2`. 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. 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 rBr/r0r1r2r3rBpaths Content-Typemultipart/form-data/images/variationsr4r5r6r7bodyfilesoptionscast_to) rrrrstrobject_postrr ImageCreateVariationParamsrr r'rBr/r0r1r2r3r4r5r6r7rLrMr(r(r)create_variation,s.0  zImages.create_variation backgroundmaskr/r0output_compression output_formatqualityr1r2r3r4r5r6r7!Union[FileTypes, List[FileTypes]]rP?Optional[Literal[('transparent', 'opaque', 'auto')]] | NotGivenFileTypes | NotGiven5Optional[Literal[('png', 'jpeg', 'webp')]] | NotGivenKOptional[Literal[('standard', 'low', 'medium', 'high', 'auto')]] | NotGivencOptional[Literal[('256x256', '512x512', '1024x1024', '1536x1024', '1024x1536', 'auto')]] | NotGivenrBpromptrWrXr/r0rYrZr[r1r2r3r4r5r6r7r$cCst||||||||| | | | d }tttttf|dgddgdggd}ddi| pVi} |jdt|tj |t | |||d t d S) NCreates an edited or extended image given one or more source images and a prompt. This endpoint only supports `gpt-image-1` and `dall-e-2`. Args: image: The image(s) to edit. Must be a supported image file or an array of images. For `gpt-image-1`, each image should be a `png`, `webp`, or `jpg` file less than 50MB. You can provide up to 16 images. For `dall-e-2`, you can only provide one image, and it should be a square `png` file less than 4MB. prompt: A text description of the desired image(s). The maximum length is 1000 characters for `dall-e-2`, and 32000 characters for `gpt-image-1`. background: Allows to set transparency for the background of the generated image(s). This parameter is only supported for `gpt-image-1`. Must be one of `transparent`, `opaque` or `auto` (default value). When `auto` is used, the model will automatically determine the best background for the image. If `transparent`, the output format needs to support transparency, so it should be set to either `png` (default value) or `webp`. mask: An additional image whose fully transparent areas (e.g. where alpha is zero) indicate where `image` should be edited. If there are multiple images provided, the mask will be applied on the first image. 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` and `gpt-image-1` are supported. Defaults to `dall-e-2` unless a parameter specific to `gpt-image-1` is used. n: The number of images to generate. Must be between 1 and 10. output_compression: The compression level (0-100%) for the generated images. This parameter is only supported for `gpt-image-1` with the `webp` or `jpeg` output formats, and defaults to 100. output_format: The format in which the generated images are returned. This parameter is only supported for `gpt-image-1`. Must be one of `png`, `jpeg`, or `webp`. The default value is `png`. quality: The quality of the image that will be generated. `high`, `medium` and `low` are only supported for `gpt-image-1`. `dall-e-2` only supports `standard` quality. Defaults to `auto`. 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. This parameter is only supported for `dall-e-2`, as `gpt-image-1` will always return base64-encoded images. size: The size of the generated images. Must be one of `1024x1024`, `1536x1024` (landscape), `1024x1536` (portrait), or `auto` (default value) for `gpt-image-1`, and one of `256x256`, `512x512`, or `1024x1024` for `dall-e-2`. 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 rBrcrWrXr/r0rYrZr[r1r2r3rBrXrErGrH /images/editsrJrK) rrrrrPrQrRrr ImageEditParamsrrr'rBrcrWrXr/r0rYrZr[r1r2r3r4r5r6r7rLrMr(r(r)editus:\( z Images.editrWr/ moderationr0rYrZr[r1r2styler3r4r5r6r7-Optional[Literal[('low', 'auto')]] | NotGivenQOptional[Literal[('standard', 'hd', 'low', 'medium', 'high', 'auto')]] | NotGiven}Optional[Literal[('auto', '1024x1024', '1536x1024', '1024x1536', '256x256', '512x512', '1792x1024', '1024x1792')]] | NotGiven2Optional[Literal[('vivid', 'natural')]] | NotGivenrcrWr/rlr0rYrZr[r1r2rmr3r4r5r6r7r$cCs@|jdt||||||||| | | | d tjt| |||dtdS) Creates an image given a prompt. [Learn more](https://platform.openai.com/docs/guides/images). Args: prompt: A text description of the desired image(s). The maximum length is 32000 characters for `gpt-image-1`, 1000 characters for `dall-e-2` and 4000 characters for `dall-e-3`. background: Allows to set transparency for the background of the generated image(s). This parameter is only supported for `gpt-image-1`. Must be one of `transparent`, `opaque` or `auto` (default value). When `auto` is used, the model will automatically determine the best background for the image. If `transparent`, the output format needs to support transparency, so it should be set to either `png` (default value) or `webp`. model: The model to use for image generation. One of `dall-e-2`, `dall-e-3`, or `gpt-image-1`. Defaults to `dall-e-2` unless a parameter specific to `gpt-image-1` is used. moderation: Control the content-moderation level for images generated by `gpt-image-1`. Must be either `low` for less restrictive filtering or `auto` (default value). n: The number of images to generate. Must be between 1 and 10. For `dall-e-3`, only `n=1` is supported. output_compression: The compression level (0-100%) for the generated images. This parameter is only supported for `gpt-image-1` with the `webp` or `jpeg` output formats, and defaults to 100. output_format: The format in which the generated images are returned. This parameter is only supported for `gpt-image-1`. Must be one of `png`, `jpeg`, or `webp`. quality: The quality of the image that will be generated. - `auto` (default value) will automatically select the best quality for the given model. - `high`, `medium` and `low` are supported for `gpt-image-1`. - `hd` and `standard` are supported for `dall-e-3`. - `standard` is the only option for `dall-e-2`. response_format: The format in which generated images with `dall-e-2` and `dall-e-3` are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated. This parameter isn't supported for `gpt-image-1` which will always return base64-encoded images. size: The size of the generated images. Must be one of `1024x1024`, `1536x1024` (landscape), `1024x1536` (portrait), or `auto` (default value) for `gpt-image-1`, one of `256x256`, `512x512`, or `1024x1024` for `dall-e-2`, and one of `1024x1024`, `1792x1024`, or `1024x1792` for `dall-e-3`. style: The style of the generated images. This parameter is only supported for `dall-e-3`. 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. 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 rcrWr/rlr0rYrZr[r1r2rmr3rJrLrNrO)rRrr ImageGenerateParamsrrr'rcrWr/rlr0rYrZr[r1r2rmr3r4r5r6r7r(r(r)generates4^zImages.generate __name__ __module__ __qualname__rr*r-rrUrjryr(r(r(r)r sZ  &N2c@seZdZeddddZeddddZeeeeeddded d d d d dddddddd ddZeeeeeeeeeedddedddddd d d ddd dddddddd d!d"Zeeeeeeeeeeeddded#ddd d$d d dd%d d&d'ddddddd(d)d*Z dS)+r!AsyncImagesWithRawResponser#cCst|Sr%)r~r&r(r(r)r*iszAsyncImages.with_raw_response AsyncImagesWithStreamingResponsecCst|Sr,)rr&r(r(r)r-ssz#AsyncImages.with_streaming_responseNr.rr8r9r:r;r<r=r>r?r@rrAc sxt||||||d} tttttf| dggd} ddi|p@i}|jdt| tj IdH| t ||| | dt d IdHS) rCrDrBrErGrHrINrJrK) rrrrrPrQrRrr rSrrrTr(r(r)rU|s.0 zAsyncImages.create_variationrVr\rPr]r^r_r`rarbcst||||||||| | | | d }tttttf|dgddgdggd}ddi| pVi} |jdt|tj Id H|t | |||d t d Id HS) rdrerBrfrXrErGrHrgNrJrK) rrrrrPrQrRrr rhrrrir(r(r)rjs:\(zAsyncImages.editrkrnrorprqrrcsL|jdt||||||||| | | | d tjIdHt| |||dtdIdHS)rsrtruNrJrv)rRrr rwrrrxr(r(r)ry@s4^zAsyncImages.generaterzr(r(r(r)r!hsZ  &N2c@seZdZdddddZdS)r"r Noneimagesr$cCs4||_t|j|_t|j|_t|j|_dSN)_imagesr Zto_raw_response_wrapperrUrjryr'rr(r(r)__init__szImagesWithRawResponse.__init__Nr{r|r}rr(r(r(r)r"sr"c@seZdZdddddZdS)r~r!rrcCs4||_t|j|_t|j|_t|j|_dSr)rr Zasync_to_raw_response_wrapperrUrjryrr(r(r)rsz#AsyncImagesWithRawResponse.__init__Nrr(r(r(r)r~sr~c@seZdZdddddZdS)r+r rrcCs.||_t|j|_t|j|_t|j|_dSr)rrrUrjryrr(r(r)rsz$ImagesWithStreamingResponse.__init__Nrr(r(r(r)r+sr+c@seZdZdddddZdS)rr!rrcCs.||_t|j|_t|j|_t|j|_dSr)rrrUrjryrr(r(r)rsz)AsyncImagesWithStreamingResponse.__init__Nrr(r(r(r)rsr)2 __future__rtypingrrrrrZtyping_extensionsrZhttpxr typesr r r _typesrrrrrr_utilsrrrr_compatrZ _resourcerr _responserrZ _base_clientrZtypes.image_modelrZtypes.images_responser__all__r r!r"r~r+rr(r(r(r)s0        RR