qi!dZddlZddlZddlZddlmZmZddlmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZddlmZmZmZdd lmZdd lmZmZmZmZeGd d eZGd deZ y)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 _NotGiven_warn_deprecated_paramc:eZdZUdZedZedzezed<edZ eezed<edZ eezed <ed Z e e e efzezed <ed Zeezed <edZeezed<edZe ezed<de e effdZy)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. ctSNrL/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/fal/image.pyzFalImageGenSettings.,sr)default_factoryNseedctSrrrrrrzFalImageGenSettings.-srnum_inference_stepsctSrrrrrrzFalImageGenSettings..s r num_imagesctSrrrrrrzFalImageGenSettings./sQZr image_sizectSrrrrrrzFalImageGenSettings.0sIr expand_promptctSrrrrrrzFalImageGenSettings.1sIrenable_safety_checkerctSrrrrrrzFalImageGenSettings.2sIrformatreturnci}|j|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.rr!r#r%r'r)r+)rr!r#r%r'r)r+)selfargss rto_api_argumentsz$FalImageGenSettings.to_api_arguments4s~! 99 99DL&*&>&> "#!__\!__\ $ 2 2_(,(B(B $%X r)__name__ __module__ __qualname____doc__rrintr__annotations__r!r#r%strrr'boolr)r+rr0rrrrrs $)9J#KD#*y K+0AR+SyS"'8I"JJiJ38IZ3[Jd38n$y0[&+r?r@FTrA)rDrr!r#r%r'r)r+rDrCrFFAL_KEYr)rrrDrr!r#r%r'r)r+ apply_updatesuper__init___aiohttp_sessionosgetenv_api_keyenviron) r.rCrGrDrErFkwargsdefault_settings __class__s rrMzFalImageGenService.__init__fs </$ !""&     "7,? I%*  "   "8-@ A(.  %7=7Q7Q 4.4.?.? +.4.?.? +171E1E .9?9U9U 6*0-- '    ) )( 3 ="2=f= /7ryyB7 $'BJJy ! rpromptr,cKdtfd}tjd|d|jdd}d|i|jj }|j jd|jj|| 4d {}|jd k7rI|jd {}td |jd | d d d d {y |jd {}d d d d {r |dddnd }|stdy tjd|tjd|d|j j|4d {}tjd||jjd {} t!j"|| d {\} } } t%|| | | } | d d d d {y 7{7W7*77#1d {7swYxYw77r7V70#1d {7swYy xYww)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_imagectj|}tj|}|j |j |j fSr)ioBytesIOr opentobytessizer+)rXbufferimages rload_image_bytesz:FalImageGenService.run_image_gen..load_image_bytess9ZZ .FJJv&EMMOUZZ> >rzGenerating image from prompt: zKey zapplication/json) Authorizationz Content-TyperVzhttps://fal.run/)jsonheadersNzFal API error (z): )errorimagesrurlzImage generation failedzImage generated at: zDownloading image z ...zDownloaded image )rhr`r^r+)bytesr debugrQr;r0rNpostrDstatustextr rcgetcontentreadasyncio to_threadr)r.rVrardpayloadresp error_textresponse image_urlrX image_bytesr^r+frames r run_image_genz FalImageGenService.run_image_gens< ?E ?  5fX>? $DMM?3. VIt~~'F'F'HI((--t~~334 5. ) ){{c!#'99;.   S 'UVV ) ) )"YY[(H ) )5=HX&q)%0$ 67 7  +I;78  ))D9:((,,Y7  8 LL,YK8 9"*"2"2"7"7"99M070A0ABRTa0b*b '[$$+DY_`EK    + ) / )) ) ) ) )* 9*b    sB I H I #H5H 6$H I %H&I +H>H?H I HA4I H.I 6H6=H0>H6H2H68 I H4I  HI HI H+H! H+& I 0H62H64I 6I<H? =II )r1r2r3r4rSettingsr6r rBraiohttp ClientSessionr7rMrrrz __classcell__)rUs@rr:r:Bs #H""i0#" )-#!26A(%A(!.. A( } A( c] A(./A(F3#3.2M3rr:)!r4rqrZrO dataclassesrrtypingrrrrr r|logurur PILr pydanticr pipecat.frames.framesr rrpipecat.services.image_servicerpipecat.services.settingsrrrrrr:rrrrsd  (==EE:dd !*! !HZZr