KikdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z ddl m Z ddlZddlmZddlmZddlmZejeZdZd Zd Zd Zd Zd ZdZdZddddZ e!e "Z#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,gdZ-dd gZ.gdZ/edd !Z0da1da2ej3Z4d"Z5d#e6d$e6fd%Z7Gd&d'Z8d(Z9d)e6d*ee6e ffd+Z: d}d-e e6ee6e;ffd.e;d/eeeeeedfd7e6d8e6d.e;d/ekrL eDd?eDd@eAZEeEs-eDdAeDdBeDdCeFd n eDdD ddlZeDdEn,#eG$r$eDdFeDdGeFd YnwxYweDdHeDdIeeDdJe$dKe%dLe0jHr(eDdMe0jIeDdNe0jIdOn eDdPeDdQeDdReDdSeDdTeDdUeDdVeDdWeDdXeDdYeDdZeDd[eDd\eDd]eDd^e-D]ZJeDd_eJeDd`eDdae/D]ZKeDd_eKeDdbeDdceDddeDdeeDdfeDdgeDdheDdieDdjeDdkeDdlddmlLmMZMdndodpdqdrdsdqgddtd dudvd7gdwdxZNdyZOeMjPdndzeNeOeBgdd{|dS)~a Image Generation Tools Module This module provides image generation tools using FAL.ai's FLUX 2 Pro model with automatic upscaling via FAL.ai's Clarity Upscaler for enhanced image quality. Available tools: - image_generate_tool: Generate images from text prompts with automatic upscaling Features: - High-quality image generation using FLUX 2 Pro model - Automatic 2x upscaling using Clarity Upscaler for enhanced quality - Comprehensive parameter control (size, steps, guidance, etc.) - Proper error handling and validation with fallback to original images - Debug logging support - Sync mode for immediate results Usage: from image_generation_tool import image_generate_tool import asyncio # Generate and automatically upscale an image result = await image_generate_tool( prompt="A serene mountain landscape with cherry blossoms", image_size="landscape_4_3", num_images=1 ) N)DictAnyOptionalUnion) urlencode) DebugSession)resolve_managed_tool_gateway)managed_nous_tools_enabledzfal-ai/flux-2-pro landscape2@pngF5landscape_16_9 square_hd portrait_16_9)r squareportraitzfal-ai/clarity-upscalerz"masterpiece, best quality, highresz.(worst quality, low quality, normal quality:2)gffffff?g333333?)rr portrait_4_3r landscape_4_3rjpeg)noneregularhigh image_toolsIMAGE_TOOLS_DEBUG)env_varcLtjdrdStdS)zOReturn managed fal-queue gateway config when direct FAL credentials are absent.FAL_KEYNz fal-queue)osgetenvr 8/home/ubuntu/hermes-agent/tools/image_generation_tool.py_resolve_managed_fal_gatewayr)Zs' yt ' 4 44r'queue_run_originreturnct|pdd}|std|dS)N/z$Managed FAL queue origin is required)strstriprstrip ValueError)r*normalized_origins r(_normalize_fal_queue_url_formatr4asW,23399;;BB3GG A?@@@ " " ""r'ceZdZdZdedefdZdddddddded eeefd ed eed eed edeeeefdee e e ffdZ dS)_ManagedFalSyncClientzPSmall per-instance wrapper around fal_client.SyncClient for managed queue hosts.keyr*cttdd}|tdttdd}|tdt||_|||_t|jdd|_t|dd|_t|dd|_t|d d|_ t|d d|_ t|d d|_ t|d d|_ |jtd |j|jtd|j tddS)N SyncClientz>fal_client.SyncClient is required for managed FAL gateway modeclientz:fal_client.client is required for managed FAL gateway mode)r7_client_maybe_retry_request_raise_for_statusSyncRequestHandleadd_hint_headeradd_priority_headeradd_timeout_headerzFfal_client.SyncClient._client is required for managed FAL gateway modezKfal_client.client request helpers are required for managed FAL gateway modezLfal_client.client.SyncRequestHandle is required for managed FAL gateway mode) getattr fal_client RuntimeErrorr4_queue_url_format _sync_client _http_clientr<r=_request_handle_class_add_hint_header_add_priority_header_add_timeout_header)selfr7r*sync_client_class client_modules r(__init__z_ManagedFalSyncClient.__init__ksg#J dCC  $_`` ` Hd;;  [\\ \!@AQ!R!R--#666#D$5y$GG$+M;QSW$X$X!!(8KT!R!R%,]"%),059/ / / / S>/  / sm / c]/ / $sCx.)/  c5j 12/ / / / / / r'r6c|jd|jf}t5tt |krtcdddSt |j|ja|atcdddS#1swxYwYdS)zQReuse the managed FAL client so its internal httpx.Client is not leaked per call.r.N)r7r*)gateway_originr1nous_user_token_managed_fal_client_lock_managed_fal_client_managed_fal_client_configr6)managed_gateway client_configs r(_get_managed_fal_clientrxs &--c22'M " # #  */I]/Z/Z& # # # # # # # #4/,;   &3"" # # # # # # # # # # # # # # # # # #sA?#A??BBmodelrWcdttji}t}|t j|||St |}||||S)zKSubmit a FAL request using direct credentials or the managed queue gateway.zx-idempotency-keyN)rWrT)r/uuiduuid4r)rCrirx)ryrWrfrvmanaged_clients r(_submit_fal_requestr~sv*C ,=,=>O244O )_UUUU,_==N    !  r'r image_sizenum_inference_stepsguidance_scale num_images output_format accelerationc4i}t|tr)|tvrtd|dt||d<nt|trd|vsd|vrtdt|dt rt|dt std|ddks |ddkrtd |dd ks |dd krtd ||d<ntd t|t r |d ks|dkrtd||d<t|t t fr |dks|dkrtdt ||d<t|t r |d ks|dkrtd||d<|tvrtd|dt||d<|tvrtd|dt||d<|S)a1 Validate and normalize image generation parameters for FLUX 2 Pro model. Args: image_size: Either a preset string or custom size dict num_inference_steps: Number of inference steps guidance_scale: Guidance scale value num_images: Number of images to generate output_format: Output format for images acceleration: Acceleration mode for generation speed Returns: Dict[str, Any]: Validated and normalized parameters Raises: ValueError: If any parameter is invalid zInvalid image_size 'z'. Must be one of: rwidthheightz8Custom image_size must contain 'width' and 'height' keysz3Custom image_size width and height must be integers@z3Custom image_size dimensions must be at least 64x64iz6Custom image_size dimensions must not exceed 2048x2048zEimage_size must be either a preset string or a dict with width/heightrdz8num_inference_steps must be an integer between 1 and 100rg?g4@z4guidance_scale must be a number between 0.1 and 20.0rrz-num_images must be an integer between 1 and 4rzInvalid output_format 'rzInvalid acceleration 'r) isinstancer/VALID_IMAGE_SIZESr2rdrnroVALID_OUTPUT_FORMATSVALID_ACCELERATION_MODES)rrrrrr validateds r(_validate_parametersrs2I*c""b . . .fJffSdffgg g", , J % % b * $ $ (B(BWXX X*W-s33 T:jQYFZ\_;`;` TRSS S g  # #z(';b'@'@RSS S g  % %H)=)D)DUVV V", ,`aaa )3 / /U3F3J3JNadgNgNgSTTT':I#$ nsEl 3 3Q~7K7K~`dOdOdOPPP"'"7"7I j# & &J*q..JNNHIII(Il000k=kkUikklll!.Io333m,mmSkmmnnn ,In r' image_urloriginal_promptc  td|td|ttt t tttd }tt|}| }|rd|vr|d}td| dd| d d|d | dd | d d d td St ddS#t$r(}t d|d Yd}~dSd}~wwxYw)a Upscale an image using FAL.ai's Clarity Upscaler. Uses the synchronous fal_client API to avoid event loop lifecycle issues when called from threaded contexts (e.g. gateway thread pool). Args: image_url (str): URL of the image to upscale original_prompt (str): Original prompt used to generate the image Returns: Dict[str, Any]: Upscaled image data or None if upscaling fails z(Upscaling image with Clarity Upscaler...z, ) rpromptupscale_factornegative_prompt creativity resemblancerrenable_safety_checkerrWimagez$Image upscaled successfully to %sx%srunknownrrerT)rerrupscaledrz"Upscaler returned invalid responseNzError upscaling image: %sexc_info)loggerinfoUPSCALER_DEFAULT_PROMPTUPSCALER_FACTORUPSCALER_NEGATIVE_PROMPTUPSCALER_CREATIVITYUPSCALER_RESEMBLANCEUPSCALER_GUIDANCE_SCALEUPSCALER_NUM_INFERENCE_STEPSUPSCALER_SAFETY_CHECKERr~UPSCALER_MODELgeterror Exception)rrupscaler_argumentshandlerresultupscaled_imagees r(_upscale_imager#s, >???#0EEOEE-7-/5#?%<    & (      g''#G_N KK>@R@RSZ\e@f@fhvhzhz|DFOiPiP Q Q Q%e,'++GQ77(,,Xq99 "1   LL= > > >4  0!d CCCtttttsD D( D(( E2EEr aspect_ratioseedc  |r&|nt}|tvr(td|tt}t|}||||||||dddddd} t j} td||dd|r:t|tr%t|dkrtd tjd s2ts$d } t!r| d z } t| t#|||||d } || d| d| d| d| dt$t&dd } |t|t(r|| d<tdtdt*td||td| dtd| dt-t*| }|}t j| z }|rd|vrtd|dg}|stdg}|D]}t|t2rd|vr|d|d d|d!dd"}t5|d|}|r||td#d|d$<|||std%t9d&|D}td't|||d|r|ddndd(}d| d)<t|| d*<|| d+<t:d,| t:tAj!|d-d.S#tD$r}t j| z }d/t|}t#d0|d1ddt|tI|j%d2}|| d3<|| d+<t:d,| t:tAj!|d-d.cYd}~Sd}~wwxYw)4a Generate images from text prompts using FAL.ai's FLUX 2 Pro model with automatic upscaling. Uses the synchronous fal_client API to avoid event loop lifecycle issues. The async API's global httpx.AsyncClient (cached via @cached_property) breaks when asyncio.run() destroys and recreates event loops between calls, which happens in the gateway's thread-pool pattern. Args: prompt (str): The text prompt describing the desired image aspect_ratio (str): Image aspect ratio - "landscape", "square", or "portrait" (default: "landscape") num_inference_steps (int): Number of denoising steps (1-50, default: 50) guidance_scale (float): How closely to follow prompt (0.1-20.0, default: 4.5) num_images (int): Number of images to generate (1-4, default: 1) output_format (str): Image format "jpeg" or "png" (default: "png") seed (Optional[int]): Random seed for reproducible results (optional) Returns: str: JSON string containing minimal generation results: { "success": bool, "image": str or None # URL of the upscaled image, or None if failed } z-Invalid aspect_ratio '%s', defaulting to '%s')rrrrrrrrNFr) parametersrsuccessimages_generatedgeneration_timez*Generating %s image(s) with FLUX 2 Pro: %sPz1Prompt is required and must be a non-empty stringr#z$FAL_KEY environment variable not setz' and managed FAL gateway is unavailablerrrrrrT) rrrrrrrsafety_tolerance sync_moderz5Submitting generation request to FAL.ai FLUX 2 Pro...z Model: %sz Aspect Ratio: %s -> %sz Steps: %sz Guidance: %srimagesz5Invalid response from FAL.ai API - no images returnedzNo images were generatedrerr)rerrz Using original image as fallbackrz%No valid image URLs returned from APIc3FK|]}|dddVdS)rFrN)r).0imgs r( z&image_generate_tool..s6YY3cggjRW>X>XYQYYYYYYr'z,Generated %s image(s) in %.1fs (%s upscaled))rrrrrimage_generate_toolr)indent ensure_asciizError generating image: z%sr)rrr error_typer)&lowerr0DEFAULT_ASPECT_RATIOASPECT_RATIO_MAPrwarningdatetimenowrrr/lenr2r$r%r)r rENABLE_SAFETY_CHECKERSAFETY_TOLERANCErn DEFAULT_MODELr~r total_secondsrdrappendsum_debuglog_callsaver]dumpsrrtyperj)rrrrrrraspect_ratio_lowerrdebug_call_data start_timemessagevalidated_paramsrWrrrrformatted_imagesroriginal_imagerupscaled_count response_datar error_msgs r(rr`sD:F_++--33555K_!111F Vjkkk1!"45J($#6,$*   O""&&((J{G @*fUXVXUXkZZZ RZ44 RFLLNN8K8Kq8P8PPQQ Q )$$ &(D(F(F &%: 0      4 5 5  $If  KLLL M=111 .0BJOOO M#34I#JKKK $&67G&HIII&     #,0022Z?NNPP V//TUU UHb)) 9788 8 < 8V>Vc`ttjdp tS)z Check if the FAL.ai API key is available in environment variables. Returns: bool: True if API key is set, False otherwise r#)boolr$r%r)r&r'r(check_fal_api_keyrs(  )$$F(D(F(F G GGr'cR tsdSddl}dS#t$rYdSwxYw)z Check if all requirements for image generation tools are met. Returns: bool: True if requirements are met, False otherwise FrNT)rrC ImportError)rCs r(#check_image_generation_requirementsr#sO  "" 5 t uus &&c4tS)z Get information about the current debug session. Returns: Dict[str, Any]: Dictionary containing debug session information )rget_session_infor&r'r(get_debug_session_infor7s  " " $ $$r'__main__u@🎨 Image Generation Tools Module - FLUX 2 Pro + Auto Upscalingz<============================================================u(❌ FAL_KEY environment variable not setz7Please set your API key: export FAL_KEY='your-key-here'zGet API key at: https://fal.ai/u✅ FAL.ai API key foundu ✅ fal_client library availableu ❌ fal_client library not foundz&Please install: pip install fal-clientu-🛠️ Image generation tools ready for use!u🤖 Using model: u🔍 Auto-upscaling with: z (zx)u&🐛 Debug mode ENABLED - Session ID: z9 Debug logs will be saved to: ./logs/image_tools_debug_z.jsonu?🐛 Debug mode disabled (set IMAGE_TOOLS_DEBUG=true to enable)z Basic usage:z7 from image_generation_tool import image_generate_toolz import asyncior-z async def main():z2 # Generate image with automatic 2x upscalingz) result = await image_generate_tool(zD prompt='A serene mountain landscape with cherry blossoms',z% image_size='landscape_4_3',z num_images=1z )z print(result)z asyncio.run(main())z Supported image sizes:z - z5 - Custom: {'width': 512, 'height': 768} (if needed)z Acceleration modes:z Example prompts:zH - 'A candid street photo of a woman with a pink bob and bold eyeliner'zE - 'Modern architecture building with glass facade, sunset lighting'z= - 'Abstract art with vibrant colors and geometric patterns'z? - 'Portrait of a wise old owl perched on ancient tree branch'z; - 'Futuristic cityscape with flying cars and neon lights'z Debug mode:z # Enable debug loggingz export IMAGE_TOOLS_DEBUG=truez= # Debug logs capture all image generation calls and resultsz5 # Logs saved to: ./logs/image_tools_debug_UUID.json)registryimage_generatea Generate high-quality images from text prompts using FLUX 2 Pro model with automatic 2x upscaling. Creates detailed, artistic images that are automatically upscaled for hi-rez results. Returns a single upscaled image URL. Display it using markdown: ![description](URL)objectstringzJThe text prompt describing the desired image. Be detailed and descriptive.)r descriptionzlThe aspect ratio of the generated image. 'landscape' is 16:9 wide, 'portrait' is 16:9 tall, 'square' is 1:1.)renumrdefault)rr)r propertiesrequired)namerrc |dd}|stjddiSt||ddddd d d S) Nrr-rz'prompt is required for image generationrr r r rr)rrrrrrr)rr]rr)argskwrs r(_handle_image_generatersn XXh # #F Pz7$MNOOO XXnk::    r' image_genu🎨)rtoolsetschemarcheck_fn requires_envis_asyncemoji)r)Qrmr]loggingr$r threadingr{typingrrrr urllib.parserrCtools.debug_helpersrtools.managed_tool_gatewayr tools.tool_backend_helpersr getLoggerrjrrrDEFAULT_NUM_INFERENCE_STEPSDEFAULT_GUIDANCE_SCALEDEFAULT_NUM_IMAGESDEFAULT_OUTPUT_FORMATrrrlistkeysVALID_ASPECT_RATIOSrrrrrrrrrrrrrrtruLockrsr)r/r4r6rxr~rnrorrrrrrrprint api_availableexitractive session_idsizemodetools.registryrIMAGE_GENERATE_SCHEMArregisterr&r'r(rs:   ------------"""""",,,,,,CCCCCCAAAAAA  8 $ $$ " " d+002233+>K!666 m-@ A A A!)9>++555#c#c####L L L L L L L L ^###( s tCH~    * FFc4S>)*FFF F  F  F #s(^FFFFR:c:C:DcN::::~-:2(.vGvG vGvGvG vG  vG  vG 3-vG vGvGvGvGrH4HHHHT(%S#X%%%% z E LMMM E(OOO&%''M * 8999 GHHH /000 Q ())) 01111  0111 6777 Q  E 9::: E .} . ./// E L~ L L L L LMMM}Q Jv7HJJKKK b&J[bbbcccc OPPP E  E CDDD E  E"III E     E >??? E 5666 E PQQQ E 1222 E "### E) E     E !""" E $%%%! mTmm E ABBB E !"""( mTmm E  E TUUU E QRRR E IJJJ E KLLL E GHHH E/ E $%%% E +,,, E IJJJ E ABBB $##### b!k !;;; N&    J,      " 0        s G&HH