qi<dZddlZddlmZmZddlmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZmZmZeGd d eZGd deZy)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 _NotGiven_warn_deprecated_paramc<eZdZUdZedZedzezed<y)OpenAIImageGenSettingszSettings for the OpenAI image generation service. Parameters: model: DALL-E model identifier. image_size: Target size for generated images. ctS)N)rO/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/openai/image.pyzOpenAIImageGenSettings.(syr)default_factoryN image_size) __name__ __module__ __qualname____doc__rrstrr__annotations__rrrrrs"*/?P)QJd Y&QrrceZdZUdZeZeed<ddddddedeede jdee d d eed eef fd Z d ede edffdZxZS)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_keyr%aiohttp_sessionr)256x256512x512 1024x1024 1792x1024 1024x1792r&r'ctdd}|tdtd||_|tdtd||_||j |t ||t|||_||_ y)aIInitialize 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=OpenAIImageGenSettings(image_size=...)`` instead. model: DALL-E model to use for generation. Defaults to "dall-e-3". .. deprecated:: 0.0.105 Use ``settings=OpenAIImageGenSettings(model=...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. zdall-e-3N)r&rr&r)r')r(r%) rrr&r apply_updatesuper__init__r _client_aiohttp_session) selfr(r%r)rr&r'default_settings __class__s rr2zOpenAIImageGenService.__init__6s@2    "7,BG L%*  "  ! "<1G V*4  '    ) )( 3 "23"7XF /rpromptreturncKtjd||jjj ||j j d|j jd{}|jdj}|stdy|jj|4d{}tj|jj!d{}t#j$|}t'|j)|j*|j,|}|dddd{y777h7 #1d{7swYyxYww)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: )r8r&nsizeNrzImage generation failed)imager=formaturl)rdebugr3imagesgenerater$r&rdatar@r r4getioBytesIOcontentreadr openr tobytesr=r?)r5r8r> image_urlresponse image_streamframes r run_image_genz#OpenAIImageGenService.run_image_genls3  5fX>?ll))22!5!5AZAZ3  JJqM%% 67 7 ((,,Y7  8::H,<,<,A,A,C&CDLJJ|,E$mmoZZ|| E K     &C    snA-E0/E0A E0=E>E0-E.E /AE E0 EE0E0EE0E-!E$ "E-)E0)rrrrrSettingsr!r raiohttp ClientSessionrr2rr rP __classcell__)r7s@rr#r#+s&H%% #' #5940403- 40 !.. 40  O P 40}401240l#.2Mrr#)rrF dataclassesrrtypingrrrrRlogururopenair PILr pipecat.frames.framesr r r pipecat.services.image_servicerpipecat.services.settingsrrrrrr#rrrr]sf  (44 ;dd R-R R`O`r