o i @s0dZddlZddlZdejd<ddlmZmZddlmZm Z m Z ddl m Z ddl mZdd lmZmZdd lmZmZmZdd lmZdd lmZdd lmZmZmZzddlmZddl m!Z!Wn e"yZ#ze $de#e $de%de#dZ#[#wweGdddeZ&GdddeZ'dS)zGoogle AI image generation service implementation. This module provides integration with Google's Imagen model for generating images from text prompts using the Google AI API. NfalseGRPC_ENABLE_FORK_SUPPORT) dataclassfield)AnyAsyncGeneratorOptional)logger)Image) BaseModelField) ErrorFrameFrameURLImageRawFrame)!update_google_client_http_options)ImageGenService) NOT_GIVENImageGenSettings _NotGiven)genai)typesz Exception: zHIn order to use Google AI, you need to `pip install pipecat-ai[google]`.zMissing module: c@sJeZdZUdZedddZeeBed<edddZ e dBeBed<dS) GoogleImageGenSettingsa Settings for the Google image generation service. Parameters: model: Google Imagen model identifier. number_of_images: Number of images to generate per request. negative_prompt: Text describing what not to include in generated images. cCtSNrrrQ/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/google/image.py2zGoogleImageGenSettings.)default_factorynumber_of_imagescCrrrrrrrr3rNnegative_prompt) __name__ __module__ __qualname____doc__rr intr__annotations__r!strrrrrr(s "rc seZdZUdZeZeed<GdddeZddddde de ed e e d e effd d Z d e fddZde d eedffddZZS)GoogleImageGenServiceaGoogle AI image generation service using Imagen models. Provides text-to-image generation capabilities using Google's Imagen models through the Google AI API. Supports multiple image generation and negative prompting for enhanced control over generated content. _settingsc@sPeZdZUdZeddddZeed<eddZe ed<ed dZ e e ed <d S) z!GoogleImageGenService.InputParamsaConfiguration parameters for Google image generation. .. deprecated:: 0.0.105 Use ``settings=GoogleImageGenService.Settings(...)`` instead. Parameters: number_of_images: Number of images to generate (1-8). Defaults to 1. model: Google Imagen model to use. Defaults to "imagen-3.0-generate-002". negative_prompt: Optional negative prompt to guide what not to include. )defaultgeler imagen-3.0-generate-002)r-modelNr!) r"r#r$r%r r r&r'r1r(r!rrrrr InputParamsAs  r2N)params http_optionssettingsapi_keyr3r4r5c s|jdddd}|dur|d|s|j|_|j|_|j|_|dur(||tjdd|i|t|}t j ||d|_ dS) aqInitialize the GoogleImageGenService with API key and parameters. Args: api_key: Google AI API key for authentication. params: Configuration parameters for image generation. .. deprecated:: 0.0.105 Use ``settings=GoogleImageGenService.Settings(...)`` instead. http_options: HTTP options for the client. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional arguments passed to the parent ImageGenService. r0r+N)r1r r!r3r5)r6r4r) Settings"_warn_init_param_moved_to_settingsr1r r! apply_updatesuper__init__rrClient_client)selfr6r3r4r5kwargsdefault_settings __class__rrr;Qs   zGoogleImageGenService.__init__returncCsdS)zCheck if this service can generate processing metrics. Returns: True, as Google image generation service supports metrics. Tr)r>rrrcan_generate_metricssz*GoogleImageGenService.can_generate_metricspromptc Cstd||IdHzQ|jjjj|jj|t j |jj |jj ddIdH}| IdH|r6|js>tdVWdS|jD]}|jj}tt|}td||j|jd}|VqAWdSty~}ztdt|VWYd}~dSd}~ww)aGenerate images from a text prompt using Google's Imagen model. Args: prompt: The text description to generate images from. Yields: Frame: Generated URLImageRawFrame objects containing the generated images, or ErrorFrame objects if generation fails. Raises: Exception: If there are issues with the Google AI API or image processing. zGenerating image from prompt: N)r r!)r1rEconfigzImage generation failed)urlimagesizeformatzImage generation error: )r debugstart_ttfb_metricsr=aiomodelsgenerate_imagesr*r1rGenerateImagesConfigr r!stop_ttfb_metricsgenerated_imagesr rH image_bytesr openioBytesIOrtobytesrIrJ Exceptionr()r>rEresponse img_responserSrHframeerrr run_image_gens>       "z#GoogleImageGenService.run_image_gen)r"r#r$r%rr7r'r r2r(rrr;boolrDrrr] __classcell__rrrArr)6s& 1"r))(r%rUosenviron dataclassesrrtypingrrrlogurur PILr pydanticr r pipecat.frames.framesr rrpipecat.services.google.utilsrpipecat.services.image_servicerpipecat.services.settingsrrrgoogler google.genairModuleNotFoundErrorr\errorrXrr)rrrrs2