+ ikxRt^RIt^RIt^RIt^RIt^RIt^RIt^RIHtH t H t H t ^RI H t ^RIt^RIHt^RIHt^RIHt]P*!]4tRtRt^2tR t^tR tR tR tRR RRRR/t ]!!] PE44t#Rt$^t%R t&Rt'Rt(Rt)Rt*^t+^t,.RrOt-RR .t..RsOt/]!RRR7t0Rs1Rs2]Pf!4t4Rt5RRlt6!RR4t7R t8R!R"lt9RtR#R$llt:R%R&lt;]]]]]R3R'R(lltR-R.lt?]R/8XEd]@!R04]@!Ru4]=!4tA]A'g"]@!R14]@!R24]@!R34]B!^4M]@!R44^RIt]@!R54]@!R84]@!R9] 24]@!R:]$ R;]% R<24]0P'd-]@!R=]0P 24]@!R>]0P R?24M]@!R@4]@!RA4]@!RB4]@!RC4]@!RD4]@!RE4]@!RF4]@!RG4]@!RH4]@!RI4]@!RJ4]@!RK4]@!RL4]@!RM4]@!RN4]-FtF]@!RO]F 24K ]@!RP4]@!RQ4]/FtG]@!RO]G 24K ]@!RR4]@!RS4]@!RT4]@!RU4]@!RV4]@!RW4]@!RX4]@!RY4]@!RZ4]@!R[4]@!R\4^R]IHHItIR^R_R`RaRbRcRdReRfRcRgR`Rh/RiRcRgRj.RvOR`RkRlR//RmRf.//tJRntK]IP!R_Ro]J]K]>.R RpRq7R# ]Cd]@!R64]@!R74]B!^4ELi;i)wa 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 landscape@pngF5landscape_16_9square square_hdportrait portrait_16_9zfal-ai/clarity-upscalerz"masterpiece, best quality, highresz.(worst quality, low quality, normal quality:2)gffffff?g333333?jpeg image_toolsIMAGE_TOOLS_DEBUG)env_varcT\P!R4'dR#\R4#)zOReturn managed fal-queue gateway config when direct FAL credentials are absent.FAL_KEYNz fal-queue)osgetenvr8/home/ubuntu/hermes-agent/tools/image_generation_tool.py_resolve_managed_fal_gatewayrZs yy ' 44rc0V^8dQhR\R\/#)queue_run_originreturnstr)formats"r __annotate__r&as##c#c#rc\T;'gR4P4PR4pV'g \R4hV R2#)/z$Managed FAL queue origin is required)r$striprstrip ValueError)r!normalized_origins& r_normalize_fal_queue_url_formatr.asG,22399;BB3G ?@@  ""rc da]tRt^htoRtV3RlRltRRRRRRR RR RR R/V3R lR lltRtVtR#)_ManagedFalSyncClientzPSmall per-instance wrapper around fal_client.SyncClient for managed queue hosts.c&<V^8dQhRS[RS[/#)r keyr!r#)r% __classdict__s"rr&"_ManagedFalSyncClient.__annotate__ksoosocorc\\RR4pVf \R4h\\RR4pVf \R4h\V4VnV!VR7Vn\VP RR4Vn\VRR4Vn\VRR4Vn\VR R4Vn \VR R4Vn \VR R4Vn \VR R4Vn VP f \R 4hVPeVPf \R4hVPf \R4hR#) SyncClientNz>fal_client.SyncClient is required for managed FAL gateway modeclientz:fal_client.client is required for managed FAL gateway mode)r2_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 RuntimeErrorr._queue_url_format _sync_client _http_clientr9r:_request_handle_class_add_hint_header_add_priority_header_add_timeout_header)selfr2r!sync_client_class client_modules&$$ r__init___ManagedFalSyncClient.__init__ks=#J dC  $_` ` Hd;  [\ \!@AQ!R-#6#D$5$5y$G$+M;QSW$X!!(8KT!R%,]/  / sm / c]/ / $sCx.)/  c5j 12/ rc VPV,p V'd V RVPR4,, p VeV R\RV/4,, p \T;'g/4p Ve VPeVP WJ4Ve+VP f \ R4hVP Wj4Ve+VPf \ R4hVPW4VPVPRV V\VPRR4V R7p VPV 4V P4p VPV R ,V R ,V R ,V R ,VPR 7#)r)? fal_webhookzGfal_client.client.add_priority_header is required for priority requestszEfal_client.client.add_timeout_header is required for timeout requestsPOSTdefault_timeoutg^@)jsontimeoutrR request_id response_url status_url cancel_url)r`rarbrcr7)rBlstriprdictrFrGrArHr9rDr?rCr:r^rE) rIrUrVrNrOrPrQrRrSurlrequest_headersresponsedatas &&&$$$$$$ rsubmit_ManagedFalSyncClient.submitsn$${2  3S)) )C  " 3M;#?@@ @Cw}}"-   5 5 A  ! !$ 8  ((0"#lmm  % %h @  $''/"#jkk  $ $] D,,     D--/@%H# -  x(}}))L)n-L)L)$$ *  r) rFrGrHrDr9rBr:rErC) __name__ __module__ __qualname____firstlineno____doc__rLrj__static_attributes____classdictcell__)r3s@rr0r0hs_Zoo4/  / # / &*/ / -1/ 6:/ / / rr0cBVPPR4VP3p\;_uu_4\e\ V8Xd\uuRRR4#\ VPVPR7sVs\uuRRR4# +'giR#;i)zQReuse the managed FAL client so its internal httpx.Client is not leaked per call.r)N)r2r!)gateway_originr+nous_user_token_managed_fal_client_lock_managed_fal_client_managed_fal_client_configr0)managed_gateway client_configs& r_get_managed_fal_clientr{s &&--c2''M " !  */I]/Z& " !4//,;; &3"" " ! ! !sB (B  B cRV^8dQhR\R\\\3,/#)r modelrVr$rr)r%s"rr&r&s"  s tCH~ rcR\\P!44/p\4pVf\P !WVR7#\ V4pVP VVVR7#)zKSubmit a FAL request using direct credentials or the managed queue gateway.zx-idempotency-key)rVrR)r$uuiduuid4rr@rjr{)r}rVrgrymanaged_clients&& r_submit_fal_requestrsc*C ,=>O24O  _UU,_=N    ! rcV^8dQhR\\\\\3,3,R\R\R\R\R\R\\\ 3,/#)r image_sizenum_inference_stepsguidance_scale num_images output_format accelerationr")rr$rrWrXr)r%s"rr&r&sqFFc4S>)*FFF F  F  F #s(^Frc/p\V\4'd&V\9d\RV R\ 24hWR&M\V\4'dRV9gRV9d \R4h\VR,\ 4'd\VR,\ 4'g \R4hVR,^@8gVR,^@8d \R4hVR,R 8gVR,R 8d \R 4hWR&M \R 4h\V\ 4'dV^8gV^d8d \R 4hWR &\V\ \ 34'dVR8gVR8d \R4h\ V4VR&\V\ 4'dV^8gV^8d \R4hW6R&V\9d\RV R\ 24hWFR&V\9d\RV R\ 24hWVR&V#)a 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 integersz3Custom 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/heightz8num_inference_steps must be an integer between 1 and 100rg?g4@z4guidance_scale must be a number between 0.1 and 20.0rz-num_images must be an integer between 1 and 4rzInvalid output_format 'rzInvalid acceleration 'r) isinstancer$VALID_IMAGE_SIZESr,rerWrXVALID_OUTPUT_FORMATSVALID_ACCELERATION_MODES)rrrrrr validateds&&&&&& r_validate_parametersrs2I*c"" . .3J?  01O3DE o 7 - / 5 !#? #%<   & (   g'#G_N KK>@R@RSZ\e@fhvhzhz|DFOiP Q~e,++GQ7.,,Xq9D /   LL= >  0!d CsC4D7D D<D77D<cV^8dQhR\R\R\R\R\R\R\\,R\/#) r r aspect_ratiorrrrseedr")r$rWrXr)r%s"rr&r&`spvGvG vGvGvG vG  vG  vG 3-vG vGrc V'dVP4P4M\pV\9d"\P RV\4\p\V,pRRVRVRVRVRVRVR VR V/R R R RR^R^/p \ P P4p \PRW@R,4V'd5\V\4'd\VP44^8Xd \R4h\P!R4'g7\4'g'Rp \!4'd V R, p \V 4h\#WW4VR4p RVP4RV R,RV R,RV R,RV R,R V R ,R\$R\&RR/ p Ve\V\(4'dWmR &\PR4\PR\*4\PRWx4\PRV R,4\PR V R,4\-\*V R!7pVP/4p\ P P4V , P14pV'dR"V9d \R#4hVP/R".4pV'g \R$4h.pVFp\V\24'gKR%V9gK$R%VR%,R&VP/R&^4R'VP/R'^4/p\5VR%,VP44pV'dVP7V4K\P R(4RVR)&VP7V4K V'g \R*4h\9R+V44p\PR,\V4VV4R RR-V'dV^,R%,MR /pRV R &\V4V R&VV R&\:P=R.V 4\:P?4\@PB!V^RR/7# \Ddp\ P P4T , P14pR0\T4 2p\PGR1TRR27R RR-R R \T4R3\IT4PJ/pTT R &TT R&\:P=R.T 4\:P?4\@PB!T^RR/7uR p?#R p?ii;i)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' parametersrrrrrrrrrNsuccessFimages_generatedgeneration_timez*Generating %s image(s) with FLUX 2 Pro: %s:NPNz1Prompt is required and must be a non-empty stringrz$FAL_KEY environment variable not setz' and managed FAL gateway is unavailablenonersafety_tolerance sync_modeTz5Submitting 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 generatedrfrrz Using original image as fallbackrz%No valid image URLs returned from APIc3X"TF qPRR4'gK^xK" R#5i)rFN)r).0imgs& r &image_generate_tool..s Y*:3ggjRW>XQQ*:s* *z,Generated %s image(s) in %.1fs (%s upscaled)rimage_generate_tool)indent ensure_asciizError generating image: z%sr error_type)&lowerr*DEFAULT_ASPECT_RATIOASPECT_RATIO_MAPrwarningdatetimenowrrr$lenr,rrrr rENABLE_SAFETY_CHECKERSAFETY_TOLERANCErW DEFAULT_MODELrr total_secondsrerappendsum_debuglog_callsaver^dumpsrrtyperl)rrrrrrraspect_ratio_lowerrdebug_call_data start_timemessagevalidated_paramsrVrrrrformatted_imagesroriginal_imagerupscaled_count response_datar error_msgs&&&&&&& rrr`sD:F++-335K_!11F Vjk1!"45J  f L * !#6 n * ] D   5A1O"""&&(J{G @*UXkZZ44FLLN8Kq8PPQ Q )$$(D(F(F #%:  0      4 5 5 $f  KL M=1 .0BO M#34I#JK $&67G&HI&   #,,002Z?NNP/TU UHb)78 8C#t$$#3u:SWWWa0cggh2""0E FLLN!K!$++N;NN#EF16N:.$++N;%( DE EY*:YY BCHXDY[jlz{ t 3C%a(/ &* ".12B.C*+-<)* -? zz-FF G#,,002Z?NNP.s1vh7  T9t 4 u T SV $q'**  $- -<)*-? zz-FF%GsA!A?Q#!F5Q#%Q#Q# B!Q#,B6Q## U.C T?9U?Uc$V^8dQhR\/#r r"bool)r%s"rr&r&sHH4Hrcd\\P!R4;'g \44#)z Check if the FAL.ai API key is available in environment variables. Returns: bool: True if API key is set, False otherwise r)rrrrrrrcheck_fal_api_keyrs$  )$FF(D(F GGrc$V^8dQhR\/#rr)r%s"rr&r&#sTrcX\4'gR#^RIpR# \dR#i;i)z Check if all requirements for image generation tools are met. Returns: bool: True if requirements are met, False otherwise FNT)rr@ ImportError)r@s r#check_image_generation_requirementsr#s-  ""  s ))cFV^8dQhR\\\3,/#r)rr$r)r%s"rr&r&7s%%S#X%rc*\P4#)z Get information about the current debug session. Returns: Dict[str, Any]: Dictionary containing debug session information )rget_session_inforrrget_debug_session_infor7s  " " $$r__main__u@🎨 Image Generation Tools Module - FLUX 2 Pro + Auto Upscalingu(❌ 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)registrynameimage_generate descriptiona 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)rrobject propertiesrstringzJThe text prompt describing the desired image. Be detailed and descriptive.renumzlThe aspect ratio of the generated image. 'landscape' is 16:9 wide, 'portrait' is 16:9 tall, 'square' is 1:1.defaultrequiredc VPRR4pV'g\P!RR/4#\VVPRR4^2R^RRR 7#) rr(rz'prompt is required for image generationrr r r N)rrrrrrr)rr^rr)argskwrs&, r_handle_image_generatersW XXh #F zz7$MNOO XXnk:  r image_genu🎨)rtoolsetschemarcheck_fn requires_envis_asyncemoji)rr portrait_4_3r landscape_4_3r)rregularhigh)rz<============================================================)r rr)Mrpr^loggingrr threadingrtypingrrrr urllib.parserr@tools.debug_helpersrtools.managed_tool_gatewayrtools.tool_backend_helpersr getLoggerrlrrrDEFAULT_NUM_INFERENCE_STEPSDEFAULT_GUIDANCE_SCALEDEFAULT_NUM_IMAGESDEFAULT_OUTPUT_FORMATrrrlistkeysVALID_ASPECT_RATIOSrrrrrrrrrrrrrrwrxLockrvrr.r0r{rrrrrrrprint api_availableexitractive session_idsizemodetools.registryrIMAGE_GENERATE_SCHEMArregisterrrrr,sF:   --",CA   8 $$ " ! k +0023+>K!6 m-@ A!$>>+5#L L ^#( FR:~-:2(.vGrH(% z LM (O&'M  89 GH /0 Q () 01  9: }o ./ &~&6b8I LM}}} 6v7H7H6IJK I&J[J[I\\abc OP  CD  "I   >? 56 PQ 12 "# )   !" $%! TFm" AB !"( TFm)  TU QR IJ KL GH / $% +, IJ AB $ b k ; N;    XJ,     " 0     01 67 Qs LL98L9