o
    i                     @   s   d dl mZ d dlmZmZ G dd deZG dd deZG dd deZG d	d
 d
eZG dd deZ	G dd deZ
G dd deZG dd deZG dd deZG dd deZG dd deZdS )    )	TypedDict)	BaseModelFieldc                   @   s&   e Zd ZU eed< eed< eed< dS )InputModerationSettingsprompt_content_moderationvisual_input_moderationvisual_output_moderationN)__name__
__module____qualname__bool__annotations__ r   r   7/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/bria.pyr      s   
 r   c                   @   s  e Zd ZU edZedB ed< edddZedB ed< edddZe	e ed< edd	dZ
edB ed
< edZedB ed< edZeed< edZeed< edZeed< edZeed< edddZeed< edddZeed< edddZeed< edddZeed< dS )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 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_moderation)r	   r
   r   r   r   strr   r   r   listr   r   r   floatr   r   intr   r   r   r   r   r   r   r   r   r   r      s@   
 r   c                   @   sf   e Zd ZU edZeed< edZeed< edddZ	eed< edddZ
eed	< edZeed
< dS )BriaRemoveBackgroundRequest.imageFsyncz7If true, returns 422 on input image moderation failure.r   r   r   r   r   N)r	   r
   r   r   r%   r    r   r&   r   r   r   r   r#   r   r   r   r   r$   0   s   
 r$   c                   @   sB   e Zd ZU edZeed< edZeed< edZedB ed< dS )BriaStatusResponse.
request_id
status_urlNwarning)	r	   r
   r   r   r(   r    r   r)   r*   r   r   r   r   r'   <   s   
 r'   c                   @      e Zd ZU edZeed< dS )BriaRemoveBackgroundResult.	image_urlN)r	   r
   r   r   r-   r    r   r   r   r   r   r,   B      
 r,   c                   @   2   e Zd ZU edZeed< edZedB ed< dS )BriaRemoveBackgroundResponse.statusNresult)	r	   r
   r   r   r1   r    r   r2   r,   r   r   r   r   r0   F      
 r0   c                   @   s.   e Zd ZU edZeed< edZeed< dS )BriaImageEditResult.structured_promptr-   N)r	   r
   r   r   r5   r    r   r-   r   r   r   r   r4   K   s   
 r4   c                   @   r/   )BriaImageEditResponse.r1   Nr2   )	r	   r
   r   r   r1   r    r   r2   r4   r   r   r   r   r6   P   r3   r6   c                   @   sb   e Zd ZU edZeed< edddZeed< edZeed< edZ	e
ed	< edZeed
< dS ) BriaRemoveVideoBackgroundRequest.videotransparentz&Background color for the output video.)defaultr   background_coloroutput_container_and_codecTpreserve_audior   N)r	   r
   r   r   r8   r    r   r;   r<   r=   r   r   r#   r   r   r   r   r7   U   s   
 r7   c                   @   r+   )BriaRemoveVideoBackgroundResult.	video_urlN)r	   r
   r   r   r?   r    r   r   r   r   r   r>   ]   r.   r>   c                   @   r/   )!BriaRemoveVideoBackgroundResponse.r1   Nr2   )	r	   r
   r   r   r1   r    r   r2   r>   r   r   r   r   r@   a   r3   r@   N)typingr   pydanticr   r   r   r   r$   r'   r,   r0   r4   r6   r7   r>   r@   r   r   r   r   <module>   s    $