o i!@sdZddlZddlZddlZddlmZmZddlmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZddlmZmZmZdd lmZdd lmZmZmZeGd d d eZGd ddeZdS)zFal's image generation service implementation. This module provides integration with Fal's image generation API for creating images from text prompts using various AI models. N) dataclassfield)AnyAsyncGeneratorDictOptionalUnion)logger)Image) BaseModel) ErrorFrameFrameURLImageRawFrame)ImageGenService) NOT_GIVENImageGenSettings _NotGivenc@seZdZUdZedddZedBeBed<edddZ eeBed<ed ddZ eeBed <ed ddZ e e e efBeBed <ed ddZeeBed<edddZeeBed<edddZe eBed<de e effddZdS)FalImageGenSettingsaGSettings for the Fal image generation service. Parameters: model: Fal.ai model identifier. seed: Random seed for reproducible generation. ``None`` uses a random seed. num_inference_steps: Number of inference steps for generation. num_images: Number of images to generate. image_size: Image dimensions as a string preset or dict with width/height. expand_prompt: Whether to automatically expand/enhance the prompt. enable_safety_checker: Whether to enable content safety filtering. format: Output image format. cCtSNrrrN/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/fal/image.py,zFalImageGenSettings.)default_factoryNseedcCrrrrrrrr-rnum_inference_stepscCrrrrrrrr.r num_imagescCrrrrrrrr/r image_sizecCrrrrrrrr0r expand_promptcCrrrrrrrr1renable_safety_checkercCrrrrrrrr2rformatreturncCsXi}|jdur |j|d<|j|d<|j|d<|j|d<|j|d<|j|d<|j|d<|S) zFBuild the Fal API arguments dict from settings, excluding None values.Nrrrrr r!r")rrrrr r!r")selfargsrrrto_api_arguments4s        z$FalImageGenSettings.to_api_arguments)__name__ __module__ __qualname____doc__rrintr__annotations__rrrstrrr boolr!r"rr&rrrrrs  &rc seZdZUdZeZeed<GdddeZeed<dddddde ede j d e e d e e d e ef fd d Z de deedffddZZS)FalImageGenServicezFal's image generation service. Provides text-to-image generation using Fal.ai's API with configurable parameters for image quality, safety, and format options. _settingsc@szeZdZUdZdZeeed<dZeed<dZ eed<dZ e e e e effed <d Zeed <d Zeed <dZe ed<dS)zFalImageGenService.InputParamsaInput parameters for Fal.ai image generation. .. deprecated:: 0.0.105 Use ``settings=FalImageGenService.Settings(...)`` instead. Parameters: seed: Random seed for reproducible generation. If None, uses random seed. num_inference_steps: Number of inference steps for generation. Defaults to 8. num_images: Number of images to generate. Defaults to 1. image_size: Image dimensions as string preset or dict with width/height. Defaults to "square_hd". expand_prompt: Whether to automatically expand/enhance the prompt. Defaults to False. enable_safety_checker: Whether to enable content safety filtering. Defaults to True. format: Output image format. Defaults to "png". Nrrr square_hdrFr Tr!pngr")r'r(r)r*rrr+r,rrrrr-rr r.r!r"rrrr InputParamsLs     r5N)paramsmodelkeysettingsr6aiohttp_sessionr7r8r9c s|jddddddddd }|dur|d d ||_|durA|d |sA|j|_|j|_|j|_|j|_|j|_|j|_|j |_ |durJ| |t j dd |i|||_ |p_td d|_|rj|tjd <dSdS)ajInitialize the FalImageGenService. Args: params: Input parameters for image generation configuration. .. deprecated:: 0.0.105 Use ``settings=FalImageGenService.Settings(...)`` instead. aiohttp_session: HTTP client session for downloading generated images. model: The Fal.ai model to use for generation. Defaults to "fal-ai/fast-sdxl". .. deprecated:: 0.0.105 Use ``settings=FalImageGenService.Settings(model=...)`` instead. key: Optional API key for Fal.ai. If provided, sets FAL_KEY environment variable. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional arguments passed to parent ImageGenService. zfal-ai/fast-sdxlNr1r2r3FTr4)r7rrrrr r!r"r7r6r9FAL_KEYr)Settings"_warn_init_param_moved_to_settingsr7rrrrr r!r" apply_updatesuper__init___aiohttp_sessionosgetenv_api_keyenviron)r$r6r:r7r8r9kwargsdefault_settings __class__rrrAfs>   zFalImageGenService.__init__promptr#c Csdtfdd}td|d|jdd}d|i|j}|jjd |jj||d 4Id H7}|j d krZ| Id H}t d |j d|dV Wd Id Hd S| Id H}Wd Id Hn 1Id HsqwY|r|dddnd }|st dVd Std|td|d|j |4Id H3}td||jId H} t|| Id H\} } } t|| | | d} | VWd Id Hd S1Id HswYd S)aGenerate an image from a text prompt. Args: prompt: The text prompt to generate an image from. Yields: URLImageRawFrame: Frame containing the generated image data and metadata. ErrorFrame: If image generation fails. encoded_imagecSs&t|}t|}||j|jfSr)ioBytesIOr opentobytessizer")rLbufferimagerrrload_image_bytess  z:FalImageGenService.run_image_gen..load_image_byteszGenerating image from prompt: zKey zapplication/json) Authorizationz Content-TyperKzhttps://fal.run/)jsonheadersNzFal API error (z): )errorimagesrurlzImage generation failedzImage generated at: zDownloading image z ...zDownloaded image )r[rSrQr")bytesr debugrEr0r&rBpostr7statustextr rVgetcontentreadasyncio to_threadr)r$rKrTrWpayloadresp error_textresponse image_urlrL image_bytesrQr"framerrr run_image_gensB     (  .z FalImageGenService.run_image_gen)r'r(r)r*rr=r,r r5raiohttp ClientSessionr-rArr rm __classcell__rrrIrr/Bs, "Cr/) r*rdrMrC dataclassesrrtypingrrrrrrnlogurur PILr pydanticr pipecat.frames.framesr r rpipecat.services.image_servicerpipecat.services.settingsrrrrr/rrrrs     $