o
    i]                     @   sL  d dl mZ d dlmZmZ d dl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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Field)Inputc                   @   s&   e Zd ZU eed< eed< eed< dS )InputShouldRemeshshould_remeshtopologytarget_polycountN)__name__
__module____qualname__str__annotations__int r   r   8/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/meshy.pyr      s   
 r   c                   @   s4   e Zd ZU eed< eed< eed< ejdB ed< dS )InputShouldTextureshould_texture
enable_pbrtexture_promptNtexture_image)r
   r   r   r   r   boolr   Imager   r   r   r   r      s
   
 r   c                   @   s   e Zd ZU edZeed< dS )MeshyTaskResponse.resultN)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ddZeed< edddZeed	< edZ	eed
< edddZ
edB ed< eddddZedB ed< edd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 )MeshyTextToModelRequestpreviewmode.X  
max_lengthpromptz'realistic' or 'sculpture'description	art_styleai_model'quad' or 'triangle'Nr   d    geler	   TAFalse returns the original mesh, ignoring topology and polycount.r   'auto', 'off' or 'on'symmetry_mode	pose_modeseedF
moderation)r
   r   r   r   r   r   r   r!   r$   r%   r   r	   r   r   r   r.   r/   r0   r1   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	dB ed< e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< dS )MeshyRefineTaskrefiner   .preview_task_idNr   r   texture_image_urlr%   Fr1   )r
   r   r   r   r   r   r   r4   r   r   r   r5   r%   r1   r   r   r   r   r2   *   s   
 r2   c                   @   s  e Zd ZU edZeed< edZeed< edddZedB ed< eddd	d
Z	e
dB ed< edddZeed< edddZeed< edZeed< edZedB ed< edZ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< dS )MeshyImageToModelRequest.	image_urlr%   r&   r"   Nr   r'   r(   r)   r	   r-   r.   Tr,   r   r   r   r/   r   r   r   r5   r0   Fr1   )r
   r   r   r   r7   r   r   r%   r   r	   r   r.   r   r   r   r   r/   r   r5   r0   r1   r   r   r   r   r6   4   s"   
 r6   c                   @   s  e Zd ZU edZee ed< edZeed< edddZ	edB ed< eddd	d
Z
edB ed< edddZeed< edddZeed< edZeed< edZedB ed< edZ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< dS )MeshyMultiImageToModelRequest.
image_urlsr%   r&   r"   Nr   r'   r(   r)   r	   r-   r.   Tr,   r   r   r   r/   r   r   r   r5   r0   Fr1   )r
   r   r   r   r9   listr   r   r%   r   r	   r   r.   r   r   r   r   r/   r   r5   r0   r1   r   r   r   r   r8   G   s"   
 r8   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 )MeshyRiggingRequest.input_task_idheight_metersNr5   )
r
   r   r   r   r<   r   r   r=   floatr5   r   r   r   r   r;   Z   s   
 r;   c                   @   s.   e Zd ZU edZeed< edZeed< dS )MeshyAnimationRequest.rig_task_id	action_idN)	r
   r   r   r   r@   r   r   rA   r   r   r   r   r   r?   `      
 r?   c                   @   sv   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< edZedB ed< dS )	MeshyTextureRequest.r<   r%   enable_original_uvr   Ntext_style_promptimage_style_url)r
   r   r   r   r<   r   r   r%   rD   r   r   rE   rF   r   r   r   r   rC   e   s   
 rC   c                   @   sN   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< dS )MeshyModelsUrls glbfbxusdzobjN)
r
   r   r   r   rI   r   r   rJ   rK   rL   r   r   r   r   rG   n   s
   
 rG   c                   @   .   e Zd ZU edZeed< edZeed< dS )MeshyRiggedModelsUrlsrH   rigged_character_glb_urlrigged_character_fbx_urlN)r
   r   r   r   rO   r   r   rP   r   r   r   r   rN   u   rB   rN   c                   @   rM   )MeshyAnimatedModelsUrlsrH   animation_glb_urlanimation_fbx_urlN)r
   r   r   r   rR   r   r   rS   r   r   r   r   rQ   z   rB   rQ   c                   @   sZ   e Zd ZU edZeed< edZedB ed< edZedB ed< edZ	edB ed< dS )MeshyResultTextureUrls.
base_colorNmetallicnormal	roughness)
r
   r   r   r   rU   r   r   rV   rW   rX   r   r   r   r   rT      s
   
 rT   c                   @   s"   e Zd ZU edZedB ed< dS )MeshyTaskErrorNmessage)r
   r   r   r   rZ   r   r   r   r   r   r   rY      s   
 rY   c                   @   s   e Zd ZU edZeed< edZeed< ee Z	eed< edZ
eed< edZedB ed< edZeed< ed	Zeed
< eg Zee dB ed< edZedB ed< dS )MeshyModelResult.idtype
model_urlsthumbnail_urlN	video_urlstatusr   progresstexture_urls
task_error)r
   r   r   r   r\   r   r   r]   rG   r^   r_   r`   ra   rb   r   rc   r:   rT   rd   rY   r   r   r   r   r[      s   
 r[   c                   @   t   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e Zeed< edZedB ed	< dS )
MeshyRiggedResult.r\   r]   ra   r   rb   r   Nrd   )r
   r   r   r   r\   r   r   r]   ra   rb   r   rN   r   rd   rY   r   r   r   r   rf         
 rf   c                   @   re   )
MeshyAnimationResult.r\   r]   ra   r   rb   r   Nrd   )r
   r   r   r   r\   r   r   r]   ra   rb   r   rQ   r   rd   rY   r   r   r   r   rh      rg   rh   N)typingr   pydanticr   r   comfy_api.latestr   r   r   r   r   r2   r6   r8   r;   r?   rC   rG   rN   rQ   rT   rY   r[   rf   rh   r   r   r   r   <module>   s*    
		