
    +jt                     >   S SK Jr  S SKJrJr   " S S\5      r " S S\5      r " S S\5      r " S	 S
\5      r " S S\5      r	 " S S\5      r
 " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      rg)    )	TypedDict)	BaseModelFieldc                   4    \ rS rSr% \\S'   \\S'   \\S'   Srg)InputModerationSettings   prompt_content_moderationvisual_input_moderationvisual_output_moderation N)__name__
__module____qualname____firstlineno__bool__annotations____static_attributes__r       9/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/bria.pyr   r      s    ##!!""r   r   c                   x   \ rS rSr% \" S5      r\S-  \S'   \" SSS9r\S-  \S'   \" SSS9r	\
\   \S	'   \" SS
S9r\S-  \S'   \" S5      r\S-  \S'   \" S5      r\\S'   \" S5      r\\S'   \" S5      r\\S'   \" S5      r\\S'   \" SSS9r\\S'   \" SSS9r\\S'   \" SSS9r\\S'   \" SSS9r\\S'   Srg)BriaEditImageRequest   .NinstructionzBUse this instead of instruction for precise, programmatic control.descriptionstructured_instructionzRRequired. Publicly available URL or Base64-encoded. Must contain exactly one item.imageszMask image (black and white). Black areas will be preserved, white areas will be edited. If omitted, the edit applies to the entire image. The input image and the input mask must be of the same size.masknegative_promptguidance_scalemodel_version	steps_numseedFzGIf true, returns a warning for potential IP content in the instruction.	ip_signalz7If true, returns 422 on instruction moderation failure.r	   z:If true, returns 422 on images or mask moderation failure.visual_input_content_moderation9If true, returns 422 on visual output moderation failure. visual_output_content_moderationr   )r   r   r   r   r   r   strr   r   r   listr   r   r    floatr!   r"   intr#   r$   r   r	   r%   r'   r   r   r   r   r   r      s   #CjKt().X*C$J  hFDI  GD#*  #(+OS4Z-!#JNE&sM3#3ZIsc
D#]It  ',T't  -2W-#T  .3V.$d r   r   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\	\S'   \" SSS9r
\	\S'   \" SS	S9r\	\S
'   \" S5      r\\S'   Srg)BriaRemoveBackgroundRequest0   .imageFsyncz7If true, returns 422 on input image moderation failure.r   r%   r&   r'   r#   r   N)r   r   r   r   r   r/   r(   r   r0   r   r%   r'   r#   r+   r   r   r   r   r-   r-   0   s`    sE3uD$,1T-#T  .3V.$d  c
D#r   r-   c                   j    \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   \" S5      r	\S-  \S'   Sr
g)BriaStatusResponse<   .
request_id
status_urlNwarningr   )r   r   r   r   r   r4   r(   r   r5   r6   r   r   r   r   r2   r2   <   s0    CjJ CjJ +GS4Z%r   r2   c                   0    \ rS rSr% \" S5      r\\S'   Srg)BriaRemoveBackgroundResultB   .	image_urlr   N)	r   r   r   r   r   r:   r(   r   r   r   r   r   r8   r8   B       3ZIsr   r8   c                   P    \ rS rSr% \" S5      r\\S'   \" S5      r\	S-  \S'   Sr
g)BriaRemoveBackgroundResponseF   .statusNresultr   )r   r   r   r   r   r?   r(   r   r@   r8   r   r   r   r   r=   r=   F   s$    *FC05dF&-;r   r=   c                   J    \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   Sr	g)BriaImageEditResultK   .structured_promptr:   r   N)
r   r   r   r   r   rD   r(   r   r:   r   r   r   r   rB   rB   K   s    "3Zs'3ZIsr   rB   c                   P    \ rS rSr% \" S5      r\\S'   \" S5      r\	S-  \S'   Sr
g)BriaImageEditResponseP   .r?   Nr@   r   )r   r   r   r   r   r?   r(   r   r@   rB   r   r   r   r   rF   rF   P   s$    *FC).tF$&4r   rF   c                       \ rS rSr% \" S5      r\\S'   \" SSS9r\\S'   \" S5      r	\\S'   \" S	5      r
\\S
'   \" S5      r\\S'   Srg) BriaRemoveVideoBackgroundRequestU   .videotransparentz&Background color for the output video.defaultr   background_coloroutput_container_and_codecTpreserve_audior#   r   N)r   r   r   r   r   rK   r(   r   rO   rP   rQ   r   r#   r+   r   r   r   r   rI   rI   U   sN    sE3!-Emncn&+Cj0 ;ND&c
D#r   rI   c                   0    \ rS rSr% \" S5      r\\S'   Srg)BriaRemoveVideoBackgroundResult]   .	video_urlr   N)	r   r   r   r   r   rU   r(   r   r   r   r   r   rS   rS   ]   r;   r   rS   c                   P    \ rS rSr% \" S5      r\\S'   \" S5      r\	S-  \S'   Sr
g)!BriaRemoveVideoBackgroundResponsea   .r?   Nr@   r   )r   r   r   r   r   r?   r(   r   r@   rS   r   r   r   r   rW   rW   a   s$    *FC5:4[F+d2@r   rW   c                       \ rS rSr% \" SSS9r\\S'   \" SSS9r\\S	'   \" S5      r	\\S
'   \" S5      r
\\S'   \" S5      r\\S'   Srg)BriaVideoGreenScreenRequestf   .z+Publicly accessible URL of the input video.r   rK   broadcast_greenzeSolid chroma-key shade applied behind the foreground (broadcast_green, chroma_green, or blue_screen).rM   green_shaderP   TrQ   r#   r   N)r   r   r   r   r   rK   r(   r   r]   rP   rQ   r   r#   r+   r   r   r   r   rZ   rZ   f   sZ    s(UVE3V!;K 
 ',Cj0 ;ND&c
D#r   rZ   c                       \ rS rSr% \" SSS9r\\S'   \" SSS9r\\S'   \" S5      r	\\S'   \" S	5      r
\\S
'   \" S5      r\\S'   Srg)!BriaVideoReplaceBackgroundRequestr   .z8Publicly accessible URL of the input (foreground) video.r   rK   zPublicly accessible URL of the background image or video to composite behind the foreground. Stretched to the foreground frame; match its aspect ratio for undistorted results.background_urlrP   TrQ   r#   r   N)r   r   r   r   r   rK   r(   r   ra   rP   rQ   r   r#   r+   r   r   r   r   r_   r_   r   sZ    s(bcE3cNC  ',Cj0 ;ND&c
D#r   r_   N)typingr   pydanticr   r   r   r   r-   r2   r8   r=   rB   rF   rI   rS   rW   rZ   r_   r   r   r   <module>rd      s     %#i #!9 !H	) 	& &   <9 <
 )  
5I 5
y  i  A	 A
	) 	
	 
r   