+j9ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZddlmZddlmZddlZddlmZdd lmZmZdd lmZddlZd Z d e!d e"e#e#e#ffdZ$d e!d e%fdZ&dZ'gZ(dZ)dZ*dZ+GddeZ,GddeZ-e GddZ.de-de!d dfdZ/GddZ0dS)N) dataclass)cached_property)Path)Dict TypedDictOptional)webversion) NotRequired) get_missing_requirements_messageget_required_packages_versions)DEFAULT_VERSION_STRINGcJdtdS)N zo The ComfyUI frontend is shipped in a pip package so it needs to be updated separately from the ComfyUI code. )r strip7/home/wildlama/comfy/ComfyUI/app/frontend_management.py frontend_install_warning_messagers/ !##    EGG  rr returncbttt|jdS)N.)tuplemapintsplitr s r parse_versionr s&SmgmC0011222rcJd}ttj||S)z@Validate if a string is a valid semantic version (X.Y.Z format).z^(\d+)\.(\d+)\.(\d+)$)boolrematch)r patterns ris_valid_versionr$#s!&G '** + ++rcFtddS)Nzcomfyui-frontend-packagergetrrrget_required_frontend_versionr((s ) + + / /0JD Q QQrcltrtSt}tpiD]W\}}|ds t |}n#t $rd}YnwxYw||||dX|S)NList installed/required versions for every comfy* package in requirements.txt.comfyN)name installedrequired)COMFY_PACKAGE_VERSIONScopyritems startswithr Exceptionappend)outr,r.r-s rget_comfy_package_versionsr6-s-%**,,, C9;;ArHHJJQQhw''    II   III  DyhOOPPPP 88::s&A66 BBcjddlm}m}g}tD]}|d}|d}|r|s ||||k}n2#|$r*}t jd|dd|Yd}~[d}~wwxYw|r||d||ft jd |d||red d |D}tj d |d tddSdS)zPWarn for every comfy* package whose installed version is below requirements.txt.r)InvalidVersionparser-r.zFailed to check r,z version: Nz{} version: {}rc36K|]\}}}d|d|d|dVdS)z Installed z version z' is lower than the recommended version rNr).0r,r-r.s r z0check_comfy_packages_versions..RsY% % )i f e e e eZb e e e% % % % % % rzs ________________________________________________________________________ WARNING WARNING WARNING WARNING WARNING z zJ ________________________________________________________________________ )packaging.versionr8r9r6loggingerrorr4infoformatjoinapploggerlog_startup_warningr r) r8 parse_pep440outdated_packagespkg installed_str required_stroutdatedepackage_warningss rcheck_comfy_packages_versionsrN=sGGGGGGGG)++ N NK( :  L   #|M22\\,5O5OOHH    MGS[GGAGG H H H HHHH   N  $ $c&k=,%O P P P P L)00VmLL M M M M 99% % ->% % %    &&     "##     EGG   sA  A9 A44A9 ceZdZUeed<dS)AsseturlN)__name__ __module__ __qualname__str__annotations__rrrrQrQfs HHHHHrrQcxeZdZUeed<eed<eed<eed<eed<eed<eed<eee ed<d S) Releaseidtag_namer, prerelease created_at published_atbodyassetsN) rSrTrUrrWrVr r listrQrrrrYrYjsl GGGMMM IIIOOO III U $$$$$$rrYceZdZUeed<eed<edefdZedefdZede e fdZ ede fdZ ede fdZ d ede fd Zd S) FrontEndProviderownerreporc$|jd|jS)N_rdreselfs r folder_namezFrontEndProvider.folder_namezs***ty***rc(d|jd|jdS)Nzhttps://api.github.com/repos//z /releasesrhris r release_urlzFrontEndProvider.release_url~sPtzPPDIPPPPrc g}|j}|rwtj|t}|||d|jvr|jdd}nd}|w|S)NtimeoutnextrR)rnrequestsr'REQUEST_TIMEOUTraise_for_statusextendjsonlinks)rjreleasesapi_urlresponses r all_releaseszFrontEndProvider.all_releasess" |G_EEEH  % % ' ' ' OOHMMOO , , ,''".07 rc|jd}tj|t}||S)Nz/latestrp)rnrsr'rtrurw)rjlatest_release_urlr{s rlatest_releasezFrontEndProvider.latest_releasesI $ 0999< 2OLLL!!###}}rcVd|jD}|std|dS)zOGet the latest pre-release version - even if it's older than the latest releasec"g|] }|d | S)r\r)r;releases r z6FrontEndProvider.latest_prerelease..s"UUUww|?TU7UUUrzNo pre-releases foundr)r| ValueError)rjrs rlatest_prereleasez"FrontEndProvider.latest_prereleases>VU$*;UUU 6455 5qzrr c|dkr|jS|dkr|jS|jD]}|d|d|fvr|cStd|d)Nlatestr\r[vzVersion z not found in releases)rrr|r)rjr rs r get_releasezFrontEndProvider.get_releases h  & &  $ $) ), # #:&7MMM*BBB"NNNCGGGGHH HrN)rSrTrUrVrWpropertyrkrnrrarYr|rrrrrrrcrcus JJJ III +S+++X+QSQQQXQ d7m   _ _ 7_ I3 I7 I I I I I Irrcrdestination_pathc.d}|dgD]}|ddkr |d}n|stdtj5}ddi}t j||d t }|||j| d tj |d 5}| |dddn #1swxYwYddddS#1swxYwYdS) z&Download dist.zip from github release.Nr`r,zdist.ziprRz(dist.zip not found in the release assetsAcceptzapplication/octet-streamT)headersallow_redirectsrqrr) r'rtempfile TemporaryFilersrtruwritecontentseekzipfileZipFile extractall)rr asset_urlassettmp_filerr{zip_refs rdownload_release_asset_ziprsIXr** =J & &e I E ' ECDDD   ! !1X78< wo    !!### x'(((  a_Xs + + 1w   / 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1111111111111111111s7A:D C2& D 2C6 6D 9C6 :D  DDceZdZeeejddz ZedefdZ edefdZ edefdZ edZ edefdZ edeeeeffd Zedeefd Zedefd Zed edeeeeffd Ze ddedeedefdZededefdZedZdS)FrontendManagerweb_custom_versionsrctS)z*Get the required frontend package version.)r(clss rr(z-FrontendManager.get_required_frontend_versions-...rcH td}|S#t$rYdSwxYw)z?Get the currently installed workflow templates package version.comfyui-workflow-templatesN)r r3)rtemplates_version_strs rget_installed_templates_versionz/FrontendManager.get_installed_templates_versions> $+,H$I$I !( (   44 s  !!cFtddS)Nrr&rs rget_required_templates_versionz.FrontendManager.get_required_templates_versions -//334PRVWWWrctS)r*)r6rs rr6z*FrontendManager.get_comfy_package_versionss*+++rc  ddl}ttj|dz S#t $rNt jdtd tj dYdSwxYw)NrstaticzK ********** ERROR *********** comfyui-frontend-package is not installed.  ********** ERROR *********** ) comfyui_frontend_packagerV importlib resourcesfiles ImportErrorr>r?rrsysexit)rrs rdefault_frontend_pathz%FrontendManager.default_frontend_paths  + + + +y*001IJJXUVV V    M "##  EGG    HRLLLLLL s25AB  B c( ddlm}m}nG#t$r:t jdt dYdSwxYw t|}n/#t$r"}t jd|Yd}~dSd}~wwxYwi} |D]*}|j D] }||j |j ||j <!+n/#t$r"}t jd|Yd}~dSd}~wwxYw|st jddS|S) zAReturn a mapping of template asset names to their absolute paths.r)get_asset_pathiter_templatesM ********** ERROR *********** comfyui-workflow-templates is not installed. rNz(Failed to enumerate workflow templates: z(Failed to resolve template asset paths: zFNo workflow template assets found. Did the packages install correctly?) comfyui_workflow_templatesrrrr>r?rrrar3r` template_idfilename)rrrtemplate_entriesexc asset_mapentryrs rtemplate_asset_mapz"FrontendManager.template_asset_maps              M "##  EGG   44  #NN$4$455      MJSJJ K K K44444 %'  )  "\E0>)5>11Ien--      MJSJJ K K K44444   Mb c c c4s@ AAAA++ B5BB-C C7C22C7c ddl}ttj|dz S#t $r:t jdtd YdSwxYw)zFReturn the legacy templates directory shipped inside the meta package.rN templatesrr) rrVrrrrr>r?rr)rrs rlegacy_templates_pathz%FrontendManager.legacy_templates_path)s  - - - -#))*DEE S     M "##  EGG   44 s25AA98A9c ddl}ttj|dz S#t $rt jdYdSwxYw)z&Get the path to embedded documentationrNdocsz'comfyui-embedded-docs package not found)comfyui_embedded_docsrVrrrrr>r@)rrs rembedded_docs_pathz"FrontendManager.embedded_docs_path@st  ( ( ( (#))*?@@6I     LB C C C44 s25AAvaluecd}tj||}|tjd||d|d|dfS)a Args: value (str): The version string to parse. Returns: tuple[str, str]: A tuple containing provider name and version. Raises: argparse.ArgumentTypeError: If the version string is invalid. zf^([a-zA-Z0-9][a-zA-Z0-9-]{0,38})/([a-zA-Z0-9_.-]+)@(v?\d+\.\d+\.\d+[-._a-zA-Z0-9]*|latest|prerelease)$NzInvalid version string: r)r!r"argparseArgumentTypeErrorgroup)rrVERSION_PATTERN match_results rparse_version_stringz$FrontendManager.parse_version_stringMssDx77  ,-O-O-OPP P!!!$$l&8&8&;&;\=O=OPQ=R=RRRrNversion_stringproviderc V|tkr"t|S||\}}}|jdrxt t |j|d|z |jdz }tj |rtj d|d|d||Stj d|d|d|d|pt||}||}|dd}t t |j|jz |z } tj | s tj| d tj d |j|| tj|t'|| tj| stj| n.#tj| stj| wwxYw| S) a Initializes the frontend for the specified version. Args: version_string (str): The version string. provider (FrontEndProvider, optional): The provider to use. Defaults to None. Returns: str: The path to the initialized frontend. Raises: Exception: If there is an error during the initialization process. main error source might be request timeout or invalid URL. rrgz6Using existing copy of specific frontend version tag: rm@zInitializing frontend: z+, requesting version details from GitHub...r[T)exist_okz,Downloading frontend(%s) version(%s) to (%s))r)rrNrrr2rVrCUSTOM_FRONTENDS_ROOTlstripospathexistsr>r@rcrrkmakedirsdebugrlistdirrmdir) rrr repo_owner repo_namer expected_pathrsemantic_versionweb_roots rinit_frontend_unsafez$FrontendManager.init_frontend_unsafe`sk$ 3 3 3 ) + + +,,.. .),)A)A.)Q)Q& Iw 7 c " " %S.//--)--. '.%%&M w~~m,, % oZooZcoofmoo%$ sj s s9 s sw s s s   F/ IFF&&w//":.55c:: * + +h.B BEU U  w~~h'' ' ' Ht4444 B($   g&&&*7XNNNNz(++'HX&&&z(++'HX&&&&'s ;AG;;+H&c ||S#t$rU}tjd|tjdt |cYd}~Sd}~wwxYw)z Initializes the frontend with the specified version string. Args: version_string (str): The version string to initialize the frontend with. Returns: str: The path of the initialized frontend. z!Failed to initialize frontend: %sz%Falling back to the default frontend.N)rr3r>r?r@rNr)rrrLs r init_frontendzFrontendManager.init_frontends /++N;; ; / / / M=q A A A L@ A A A ) + + +,,.. . . . . . .  /s A6A A1+A61A6ct|sdSdtjdtjffd }|S)NrequestrcK|jdd}|}|tjtj|S)Nr) match_infor'r HTTPNotFound FileResponse)rrel_pathtargetr`s rserve_templatez>FrontendManager.template_asset_handler..serve_templatesU)--fb99HZZ))F~&(((#F++ +r)rr RequestStreamResponse)rrr`s @rtemplate_asset_handlerz&FrontendManager.template_asset_handlersZ'')) 4 ,#+ ,#:L , , , , , ,r)N)rSrTrUrVr__file__parentsr classmethodr(rrr6rrrrrrrrrcrrrrrrrrsJCX 6q 9> >,45E,F> >>>[>@/3/3///[/"  [   rr)1rr>rr!rrrr dataclassesr functoolsrpathlibrtypingrrraiohttpr importlib.metadatar rstyping_extensionsr utils.install_utilr rcomfy.cli_argsr app.loggerrCrrVrrrr r$r(r/r6rNrtrQrYrcrrrrrrs !!!!!!%%%%%%,,,,,,,,,,&&&&&&))))))________111111   3335c3#73333,c,d,,,, RRR    # # # L     I   %%%%%i%%% 6I6I6I6I6I6I6I 6Ir113141111