o i7@sZddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZddlmZddlmZddlZdd lmZdd lmZmZdd lmZddl Z!d d Z"d+ddZ#d,ddZ$ddZ%ddZ&ddZ'dZ(GdddeZ)Gd d!d!eZ*e Gd"d#d#Z+d-d'd(Z,Gd)d*d*Z-dS).) annotationsN) dataclass)cached_property)Path)Dict TypedDictOptional)webversion) NotRequired) get_missing_requirements_messageget_required_packages_versions)DEFAULT_VERSION_STRINGcCsdtdS)N z This error is happening because the ComfyUI frontend is no longer shipped as part of the main repo but as a pip package instead. )r striprr5/mnt/c/Users/fbmor/ComfyUI/app/frontend_management.py frontend_install_warning_messages rr strreturntuple[int, int, int]cCsttt|dS)N.)tuplemapintsplitr rrr parse_version!srboolcCsd}tt||S)z@Validate if a string is a valid semantic version (X.Y.Z format).z^(\d+)\.(\d+)\.(\d+)$)rrematch)r patternrrris_valid_version$sr"cCs td}|S)z5Get the currently installed frontend package version.comfyui-frontend-packager )frontend_version_strrrrget_installed_frontend_version)sr%cCtddS)Nr#rgetrrrrget_required_frontend_version/sr)c Csz?t}t|}t}t|}||kr5tjddtt|ddtt|dt d WdSt d |WdStyZ}zt d|WYd}~dSd}~ww) z,Check if the frontend version is up to date.z ________________________________________________________________________ WARNING WARNING WARNING WARNING WARNING Installed frontend version rz' is lower than the recommended version z. zJ ________________________________________________________________________ zComfyUI frontend version: {}z"Failed to check frontend version: N)r%rr)apploggerlog_startup_warningjoinrrrrlogginginfoformat Exceptionerror)r$frontend_versionrequired_frontend_strrequired_frontenderrrcheck_frontend_version3s*  r7 c@seZdZUded<dS)AssetrurlN__name__ __module__ __qualname____annotations__rrrrr9Ps r9c@sNeZdZUded<ded<ded<ded<ded<ded <ded <d ed <d S)Releaseridrtag_namenamer prerelease created_at published_atbodyzNotRequired[list[Asset]]assetsNr;rrrrr@Ts  r@c@sneZdZUded<ded<edddZedddZedd d Zedd dZ edddZ dddZ dS)FrontEndProviderrownerreporcCs|jd|jS)N_rJrKselfrrr folder_namedszFrontEndProvider.folder_namecCsd|jd|jdS)Nzhttps://api.github.com/repos//z /releasesrMrNrrr release_urlhszFrontEndProvider.release_url list[Release]cCsXg}|j}|r*tj|td}|||d|jvr&|jdd}nd}|s|S)Ntimeoutnextr:)rRrequestsr(REQUEST_TIMEOUTraise_for_statusextendjsonlinks)rOreleasesapi_urlresponserrr all_releasesls  zFrontEndProvider.all_releasesr@cCs*|jd}tj|td}||S)Nz/latestrT)rRrWr(rXrYr[)rOlatest_release_urlr_rrrlatest_release{s zFrontEndProvider.latest_releasecCs$dd|jD}|std|dS)zOGet the latest pre-release version - even if it's older than the latest releasecSsg|]}|dr|qS)rDr).0releaserrr sz6FrontEndProvider.latest_prerelease..zNo pre-releases foundr)r` ValueError)rOrdrrrlatest_prereleasesz"FrontEndProvider.latest_prereleaser cCsV|dkr|jS|dkr|jS|jD]}|d|d|fvr"|Sqtd|d)NlatestrDrBvzVersion z not found in releases)rbrgr`rf)rOr rdrrr get_releases zFrontEndProvider.get_releaseNrr)rrS)rr@)r rrr@) r<r=r>r?propertyrPrRrr`rbrgrjrrrrrI_s       rIrddestination_pathNonec Csd}|dgD]}|ddkr|d}nq|stdtH}ddi}tj||d td }|||j| d t |d  }| |Wdn1sTwYWddSWddS1slwYdS) z&Download dist.zip from github release.NrHrCzdist.zipr:z(dist.zip not found in the release assetsAcceptzapplication/octet-streamT)headersallow_redirectsrUrr) r(rftempfile TemporaryFilerWrXrYwritecontentseekzipfileZipFile extractall)rdrm asset_urlassettmp_filerpr_zip_refrrrdownload_release_asset_zips,     "rc@seZdZeeejddZed#ddZ ed#ddZ ed#d d Z ed#d d Z ed$ddZ ed%ddZed#ddZed&ddZe d'd(ddZed)dd Zed!d"ZdS)*FrontendManagerweb_custom_versionsrrcCstS)z*Get the required frontend package version.)r)clsrrrr)sz-FrontendManager.get_required_frontend_versioncCs$ztd}|WStyYdSw)z?Get the currently installed workflow templates package version.comfyui-workflow-templatesN)r r1)rtemplates_version_strrrrget_installed_templates_versions  z/FrontendManager.get_installed_templates_versioncCr&)Nrr'rrrrget_required_templates_versionsz.FrontendManager.get_required_templates_versioncCsVzddl}ttj|dWSty*tdtd t dYdSw)NrstaticzK ********** ERROR *********** comfyui-frontend-package is not installed.  ********** ERROR *********** ) comfyui_frontend_packager importlib resourcesfiles ImportErrorr.r2rrsysexit)rrrrrdefault_frontend_paths  z%FrontendManager.default_frontend_pathOptional[Dict[str, str]]c Csz ddlm}m}Wnty tdtdYdSwzt|}Wnt yC}ztd|WYd}~dSd}~wwi}z|D]}|j D] }||j |j ||j <qNqIWnt yx}ztd|WYd}~dSd}~ww|stddS|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.r2rrlistr1rH template_idfilename)rrrtemplate_entriesexc asset_mapentryr|rrrtemplate_asset_mapsH     z"FrontendManager.template_asset_map Optional[str]cCsLzddl}ttj|dWSty%tdtd YdSw)zFReturn the legacy templates directory shipped inside the meta package.rN templatesrr) rrrrrrr.r2rr)rrrrrlegacy_templates_paths  z%FrontendManager.legacy_templates_pathcCs>zddl}ttj|dWStytdYdSw)z&Get the path to embedded documentationrNdocsz'comfyui-embedded-docs package not found)comfyui_embedded_docsrrrrrr.r/)rrrrrembedded_docs_path%s  z"FrontendManager.embedded_docs_pathvaluetuple[str, str, str]cCsDd}t||}|durtd||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)rr argparseArgumentTypeErrorgroup)rrVERSION_PATTERN match_resultrrrparse_version_string2s  z$FrontendManager.parse_version_stringNversion_stringproviderOptional[FrontEndProvider]c CsX|tkr t|S||\}}}|drAtt|j|d||d}t j |rAt d|d|d||St d|d|d|d|pVt||}||}|dd}tt|j|j|} t j | sz(t j| d d t d |j|| t |t|| d Wt | st | | St | st | ww| 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. rirLz6Using existing copy of specific frontend version tag: rQ@zInitializing frontend: z+, requesting version details from GitHub...rBT)exist_okz,Downloading frontend(%s) version(%s) to (%s))rm)rr7rr startswithrrCUSTOM_FRONTENDS_ROOTlstripospathexistsr.r/rIrjrPmakedirsdebugrlistdirrmdir) rrr repo_owner repo_namer expected_pathrdsemantic_versionweb_rootrrrinit_frontend_unsafeEsX         z$FrontendManager.init_frontend_unsafec CsVz||WSty*}ztd|tdt|WYd}~Sd}~ww)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)rr1r.r2r/r7r)rrr6rrr init_frontends   zFrontendManager.init_frontendcs"|sdSdfdd }|S)Nrequest web.Requestrweb.StreamResponsecs4|jdd}|}|durtt|S)Nr) match_infor(r HTTPNotFound FileResponse)rrel_pathtargetrHrrserve_templates   z>FrontendManager.template_asset_handler..serve_template)rrrr)r)rrrrrtemplate_asset_handlers z&FrontendManager.template_asset_handlerrk)rr)rr)rrrr)N)rrrrrr)rrrr)r<r=r>rr__file__parentsr classmethodr)rrrrrrrrrrrrrrrs2     -   @ r)r rrr)r rrr)rdr@rmrrrn). __future__rrr.rrrrsrxr dataclassesr functoolsrpathlibrtypingrrraiohttpr importlib.metadatar rWtyping_extensionsr utils.install_utilr rcomfy.cli_argsr app.loggerr*rrr"r%r)r7rXr9r@rIrrrrrrsB            9