o i'@sddlmZddlmZddlmZmZmZddlm Z m Z m Z m Z Gddde eZGddde ZGd d d e ZGd d d e ZGd dde ZGddde ZGddde ZGddde ZGddde eZGddde ZdS)) annotations)Enum)AnyDictOptional) BaseModelFieldconfloatconintc@seZdZdZdZdS)BFLOutputFormatpngjpegN)__name__ __module__ __qualname__r r rr6/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/bfl.pyr sr c@seZdZUedddZded<edddZded <edd dZd ed <edd dZded<edddZ ded<edddZ ded<edddZ ded<edddZ ded<edddZ ded<edddZded<eejd d!gd"Zd#ed$<edd%dZded&<dS)'BFLFluxExpandImageRequest.The description of the changes you want to make. This text guides the expansion process, allowing you to specify features, styles, or modifications for the expanded areas. descriptionstrpromptNwWhether to perform upsampling on the prompt. If active, automatically modifies the prompt for more creative generation.Optional[bool]prompt_upsampling#The seed value for reproducibility. Optional[int]seedz2Number of pixels to expand at the top of the imagezconint(ge=0, le=2048)topz5Number of pixels to expand at the bottom of the imagebottomz8Number of pixels to expand at the left side of the imageleftz9Number of pixels to expand at the right side of the imageright0Number of steps for the image generation processconint(ge=15, le=50)steps2Guidance strength for the image generation processconfloat(ge=1.5, le=100)guidance{Tolerance level for input and output moderation. Between 0 and 6, 0 being most strict, 6 being least strict. Defaults to 2.Optional[conint(ge=0, le=6)]safety_tolerance>Output format for the generated image. Can be 'jpeg' or 'png'.r rexamplesOptional[BFLOutputFormat] output_formatzAA Base64-encoded string representing the image you wish to expandimage)rrrrr__annotations__rrrr r!r"r%r(r,r r r1r2rrrrrs&  rc@seZdZUedddZded<edddZded <edd dZd ed <edd dZded<edddZ ded<edddZ ded<ee j ddgdZ ded<edddZded<edddZded<dS) BFLFluxFillImageRequest.rrrrNrrrrrrr#r$r%r&r'r(r)r*r+r,r-r r.r0r1zeA Base64-encoded string representing the image you wish to modify. Can contain alpha mask if desired.r2zNA Base64-encoded string representing the mask of the areas you with to modify.mask)rrrrrr3rrr%r(r,r r r1r2r5rrrrr4#s  r4c@seZdZUedddZded<edddZded <edd dZd ed <ed ddZded<edddZ ded<edddZ ded<ee j ddgdZ ded<edddZded<dS) BFLFluxProGenerateRequest.%The text prompt for image generation.rrrNrrrrrrzAWidth of the generated image in pixels. Must be a multiple of 32.zconint(ge=256, le=1440)widthzBHeight of the generated image in pixels. Must be a multiple of 32.heightr)r*r+r,r-r r.r0r1(Optional image to remix in base64 format Optional[str] image_prompt)rrrrrr3rrr9r;r,r r r1r>rrrrr65s  r6c@sFeZdZUedZded<edddZded<ed ddZded <ed Zd ed <ed Z ded<ed ddZ ded<ed ddZ ded<ed ddZ ded<ed ddZ ded<ed ddZded<ed ddZded<ed ddZded<ed ddZded<ed ddZded<edddddZd ed<ed d!dZded"<d S)#Flux2ProGenerateRequest.rrr8zMust be a multiple of 32.rintr9r:r;Nz int | Nonerz bool | Nonerz2Base64 encoded image for image-to-image generationz str | None input_image input_image_2 input_image_3 input_image_4 input_image_5 input_image_6 input_image_7 input_image_8 input_image_9zKTolerance level for input and output moderation. Value 0 being most strict.rrgeler,r r-r1)rrrrrr3r9r;rrrArBrCrDrErFrGrHrIr,r1rrrrr?Is* r?c@seZdZUedddZded<edddZded <edd dZd ed <edd dZded<edddZ ded<edddZ ded<ee j ddgdZ ded<edddZded<edddZded <dS)! BFLFluxKontextProGenerateRequest.z+The text prompt for what you wannt to edit.rrrNzImage to edit in base64 formatr=rArrrr&zconfloat(ge=0.1, le=99.0)r(r#zconint(ge=1, le=150)r%z{Tolerance level for input and output moderation. Between 0 and 2, 0 being most strict, 6 being least strict. Defaults to 2.zOptional[conint(ge=0, le=2)]r,r-r r.r0r10Aspect ratio of the image between 21:9 and 9:21. aspect_ratiorrr)rrrrrr3rArr(r%r,r r r1rQrrrrrrN`s  rNc@seZdZUedddZded<edddZded <edd dZd ed <edd dZded<edddZ ded<ee j ddgdZ ded<edddZ ded<edddZded<edddZded<dS) BFLFluxProUltraGenerateRequest.r7rrrNrrrrrrrPr=rQr)r*r+r,r-r r.r0r1z5Generate less processed, more natural-looking images.rawr<r>z.Blend between the prompt and the image prompt.z"Optional[confloat(ge=0.0, le=1.0)]image_prompt_strength)rrrrrr3rrrQr,r r r1rSr>rTrrrrrRrs$  rRc@sJeZdZUedddZded<edddZded<edd dZd ed <dS) BFLFluxProGenerateResponse..The unique identifier for the generation task.rridz&URL to poll for the generation result. polling_urlNzPrice in centsz float | Nonecost)rrrrrWr3rXrYrrrrrUs rUc@s$eZdZdZdZdZdZdZdZdS) BFLStatuszTask not foundPendingzRequest ModeratedzContent ModeratedReadyErrorN) rrrtask_not_foundpendingrequest_moderatedcontent_moderatedreadyerrorrrrrrZsrZc@sbeZdZUedddZded<edddZded<ed d dZd ed <ed d dddZded<d S)BFLFluxStatusResponse.rVrrrWzThe status of the task.rZstatusNz/The result of the task (null if not completed).zOptional[Dict[str, Any]]resultz&The progress of the task (0.0 to 1.0).gg?rKzOptional[float]progress) rrrrrWr3rerfrgrrrrrds rdN) __future__renumrtypingrrrpydanticrrr r rr rr4r6r?rNrRrUrZrdrrrrs