o
    iQ                     @   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dS )    )	TypedDict)	BaseModelFieldc                   @   s   e Zd ZU eed< eed< dS )InputVideoModelmodel
resolutionN)__name__
__module____qualname__str__annotations__ r   r   9/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/hitpaw.pyr      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ed< edZ	e
ed< edZedB ed	< dS )
ImageEnhanceTaskCreateRequest.
model_nameimg_urlz.png	extensionFexifNDPI)r   r	   r
   r   r   r   r   r   r   r   boolr   intr   r   r   r   r      s   
 r   c                   @   sr   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< edddZ	e
e ed	< edd
dZe
e ed< dS )VideoEnhanceTaskCreateRequest.	video_urlz.mp4r   Nr   z!Target resolution [width, height])descriptionr   z)Original video resolution [width, height]original_resolution)r   r	   r
   r   r   r   r   r   r   r   listr   r   r   r   r   r   r      s   
 r   c                   @   s2   e Zd ZU edZeed< edZedB ed< dS )TaskCreateDataResponse.job_idNconsume_coins)	r   r	   r
   r   r   r   r   r   r   r   r   r   r   r      s   
 r   c                   @   s   e Zd ZU edZeed< dS )TaskStatusPollRequest.r   N)r   r	   r
   r   r   r   r   r   r   r   r   r       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 )TaskCreateResponse.codemessageNdata)r   r	   r
   r   r!   r   r   r"   r   r#   r   r   r   r   r   r    $   s   
 r    c                   @   s>   e Zd ZU edZeed< edZeed< edZeed< dS )TaskStatusDataResponse.r   status res_urlN)	r   r	   r
   r   r   r   r   r%   r'   r   r   r   r   r$   *      
 r$   c                   @   s>   e Zd ZU edZeed< edZeed< edZ	e
ed< dS )TaskStatusResponse.r!   r"   r#   N)r   r	   r
   r   r!   r   r   r"   r   r#   r$   r   r   r   r   r)   0   r(   r)   N)typingr   pydanticr   r   r   r   r   r   r   r    r$   r)   r   r   r   r   <module>   s    