o i@sddlmZddlmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZmZmZmZmZmZGdddejZGdddejZGd d d ejZGd d d ejZGd ddejZ GdddejZ!GdddejZ"GdddeZ#de#fddZ$dS))override)IOComfyExtensionInput) InputShouldRemeshInputShouldTextureMeshyAnimationRequestMeshyAnimationResultMeshyImageToModelRequestMeshyModelResultMeshyMultiImageToModelRequestMeshyRefineTaskMeshyRiggedResultMeshyRiggingRequestMeshyTaskResponseMeshyTextToModelRequestMeshyTextureRequest) ApiEndpointdownload_url_to_file_3dpoll_opsync_opupload_images_to_comfyapivalidate_stringc@sHeZdZeddZededededededed ed ej fd d Z d S)MeshyTextToModelNodecCs4tjdddtjjddgdtjjddd d tjjd d d gdtjjdtjdtjjdddgdtjjddddtjj dgtjdggddtjjdgdddtjjdgdddd tjjd!d"d"d#tjj dd$d%gtjj d&d't d(j d)d'tj j d*d'tj j d+d'gtjjtjjtjjgddtjd,d-d. S)/NrzMeshy: Text to Modelapi node/3d/MeshymodellatestoptionspromptT) multilinedefaultstyle realistic sculpture should_remeshtruetopologytrianglequadtarget_polycountdr"minmax display_modefalse:When set to false, returns an unprocessed triangular mesh.rtooltip symmetry_modeautoonoffradvanced pose_moder zA-posezT-pose.Specify the pose mode for the generated model.rr5r<seedr_Seed controls whether the node should re-run; results are non-deterministic regardless of seed.r"r/r0r1control_after_generater5 model_file display_name MESHY_TASK_ID meshy_task_idGLBFBXz{"type":"usd","usd":0.8}expr node_idrHcategoryinputsoutputshidden is_api_nodeis_output_node price_badge)rSchemaComborString DynamicComboOptionInt NumberDisplaynumberOutputCustom File3DGLB File3DFBXHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgeclsrk9/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_meshy.py define_schemasr  +  z"MeshyTextToModelNode.define_schemarrr#r&r6r=rAreturnc st|ddddt|tdddtt||||dd|d d|d d k|||d d IdH}|j} t|td| dt dddddIdH} t | d| t | j jd| dIdHt | j jd| dIdHS)NrX) field_name min_length max_length"/proxy/meshy/openapi/v2/text-to-3dPOSTpathmethodr(r+r&r') r art_styleai_modelr(r+r&r6r=rAresponse_modeldata#/proxy/meshy/openapi/v2/text-to-3d/rwcS|jSNstatusrrkrkrlz.MeshyTextToModelNode.execute..cSrrprogressrrkrkrlrrr|status_extractorprogress_extractor.glbglbtask_idfbx)rrrrrgetlowerresultrr r NodeOutputr model_urlsrr) rjrrr#r&r6r=rAresponserrrkrkrlexecuteas@       zMeshyTextToModelNode.executeN) __name__ __module__ __qualname__ classmethodrmstrrintrrrrkrkrkrlrs* A rc@sJeZdZeddZe d dededededejdBd e j f d d Z dS) MeshyRefineNodecCstjddddtjjddgdtdd tjjd d d d dtjjddd ddtjjddd dgtjjddtdjd dtj jddtj jddgtj j tj j tj jgd d tjddd S)NrzMeshy: Refine Draft Modelrz(Refine a previously created draft model.rrrrIrJ enable_pbrFzGenerate PBR Maps (metallic, roughness, normal) in addition to the base color. Note: this should be set to false when using Sculpture style, as Sculpture style generates its own set of PBR maps.Tr"r5r<texture_promptr Provide a text prompt to guide the texturing process. Maximum 600 characters. Cannot be used at the same time as 'texture_image'.r"r!r5 texture_imageMOnly one of 'texture_image' or 'texture_prompt' may be used at the same time.r5optionalrFrGrKrL{"type":"usd","usd":0.4}rM rPrHrQ descriptionrRrSrTrUrVrWrrXrYrraBooleanrZImager`rbrcrdrerfrgrhrirkrkrlrmsP   zMeshyRefineNode.define_schemaNrrJrrrrnc s|r |dur tdd}|rt|ddd|dur&t||ddIdHd}t|tdd d tt|||r5|nd||d d IdH}|j}t|td |dt dddddIdH} t |d|t | j jd|dIdHt | j jd|dIdHS)N@texture_prompt and texture_image cannot be used at the same timerrprqrsUploading texture wait_labelrrtrurv)preview_task_idrrtexture_image_urlrzendpointr|r}r~rcSrrrrrkrkrlrrz)MeshyRefineNode.execute..cSrrrrrkrkrlrrrrrrr) ValueErrorrrrrrr rrr rrrrrr) rjrrJrrrrrrrrkrkrlrsD      zMeshyRefineNode.executer rrrrrmrboolrrrrrrkrkrkrlrs$ 0rc@sJeZdZeddZededejdedede ded e d e j fd d Z d S)MeshyImageToModelNodecCstjdddtjjddgdtjdtjjdtjd tjjd d d gdtjjd dddtjj dgtjdggddtjjdgddtjjdtjd tj jddddtj jdddddtjjd d!dd"gtjdggd#dtjjd$gd%d&dd'tjjd(d)d)d*tjj dd+d,gtj j d-d.t d/j d0d.tjj d1d.tjj d2d.gtjjtjjtjjgddtjtjdgd3d4d5d6 S)7NrzMeshy: Image to Modelrrrrimager&r'r(r)r*r+r,r-r.r2r3r4r6r7should_texturerFNGenerate PBR Maps (metallic, roughness, normal) in addition to the base color.r"r5rr Trrrrr{Determines whether textures are generated. Setting it to false skips the texture phase and returns a mesh without textures.r=r>r?r@rArrBrCrDrFrGrIrJrKrLwidgetsz ( $prices := {"true": 1.2, "false": 0.8}; {"type":"usd","usd": $lookup($prices, widgets.should_texture)} )  depends_onrNrO)rrXrYrrr[r\r]r^r_rrZr`rarbrcrdrerfrgrhPriceBadgeDependsrirkrkrlrms     J   z#MeshyImageToModelNode.define_schemarrr&r6rr=rArncsb|ddk}d} } |r@|dr|ddurtd|dr,t|dddd|d} |ddur@t||ddd IdHd } t|td d d ttt||dd IdHd ||dd|dd||ddk||dd|| | |d dIdH} | j } t |td| dt dddddIdH} t | d| t| jjd| dIdHt| jjd| dIdHS)Nrr'rrrrprrrrz#/proxy/meshy/openapi/v1/image-to-3drurvzUploading base imager(r+r&r) image_urlrzr(r+r6r&rrr=rrrAr{z$/proxy/meshy/openapi/v1/image-to-3d/rcSrrrrrkrkrlrrz/MeshyImageToModelNode.execute..cSrrrrrkrkrlrrrrrrr)rrrrrrr rrrrr rrrrrr)rjrrr&r6rr=rAtexturerrrrrrkrkrlrUs`           zMeshyImageToModelNode.executeN)rrrrrmrrrrrrrrrrkrkrkrlrs* f rc@sLeZdZeddZededejjde dede ded e d ej fd d Z d S)MeshyMultiImageToModelNodecCstjdddtjjddgdtjjdtjjtjddd d d d tjjd tjdtjjdddgdtj jddddtj j dgtjdggddtjjdgdddtjjdtjdtj jddd d!tj jd"d#dd$d%tjjd&d'dd(gtjdggd)dtjjd*gd+d,dd-tj jd.d/d/d0tj j dd1d2gtj jd3d4td5jd6d4tjjd7d4tjjd8d4gtjjtjjtjjgddtjtjdgd9d:d;d< S)=NrzMeshy: Multi-Image to Modelrrrrimagesr)prefixr/r0)templater&r'r(r)r*r+r,r-r.r2r3r4r6r7Tr;rrFrrrr rrrrrrr=r>r?r@rArrBrCrDrFrGrIrJrKrLrz ( $prices := {"true": 0.6, "false": 0.2}; {"type":"usd","usd": $lookup($prices, widgets.should_texture)} ) rrO)rrXrYrAutogrowTemplatePrefixrr[r\r]r^r_rrZr`rarbrcrdrerfrgrhrrirkrkrlrms    M   z(MeshyMultiImageToModelNode.define_schemarrr&r6rr=rArncsf|ddk}d} } |r@|dr|ddurtd|dr,t|dddd|d} |ddur@t||ddd IdHd } t|td d d ttt|t|dd IdH|| dd| dd||ddk|| dd| | | |d dIdH} | j } t |td| dt dddddIdH} t| d| t| jjd| dIdHt| jjd| dIdHS)Nrr'rrrrprrrrz)/proxy/meshy/openapi/v1/multi-image-to-3drurvzUploading base imagesr(r+r&r) image_urlsrzr(r+r6r&rrr=rrrAr{z*/proxy/meshy/openapi/v1/multi-image-to-3d/rcSrrrrrkrkrlr0rz4MeshyMultiImageToModelNode.execute..cSrrrrrkrkrlr1rrrrrr)rrrrrrr listvaluesrrrrr rrrrrr)rjrrr&r6rr=rArrrrrrrkrkrlrsd            z"MeshyMultiImageToModelNode.executeN)rrrrrmrrrTyperrrrrrkrkrkrlrs* i rc @sBeZdZeddZe d dededejdBde j fdd Z dS) MeshyRigModelNodec Cstjddddtddtjjddd d d d tjjd dddgtjjddtdjddtjjddtj jddgtj j tj j tj j gddtjddd S)NrzMeshy: Rig ModelrzProvides a rigged character in standard formats. Auto-rigging is currently not suitable for untextured meshes, non-humanoid assets, or humanoid assets with unclear limb and body structure.rIrJ height_metersg?g.@g333333?zcThe approximate height of the character model in meters. This aids in scaling and rigging accuracy.)r/r0r"r5rz2The model's UV-unwrapped base color texture image.TrrFrGMESHY_RIGGED_TASK_ID rig_task_idrKrLz{"type":"usd","usd":0.2}rMr)rrXrarFloatrrZr`rbrcrdrerfrgrhrirkrkrlrm=sD   zMeshyRigModelNode.define_schemaNrJrrrnc sd}|durt||ddIdHd}t|tdddtt|||ddIdH}|j}t|td |d td d d d dIdH}t |d|t |jj d|dIdHt |jj d|dIdHS)Nrrrz/proxy/meshy/openapi/v1/riggingrurv) input_task_idrrrz /proxy/meshy/openapi/v1/rigging/rcSrrrrrkrkrlrrz+MeshyRigModelNode.execute..cSrrrrrkrkrlrrrrrrr) rrrrrrrrrrrrigged_character_glb_urlrigged_character_fbx_url)rjrJrrrrrrrkrkrlrhs8    zMeshyRigModelNode.executer) rrrrrmrfloatrrrrrrkrkrkrlr;s *rc@s4eZdZeddZedededejfddZ dS) MeshyAnimateModelNodec Cs~tjddddtddtjjdddd d d gtjjd d tjjdd tjjdd gtj j tj j tj j gddtj ddd S)NrzMeshy: Animate ModelrzCApply a specific animation action to a previously rigged character.rr action_idrizTVisit https://docs.meshy.ai/en/api/animation-library for a list of available values.)r"r/r0r5rFrGrKrLTz{"type":"usd","usd":0.12}rMr)rrXrarr]rZr`rbrcrdrerfrgrhrirkrkrlrms8  z#MeshyAnimateModelNode.define_schemarrrnc st|tdddtt||ddIdH}|j}t|td|dtdd d d d IdH}t|d t |jj d |dIdHt |jj d|dIdHS)Nz"/proxy/meshy/openapi/v1/animationsrurv)rrrz#/proxy/meshy/openapi/v1/animations/rcSrrrrrkrkrlrrz/MeshyAnimateModelNode.execute..cSrrrrrkrkrlrrrrrrr) rrrrrrr rrranimation_glb_urlanimation_fbx_url)rjrrrrrrkrkrlrs.    zMeshyAnimateModelNode.executeN) rrrrrmrrrrrrkrkrkrlrs !rc@sNeZdZeddZe d dededededed ejdBd e j fd d Z dS)MeshyTextureNodecCstjdddtjjddgdtddtjjd d d d d tjjd dd dtjjddd ddtjjdd ddgtjjddtdjddtj jddtj jddgtj j tj j tj jgd d tjddd S)NrzMeshy: Texture ModelrrrrrIrJenable_original_uvTzUse the original UV of the model instead of generating new UVs. When enabled, Meshy preserves existing textures from the uploaded model. If the model has no original UV, the quality of the output might not be as good.rpbrF)r"r<text_style_promptr zDescribe your desired texture style of the object using text. Maximum 600 characters.Maximum 600 characters. Cannot be used at the same time as 'image_style'.r image_stylezeA 2d image to guide the texturing process. Can not be used at the same time with 'text_style_prompt'.)rr5rFrG MODEL_TASK_IDrKrLrrMrOrrirkrkrlrmsP   zMeshyTextureNode.define_schemaNrrJrrrrrnc s|r |dur td|s|durtdd}|dur't||ddIdHd}t|tdddtt|||||r8|nd|d d IdH}|j} t|td | d td ddddIdH} t | d| t | j j d| dIdHt | j jd| dIdHS)NzAtext_style_prompt and image_style cannot be used at the same timez3Either text_style_prompt or image_style is requiredzUploading stylerrz!/proxy/meshy/openapi/v1/retexturerurv)rrzrrrimage_style_urlrz"/proxy/meshy/openapi/v1/retexture/rcSrrrrrkrkrlr%rz*MeshyTextureNode.execute..cSrrrrrkrkrlr&rrrrrr)rrrrrrrrr rrrrrr) rjrrJrrrrrrrrrkrkrlrsF       zMeshyTextureNode.executerrrkrkrkrlrs( 1rc@s(eZdZedeeejfddZdS)MeshyExtensionrncstttttttgSr)rrrrrrr)selfrkrkrl get_node_list1szMeshyExtension.get_node_listN) rrrrrtyper ComfyNoderrkrkrkrlr0srrncstSr)rrkrkrkrlcomfy_entrypoint>srN)%typing_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.meshyrrrr r r r r rrrrrcomfy_api_nodes.utilrrrrrrrrrrrrrrrrrkrkrkrls <  p_&+QBb