
    +j                     &   d dl mZ d dlmZmZ d dlmZ  G d de          Z G d de          Z G d d	e          Z	 G d
 de          Z
 G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d  d!e          Z G d" d#e          Z G d$ d%e          Z G d& d'e          Zd(S ))    )	TypedDict)	BaseModelField)Inputc                   .    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     :/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/meshy.pyr   r      s3         MMMr   r   c                   H    e Zd ZU eed<   eed<   eed<   ej        dz  ed<   dS )InputShouldTextureshould_texture
enable_pbrtexture_promptNtexture_image)r   r   r   r   r   boolr   Imager   r   r   r   r      sI         ;%%%%%%r   r   c                   0    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*         %**FCr   r   c                      e Zd ZU  ed          Zeed<    edd          Zeed<    edd          Zeed	<    ed          Z	eed
<    edd          Z
edz  ed<    eddd          Zedz  ed<    edd          Zeed<    e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   r4   r5   r6   r7   r   r   r   r!   r!      sd        i  D#   %,,,FC,,,U3,HIIIIsIIIE#JJHc 52HIIIHcDjIII#(5#@#@#@cDj@@@%W  M4    s0GHHHM3HHHU3ZZIsc

D#uU||J#####r   r!   c                      e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Ze	dz  ed<    ed          Z
edz  ed<    ed          Zedz  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+   Fr7   )r   r   r   r   r#   r   r   r;   r   r   r   r<   r+   r7   r   r   r   r9   r9   *   s         hD# 5::OS%%%#eCjjJt(((!&sNC$J+++$)E#JJsTz...E#JJHcuU||J#####r   r9   c                      e Zd ZU  ed          Zeed<    ed          Zeed<    edd          Zedz  ed<    eddd	
          Z	e
dz  ed<    edd          Zeed<    edd          Zeed<    ed          Zeed<    ed          Zedz  ed<    ed          Zeed<    edd          Zedz  ed<    ed          Zedz  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   r3   r4   Tr2   r	   r   r   r5   r$   r%   r   r<   r6   Fr7   )r   r   r   r   r?   r   r   r+   r
   r   r   r4   r	   r   r   r   r5   r   r<   r6   r7   r   r   r   r>   r>   4   s        U3ZZIsE#JJHc 52HIIIHcDjIII#(5#@#@#@cDj@@@s0GHHHM3HHH%W  M4    !5::ND%%%#eCjjJt(((U3ZZIs!&t!<!<!<NC$J<<<$)E$KKsTz///c

D#uU||J#####r   r>   c                      e Zd ZU  ed          Zee         ed<    ed          Zeed<    edd          Z	edz  ed<    eddd	
          Z
edz  ed<    edd          Zeed<    edd          Zeed<    ed          Zeed<    ed          Zedz  ed<    ed          Zeed<    edd          Zedz  ed<    ed          Zedz  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   r3   r4   Tr2   r	   r   r   r5   r$   r%   r   r<   r6   Fr7   )r   r   r   r   rB   listr   r   r+   r
   r   r   r4   r	   r   r   r   r5   r   r<   r6   r7   r   r   r   rA   rA   G   s        !E#JJJS	&&&E#JJHc 52HIIIHcDjIII#(5#@#@#@cDj@@@s0GHHHM3HHH%W  M4    !5::ND%%%#eCjjJt(((U3ZZIs!&t!<!<!<NC$J<<<$)E$KKsTz///c

D#uU||J#####r   rA   c                   v    e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	edz  ed<   dS )MeshyRiggingRequest.input_task_idheight_metersNr<   )
r   r   r   r   rF   r   r   rG   floatr<   r   r   r   rE   rE   Z   s_         sM3### 5::M5%%%$)E#JJsTz.....r   rE   c                   P    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   rK   r   r   rL   r   r   r   r   rJ   rJ   `   sB         uSzzK!!!U3ZZIsr   rJ   c                       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z  ed<    ed          Zedz  ed<   dS )	MeshyTextureRequest.rF   r+   enable_original_uvr   Ntext_style_promptimage_style_url)r   r   r   r   rF   r   r   r+   rO   r   r   rP   rQ   r   r   r   rN   rN   e   s         sM3###E#JJHc$uSzz)))uSzzJ!!!$)E#JJsTz..."'%**OS4Z,,,,,r   rN   c                       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   rU   r   r   rV   rW   rX   r   r   r   rS   rS   n   sr         uRyyCuRyyCb		D#uRyyCr   rS   c                   P    e Zd ZU  ed          Zeed<    ed          Zeed<   dS )MeshyRiggedModelsUrlsrT   rigged_character_glb_urlrigged_character_fbx_urlN)r   r   r   r   r[   r   r   r\   r   r   r   rZ   rZ   u   sD         $)E"IIc---$)E"IIc-----r   rZ   c                   P    e Zd ZU  ed          Zeed<    ed          Zeed<   dS )MeshyAnimatedModelsUrlsrT   animation_glb_urlanimation_fbx_urlN)r   r   r   r   r_   r   r   r`   r   r   r   r^   r^   z   sD         "U2YYs&&&"U2YYs&&&&&r   r^   c                       e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Zedz  ed<    ed          Z	edz  ed<   dS )MeshyResultTextureUrls.
base_colorNmetallicnormal	roughness)
r   r   r   r   rc   r   r   rd   re   rf   r   r   r   rb   rb      s~         eCjjJ    5;;HcDj&&&tFC$J$$$!E$KKIsTz'''''r   rb   c                   6    e Zd ZU  ed          Zedz  ed<   dS )MeshyTaskErrorNmessage)r   r   r   r   ri   r   r   r   r   r   rh   rh      s.         %++GS4Z%%%%%r   rh   c                   ^   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z  ed<    ed          Zeed<    ed	          Zeed
<    eg           Zee         dz  ed<    ed          Zedz  ed<   dS )MeshyModelResult.idtype
model_urlsthumbnail_urlN	video_urlstatusr   progresstexture_urls
task_error)r   r   r   r   rl   r   r   rm   rS   rn   ro   rp   rq   rr   r   rs   rC   rb   rt   rh   r   r   r   rk   rk      s        eCjjBc

D#"'%(9(9":":J:::sM3###!E$KKIsTz'''%**FCE!HHHc8=b		L$-.5AAA(-dJ%33333r   rk   c                       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z  ed	<   dS )
MeshyRiggedResult.rl   rm   rq   r   rr   r   Nrt   )r   r   r   r   rl   r   r   rm   rq   rr   r   rZ   r   rt   rh   r   r   r   rv   rv      s         eCjjBc

D#%**FCE!HHHc$)E*?*?*A*A$B$BF!BBB(-dJ%33333r   rv   c                       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z  ed	<   dS )
MeshyAnimationResult.rl   rm   rq   r   rr   r   Nrt   )r   r   r   r   rl   r   r   rm   rq   rr   r   r^   r   rt   rh   r   r   r   rx   rx      s         eCjjBc

D#%**FCE!HHHc&+e,C,C,E,E&F&FF#FFF(-dJ%33333r   rx   N)typingr   pydanticr   r   comfy_api.latestr   r   r   r   r!   r9   r>   rA   rE   rJ   rN   rS   rZ   r^   rb   rh   rk   rv   rx   r   r   r   <module>r|      s?         % % % % % % % % " " " " " "    	   & & & & & & & &    	   $ $ $ $ $i $ $ $"$ $ $ $ $i $ $ $$ $ $ $ $y $ $ $&$ $ $ $ $I $ $ $&/ / / / /) / / /         I      
- - - - -) - - -    i   . . . . .I . . .
' ' ' ' 'i ' ' '
( ( ( ( (Y ( ( (& & & & &Y & & &	4 	4 	4 	4 	4y 	4 	4 	44 4 4 4 4	 4 4 44 4 4 4 49 4 4 4 4 4r   