o i`@srUddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZdd lmZmZdd lmZmZmZmZmZmZdd lmZdd lm Z!dd l"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+Z,GdddeZ-GdddeZ.GdddZ/GdddZ0GdddZ1e-j2Z2e-Z3erddl4Z5de6d<e e-Z7eZ8e!Z9gdZ:dS)) annotations)ABCabstractmethod) TYPE_CHECKING) ComfyAPIBase)ProxiedSingleton)create_sync_class) ImageInput AudioInput MaskInput LatentInput VideoInput) VideoFromFileVideoFromComponents) VideoCodecVideoContainerVideoComponentsMESHVOXELFile3D) _io_public) _ui_public)get_executing_context)get_progress_statePreviewImageTuple)Image)argsNcsTeZdZdZdZfddZGdddeZGdddeZGd d d eZ Z S) ComfyAPI_latestlatestFcs,t||_||_||_dS)N)super__init__NodeReplacementnode_replacement Execution executionCachingcachingself __class__7/mnt/c/Users/fbmor/ComfyUI/comfy_api/latest/__init__.pyr!s   zComfyAPI_latest.__init__c@seZdZdddZdS) zComfyAPI_latest.NodeReplacement node_replaceio.NodeReplacereturnNonecs ddlm}|jj|dS)z$Register a node replacement mapping.r) PromptServerN)serverr2instancenode_replace_managerregister)r)r.r2r,r,r-r6s z(ComfyAPI_latest.NodeReplacement.registerN)r.r/r0r1)__name__ __module__ __qualname__r6r,r,r,r-r"sr"c@seZdZ   ddddZdS)zComfyAPI_latest.ExecutionNFvaluefloat max_valuenode_id str | None preview_imageImage.Image | ImageInput | Noneignore_size_limitboolr0r1c st}|dur|dur|j}|durtd|}|durZt|tr@|}t|jdkr/|d}|d t j } t | }t|t j rZ|jrL|jnd} |rRdntj} | || f}tj||||ddS)aM Update the progress bar displayed in the ComfyUI interface. This function allows custom nodes and API calls to report their progress back to the user interface, providing visual feedback during long operations. Migration from previous API: comfy.utils.PROGRESS_BAR_HOOK Nz4node_id must be provided if not in executing contextrJPEG)r=r:r<image)rr= ValueError isinstancer lenshapecpunumpyastypenpuint8r fromarrayformatr preview_sizerupdate_progress) r)r:r<r=r?rAexecuting_context to_displaytensorimage_np image_formatrRr,r,r- set_progress%s0     z&ComfyAPI_latest.Execution.set_progress)NNF) r:r;r<r;r=r>r?r@rArBr0r1)r7r8r9rYr,r,r,r-r$$s r$c@s8eZdZdZddlmZmZmZd dd Zd d d Z d S)zComfyAPI_latest.Cachinga External cache provider API for sharing cached node outputs across ComfyUI instances. Example:: from comfy_api.latest import Caching class MyCacheProvider(Caching.CacheProvider): async def on_lookup(self, context): ... # check external storage async def on_store(self, context, value): ... # store to external storage Caching.register_provider(MyCacheProvider()) r ) CacheProvider CacheContext CacheValueprovider''ComfyAPI_latest.Caching.CacheProvider'r0r1cddlm}||dS)zPRegister an external cache provider. Providers are called in registration order.r)register_cache_providerN)comfy_execution.cache_providerr`)r)r]r`r,r,r-register_providerl  z)ComfyAPI_latest.Caching.register_providercr_)z2Unregister a previously registered cache provider.r)unregister_cache_providerN)rard)r)r]rdr,r,r-unregister_providerqrcz+ComfyAPI_latest.Caching.unregister_providerN)r]r^r0r1) r7r8r9__doc___cachingrZr[r\rbrer,r,r,r-r&Xs  r&) r7r8r9VERSIONSTABLEr!rr"r$r& __classcell__r,r,r*r-rs 4rc@s$eZdZd ddZed ddZdS) ComfyExtensionr0r1cdS)z Called when an extension is loaded. This should be used to initialize any global resources needed by the extension. Nr,r(r,r,r-on_loadwzComfyExtension.on_loadlist[type[io.ComfyNode]]crl)zG Returns a list of nodes that this extension provides. Nr,r(r,r,r- get_node_list}rnzComfyExtension.get_node_listN)r0r1)r0ro)r7r8r9rmrrpr,r,r,r-rkvs rkc@s eZdZeZeZeZe Z e Z dS)InputN) r7r8r9r rr Audior Maskr LatentrVideor,r,r,r-rqs rqc@seZdZeZeZdS) InputImplN)r7r8r9rrr,r,r,r-rvsrvc@s$eZdZeZeZeZeZeZeZdS)TypesN) r7r8r9rrrrrrr,r,r,r-rwsrwzBtype[comfy_api.latest.generated.ComfyAPISyncStub.ComfyAPISyncStub] ComfyAPISync) ComfyAPIrxrqrvrwr&rkioIOuiUI); __future__rabcrrtypingrcomfy_api.internalrcomfy_api.internal.singletonr comfy_api.internal.async_to_syncr_inputr r r r r _input_implrr_utilrrrrrrrrzrr|comfy_execution.utilsrcomfy_execution.progressrrPILrcomfy.cli_argsrrLrNrrkrqrvrwr&ry+comfy_api.latest.generated.ComfyAPISyncStub comfy_api__annotations__rxr{r}__all__r,r,r,r-s<          b