o i@sdZddlZddlmZmZddlmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZmZeGd d d eZGd ddeZdS)zOpenAI image generation service implementation. This module provides integration with OpenAI's DALL-E image generation API for creating images from text prompts. N) dataclassfield)AsyncGeneratorLiteralOptional)logger) AsyncOpenAI)Image) ErrorFrameFrameURLImageRawFrame)ImageGenService) NOT_GIVENImageGenSettings _NotGivenc@s0eZdZUdZedddZedBeBed<dS)OpenAIImageGenSettingszSettings for the OpenAI image generation service. Parameters: model: DALL-E model identifier. image_size: Target size for generated images. cCstS)N)rrrQ/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/openai/image.py(szOpenAIImageGenSettings.)default_factoryN image_size) __name__ __module__ __qualname____doc__rrstrr__annotations__rrrrrs "rcseZdZUdZeZeed<ddddddedeede j dee d d eed eef fd d Z dede edffddZZS)OpenAIImageGenServicezOpenAI DALL-E image generation service. Provides image generation capabilities using OpenAI's DALL-E models. Supports various image sizes and can generate images from text prompts with configurable quality and style parameters. _settingsN)base_urlrmodelsettingsapi_keyraiohttp_sessionr)256x256512x512 1024x1024 1792x1024 1024x1792r r!csz|jddd}|dur|dd||_|dur!|dd||_|dur*||tj|dt||d|_||_ dS)aYInitialize the OpenAI image generation service. Args: api_key: OpenAI API key for authentication. base_url: Custom base URL for OpenAI API. If None, uses default. aiohttp_session: HTTP session for downloading generated images. image_size: Target size for generated images. Defaults to "1024x1024". .. deprecated:: 0.0.105 Use ``settings=OpenAIImageGenService.Settings(image_size=...)`` instead. model: DALL-E model to use for generation. Defaults to "dall-e-3". .. deprecated:: 0.0.105 Use ``settings=OpenAIImageGenService.Settings(model=...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. zdall-e-3N)r rr r)r!)r"r) Settings"_warn_init_param_moved_to_settingsr r apply_updatesuper__init__r_client_aiohttp_session)selfr"rr#rr r!default_settings __class__rrr-6s     zOpenAIImageGenService.__init__promptreturnc Cstd||jjj||jjd|jjdIdH}|jdj }|s*t dVdS|j |4IdH+}t |jIdH}t|}t||j|j|d}|VWdIdHdS1IdHsewYdS)zGenerate an image from a text prompt using OpenAI's DALL-E. Args: prompt: Text description of the image to generate. Yields: Frame: URLImageRawFrame containing the generated image data. zGenerating image from prompt: )r4r nsizeNrzImage generation failed)imager8formaturl)rdebugr.imagesgeneraterr rdatar;r r/getioBytesIOcontentreadr openr tobytesr8r:)r0r4r9 image_urlresponse image_streamframerrr run_image_genls(    .z#OpenAIImageGenService.run_image_gen)rrrrrr)rrraiohttp ClientSessionrr-rr rK __classcell__rrr2rr+s0   "6r)rrA dataclassesrrtypingrrrrLlogururopenairPILr pipecat.frames.framesr r r pipecat.services.image_servicer pipecat.services.settingsrrrrrrrrrs