i^UdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z ddl m Z ddlZddlmZddlmZddlmZmZmZejeZdd d d d d dddddddhddd ddddd d ddddddddddd hd!dd d"d#d$d%d d dddd&ddddd'hd(dd d)d*d+d,d-d.d/d0ddddd1d2hd3dd d4d5d6d7d8d9d:d;dddd d?d@dAdBd dCddDddedkre?de?de=s,e?de?de?de@de?d ddlZe?dn#eA$re?de@dwxYwe7\ZBZCe?deCDdeBdeBde?deCDdddeCDdde?deCDdrdnde?deED]O\ZFZGeFeBkrdndZHe?deFddeGDddddeGDddeHPe)jIre?de)jJddlKmLZLmMZMdddddddeNededddxgddZOdZPdxed-efdZQdZReLjSddeOeRe>gdddS)u Image Generation Tools Module Provides image generation via FAL.ai. Multiple FAL models are supported and selectable via ``hermes tools`` → Image Generation; the active model is persisted to ``image_gen.model`` in ``config.yaml``. Architecture: - ``FAL_MODELS`` is a catalog of supported models with per-model metadata (size-style family, defaults, ``supports`` whitelist, upscaler flag). - ``_build_fal_payload()`` translates the agent's unified inputs (prompt + aspect_ratio) into the model-specific payload and filters to the ``supports`` whitelist so models never receive rejected keys. - Upscaling via FAL's Clarity Upscaler is gated per-model via the ``upscale`` flag — on for FLUX 2 Pro (backward-compat), off for all faster/newer models where upscaling would either hurt latency or add marginal quality. Pricing shown in UI strings is as-of the initial commit; we accept drift and update when it's noticed. N)AnyDictOptionalUnion) urlencode) DebugSession)resolve_managed_tool_gateway)fal_key_is_configuredmanaged_nous_tools_enabledprefers_gatewayzFLUX 2 Klein 9Bz<1szFast, crisp textz $0.006/MPimage_size_presetlandscape_16_9 square_hd portrait_16_9) landscapesquareportraitpngF)num_inference_steps output_formatenable_safety_checker>seedprompt image_sizerrr) displayspeed strengthsprice size_stylesizesdefaultssupportsupscalez FLUX 2 Proz~6szStudio photorealismz$0.03/MP2g@5T)rguidance_scale num_imagesrrsafety_tolerance sync_mode> rrr+rr)rr(r*rrz Z-Image Turboz~2szBilingual EN/CN, 6Bz $0.005/MP)rr)rrenable_prompt_expansion>rrrr)rrrr-z$Nano Banana Pro (Gemini 3 Pro Image)z~8sz-Gemini 3 Pro, reasoning depth, text renderingz$0.15/image (1K) aspect_ratioz16:9z1:1z9:161K)r)rr* resolution> rrr+r)r0r.rr*enable_web_searchlimit_generationsz GPT Image 1.5z~15szPrompt adherencez $0.034/image gpt_literal 1536x1024 1024x1024 1024x1536medium)qualityr)r>rr8r+ backgroundrr)rz GPT Image 2z~20sz3SOTA text rendering + CJK, world-aware photorealismu$0.04–0.06/image landscape_4_3 portrait_4_3>rr8r+rr)rz Ideogram V3z~5szBest typographyz$0.03-0.09/imageBALANCEDAUTO)rendering_speed expand_promptstyle>rr@rrr?r>zRecraft V4 Proz'Design, brand systems, production-readyz $0.25/imager>colorsrrbackground_colorrz Qwen Imagez~12szLLM-based, complex textz$0.02/MPg@regular)rr(r)r acceleration> rrr+rr)rErr(r) fal-ai/flux-2/klein/9bzfal-ai/flux-2-prozfal-ai/z-image/turbozfal-ai/nano-banana-prozfal-ai/gpt-image-1.5zfal-ai/gpt-image-2zfal-ai/ideogram/v3z#fal-ai/recraft/v4/pro/text-to-imagezfal-ai/qwen-image FAL_MODELSrFrzfal-ai/clarity-upscalerz"masterpiece, best quality, highresz.(worst quality, low quality, normal quality:2)gffffff?g333333? image_toolsIMAGE_TOOLS_DEBUG)env_varc^trtdsdStdS)zsReturn managed fal-queue gateway config when the user prefers the gateway or direct FAL credentials are absent. image_genNz fal-queue)r r r @/home/ubuntu/.hermes/hermes-agent/tools/image_generation_tool.py_resolve_managed_fal_gatewayrRBs4{'C'Ct ' 4 44rPqueue_run_originreturnct|pdd}|std|dS)N/z$Managed FAL queue origin is required)strstriprstrip ValueError)rSnormalized_origins rQ_normalize_fal_queue_url_formatr]JsW,23399;;BB3GG A?@@@ " " ""rPceZdZdZdedefdZdddddddded eeefd ed eed eed edeeeefdee e e ffdZ dS)_ManagedFalSyncClientzPSmall per-instance wrapper around fal_client.SyncClient for managed queue hosts.keyrScttdd}|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)r`_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_clientrerf_request_handle_class_add_hint_header_add_priority_header_add_timeout_header)selfr`rSsync_client_class client_modules rQ__init__z_ManagedFalSyncClient.__init__Tsg#J dCC  $_`` ` Hd;;  [\\ \!@AQ!R!R--#666#D$5y$GG$+M;QSW$X$X!!(8KT!R!R%,]"%),059/ / / / S>/  / sm / c]/ / $sCx.)/  c5j 12/ / / / / / rPr_c|jd|jf}t5tt |krtcdddSt |j|ja|atcdddS#1swxYwYdS)zQReuse the managed FAL client so its internal httpx.Client is not leaked per call.rWN)r`rS)gateway_originrZnous_user_token_managed_fal_client_lock_managed_fal_client_managed_fal_client_configr_)managed_gateway client_configs rQ_get_managed_fal_clientrs &--c22'M " # #  */I]/Z/Z& # # # # # # # #4/,;   &3"" # # # # # # # # # # # # # # # # # #sA?#A??BBmodelrczdttji}t}|t j|||St |} ||||S#t$r>}t|}|'d|cxkrdkrnntd|d|d|d}~wwxYw) zKSubmit a FAL request using direct credentials or the managed queue gateway.zx-idempotency-keyN)rr}iiz*Nous Subscription gateway rejected model 'z' (HTTP u). This model may not yet be enabled on the Nous Portal's FAL proxy. Either: • Set FAL_KEY in your environment to use FAL.ai directly, or • Pick a different model via `hermes tools` → Image Generation.) rXuuiduuid4rRrlrr Exception_extract_http_statusr[)rrrrmanaged_clientexcstatuss rQ_submit_fal_requestrs*C ,=,=>O244O )_UUUU,_==N$$ #%    &c**  #"5"5"5"5#"5"5"5"5"5YUYYYYY    sA22 B:<9B55B:rct|dd}|(t|dd}t|tr|St|dd}t|tr|SdS)uReturn an HTTP status code from httpx/fal exceptions, else None. Defensive across exception shapes — httpx.HTTPStatusError exposes ``.response.status_code`` while fal_client wrappers may expose ``.status_code`` directly. rN status_code)rk isinstancer)rrrs rQrrspsJ--H=$77 fc " " M S- . .F&# 4rPcd} ddlm}|}t|tr|dnd}t|tr>|d}t|t r|}n2#t$r%}t d|Yd}~nd}~wwxYw|s'tj dd}|stttfS|tvr:td |ttttfS|t|fS) zResolve the active FAL model from config.yaml (primary) or default. Returns (model_id, metadata_dict). Falls back to DEFAULT_MODEL if the configured model is unknown (logged as a warning). rVr load_configrNNrz.Could not load image_gen.model from config: %sFAL_IMAGE_MODELz4Unknown FAL model '%s' in config; falling back to %s)hermes_cli.configrrrgetrXrYrloggerdebugosgetenv DEFAULT_MODELrGwarning)model_idrcfgimg_cfgrawrs rQ_resolve_fal_modelrsf H L111111kmm*4S$*?*?I#''+&&&T gt $ $ '++g&&C#s## '99;; LLL EsKKKKKKKKL <9.3399;; 8j777z!! B m   j777 Z) ))sBB CB>>Crrr overridesc\ t|}|d}|d}|pt}||vrt}t |di} |pd| d<|dvr ||| d<n$|dkr ||| d<nt d ||t|tr|| d <|r!| D] \} } | | | | < |d  fd | DS)a)Build a FAL request payload for `model_id` from unified inputs. Translates aspect_ratio into the model's native size spec (preset enum, aspect-ratio enum, or GPT literal string), merges model defaults, applies caller overrides, then filters to the model's ``supports`` whitelist. r r!r"rVr)r r3rr.zUnknown size_style: Nrr#c$i|] \}}|v || SrOrO).0kvr#s rQ z&_build_fal_payload..3s$ > > >TQX Aq rP) rGDEFAULT_ASPECT_RATIOlowerrYrrr[rritems) rrr.rrmetar r!aspectpayloadrrr#s @rQ_build_fal_payloadr sX h Dl#J ME22 9 9 ; ; A A C CF U%"488J#;#;< >>??? JtS11OO%%  DAq} JH > > > >W]]__ > > >>rP 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)zUpscale an image using FAL.ai's Clarity Upscaler. Returns upscaled image dict, or None on failure (caller falls back to the original image). z(Upscaling image with Clarity Upscaler...z, ) rrupscale_factornegative_prompt creativity resemblancer(rrrimagez$Image upscaled successfully to %sx%swidthunknownheightrrT)rrrupscaledrz"Upscaler returned invalid responseNzError upscaling image: %sexc_info)rinfoUPSCALER_DEFAULT_PROMPTUPSCALER_FACTORUPSCALER_NEGATIVE_PROMPTUPSCALER_CREATIVITYUPSCALER_RESEMBLANCEUPSCALER_GUIDANCE_SCALEUPSCALER_NUM_INFERENCE_STEPSUPSCALER_SAFETY_CHECKERrUPSCALER_MODELrerrorr)rrupscaler_argumentshandlerresultupscaled_imagees rQ_upscale_imager9sr % >???#0EEOEE-7-/5#?%<   &n@RSSS  g''#G_N KK6""7I66""8Y77    &e,'++GQ77(,,Xq99 "1    9:::t  0!d CCCtttttsD D( D(( E2EErr(r)rc  t\}}||||||||dddddd} tj} |r:t|tr%t |dkrtdts2ts$d} tr| dz } t| |pt } | tvr(td |tt} i} ||| d <||| d <||| d <||| d <t!||| || }td|d|||ddt'||}|}tj| z }|rd|vrtd|dg}|stdt+|dd}g}|D]}t|t,rd|vs|d|dd|ddd}|rZt/|d|}|r||tdd|d<|||stdt3d|D}tdt ||||d |r|ddndd!}d | d"<t || d#<|| d$<t4d%| t4t;j|d&d'S#t>$r}tj| z }d(t |}t d)|d *ddt |tC|j"d+}|| d,<|| d$<t4d%| t4t;j|d&d'cYd}~Sd}~wwxYw)-aGenerate an image from a text prompt using the configured FAL model. The agent-facing schema exposes only ``prompt`` and ``aspect_ratio``; the remaining kwargs are overrides for direct Python callers and are filtered per-model via the ``supports`` whitelist (unsupported overrides are silently dropped so legacy callers don't break when switching models). Returns a JSON string with ``{"success": bool, "image": url | None, "error": str, "error_type": str}``. )rr.rr(r)rrNFr)r parametersrsuccessimages_generatedgeneration_timez1Prompt is required and must be a non-empty stringz$FAL_KEY environment variable not setz' and managed FAL gateway is unavailablez-Invalid aspect_ratio '%s', defaulting to '%s'rr(r)r)rru,Generating image with %s (%s) — prompt: %srPrimagesu7Invalid response from FAL.ai API — no images returnedzNo images were generatedr$rrr)rrrz1Using original image as fallback (upscale failed)rz%No valid image URLs returned from APIc3DK|]}|ddVdS)rr&N)r)rimgs rQ z&image_generate_tool..s3RR3cggj>Q>QRQRRRRRRrPz3Generated %s image(s) in %.1fs (%s upscaled) via %sT)rrrrrimage_generate_toolrH)indent ensure_asciizError generating image: z%srrrr error_typer)#rdatetimenowrrXlenrYr[r rRr rrVALID_ASPECT_RATIOSrrrrrr total_secondsboolrrappendsum_debuglog_callsaverdumpsrrtyper)rr.rr(r)rrrrdebug_call_data start_timemessage aspect_lcrrrrrrshould_upscaleformatted_imagesroriginal_imagerupscaled_count response_datar error_msgs rQrrjsc&())NHd(#6,$*  O""&&((JnG RZ44 RFLLNN8K8Kq8P8PPQQ Q%'' &+G+I+I &AEE!E!c |dd}|stdS|dt}t||}||St ||S)NrrVz'prompt is required for image generationr.r,)rr%rr=r)argskwrr. dispatcheds rQ_handle_image_generaterBs{ XXh # #F ECDDD88N,@AAL.flCCJ !   rPrNu🎨)r/toolsetschemarcheck_fn requires_envis_asyncemoji)Trrloggingrr threadingrtypingrrrr urllib.parserrltools.debug_helpersrtools.managed_tool_gatewayr tools.tool_backend_helpersr r r getLoggerrrrGrX__annotations__rrrrrrrrrrrrrrrLockrrRr]r_rr BaseExceptionrrtuplerrrrrrrrprint SystemExitrrrrrmidmmarkeractive session_idtools.registryr$r%listIMAGE_GENERATE_SCHEMAr3r=rBregisterrOrPrQr`s *   ------------"""""",,,,,,CCCCCC  8 $ $4%'))!'  $%"%*     ). *))!'  $&!"%* #     38#*))!'  $%"%*',      /4:D#$  " #      16#'#$!#   "      -2!J% *(!&   "      =B!&#))!'  *!     ).$>))!'   $U    ',,, .))!'  $&!"%      -Q`)`) Dd38n$ %```F) "9 +>K! m-@ A A A!)9>++ 555#c#c####L L L L L L L L ^###(stCH~>m * *E * * * *L-*. (?(?(? (?(? 3- (? S#X' (?  #s(^ (?(?(?(?\+c+C+HT#s(^>> -...jmm E $%%% 01111  KLLLjmm('))NHd E LH = = L L L L LMMM E Ttxx-- T TDHHWc