o iI@sdZddlZddlZddlmZmZddlmZmZddl Z ddl m Z ddl m Z mZmZddlmZddlmZmZmZeGd d d eZGd d d eZdS) zAzure OpenAI image generation service implementation. This module provides integration with Azure's OpenAI image generation API using REST endpoints for creating images from text prompts. N) dataclassfield)AsyncGeneratorOptional)Image) ErrorFrameFrameURLImageRawFrame)ImageGenService) NOT_GIVENImageGenSettings _NotGivenc@s0eZdZUdZedddZedBeBed<dS)AzureImageGenSettingszSettings for the Azure image generation service. Parameters: model: Azure image generation model identifier. image_size: Target size for generated images. cCstS)N)r rrP/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/azure/image.py#szAzureImageGenSettings.)default_factoryN image_size) __name__ __module__ __qualname____doc__rrstrr __annotations__rrrrrs "rcs|eZdZUdZeZeed<ddddddeededed eed e j d eef fd d Z dede e dffddZZS)AzureImageGenServiceRESTzAzure OpenAI REST-based image generation service. Provides image generation using Azure's OpenAI service via REST API. Supports asynchronous image generation with polling for completion and automatic image download and processing. _settingsNz2023-06-01-preview)rmodel api_versionsettingsrapi_keyendpointraiohttp_sessionrc s~|jddd}|dur|dd||_|dur!|dd||_|dur*||tj|d||_||_||_ ||_ dS)a{Initialize the AzureImageGenServiceREST. Args: image_size: Size specification for generated images (e.g., "1024x1024"). .. deprecated:: 0.0.105 Use ``settings=AzureImageGenServiceREST.Settings(image_size=...)`` instead. api_key: Azure OpenAI API key for authentication. endpoint: Azure OpenAI endpoint URL. model: The image generation model to use. .. deprecated:: 0.0.105 Use ``settings=AzureImageGenServiceREST.Settings(model=...)`` instead. aiohttp_session: Shared aiohttp session for HTTP requests. api_version: Azure API version string. Defaults to "2023-06-01-preview". settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. N)rrrr)r) Settings"_warn_init_param_moved_to_settingsrr apply_updatesuper__init___api_key_azure_endpoint _api_version_aiohttp_session) selfrrr rr!rrdefault_settings __class__rrr&1s"     z!AzureImageGenServiceREST.__init__promptreturnc Cs|jd|j}|jdd}|dd}|jjdur!|jj|d<|jj|||d4IdH}|jd }d }d }d} |d kry|d8}|d krWtdV WdIdHdSt dIdH|jj ||dIdH} | IdH} | d}|d ks>| r| ddd dnd} | stdV WdIdHdS|j | 4IdH*} t | jIdH} t| } t| | | j| jd}|VWdIdHn 1IdHswYWdIdHdS1IdHswYdS)aGenerate an image from a text prompt using Azure OpenAI. Args: prompt: The text prompt describing the desired image. Yields: URLImageRawFrame containing the generated image data, or ErrorFrame if generation fails. z-openai/images/generations:submit?api-version=zapplication/json)zapi-keyz Content-Type)r/nNsize)headersjsonzoperation-locationx succeededrzImage generation timed out)r4statusresultdataurlzImage generation failed)r<imager3format)r(r)r'rrr*postr4rasynciosleepgetr5ioBytesIOcontentreadropenr tobytesr3r>)r+r/r<r4body submissionoperation_locationr9 attempts_left json_responseresponse image_url image_streamr=framerrr run_image_genjsL        *.z&AzureImageGenServiceREST.run_image_gen)rrrrrr"rrraiohttp ClientSessionr&rrrR __classcell__rrr-rr&s,  "9r)rr@rC dataclassesrrtypingrrrSPILrpipecat.frames.framesrrr pipecat.services.image_servicer pipecat.services.settingsr r r rrrrrrs