o ٷi- @s&ddlmZddlZddlZddlZddlmZddlmZddl Z ddl m Z ddl m ZddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZmZdd lmZgdZdZ ddZ!d@ddZ"dAddZ#dBdd Z$ed!d"d#d$d$d#ded!dd%d%df dCdd?Z&dS)E) annotationsN)Path) Annotated)HfApi)Version)print)Console)Panel)ConfirmPrompt)parse)ArgumentOption)custom_component_analytics)redyellowgreenblueindigopurplepinkgrayz"https://pypi.org/account/register/cCs8g}|D]}d|vs|ds|dr||q|S)N __pycache__dist node_modules) startswithappend)_srcnamesignorednr!Z/home/ubuntu/.local/lib/python3.10/site-packages/gradio/cli/commands/components/publish.py_ignores  r# dist_filerreturn str | NonecCs td|j}|r|dSdS)Nz-(\d+\.\d+\.\d+[a-zA-Z]*\d*)-)researchnamegroup)r$matchr!r!r"_get_version_from_file"s r-version_stringstr/semantic_version.Version | PEP440Version | Nonec CsBzt|WSty zt|WYStyYYdSww)zCParse a version string, trying semantic_version first, then PEP440.N)semantic_versionr ValueError PEP440Version Exception)r.r!r!r"_parse_version)s   r5distribution_files list[Path]cCsLg}|D]}t|}|rt|}|r||qdSq|r$tt|SdS)N)r-r5rr/max)r6versionspversionparsedr!r!r"_get_max_version6s r=.rTFdist_dir_Annotated[Path, Argument(help=f"Path to the wheel directory. Default is {Path('.') / 'dist'}")] upload_pypi:Annotated[bool, Option(help='Whether to upload to PyPI.')] pypi_username5Annotated[str, Option(help='The username for PyPI.')] pypi_password5Annotated[str, Option(help='The password for PyPI.')] upload_demoFAnnotated[bool, Option(help='Whether to upload demo to HuggingFace.')]demo_dirBAnnotated[Path | None, Option(help='Path to the demo directory.')] source_dirUAnnotated[Path, Option(help='Path to the source directory of the custom component.')]hf_tokenAnnotated[str | None, Option(help='HuggingFace token for uploading demo. Can be omitted if already logged in via huggingface cli.')] prefer_localvAnnotated[bool, Option(help='Install the package from the local wheel in the demo space, even if it exists on PyPi.')] upload_sourcewAnnotated[bool, Option(help='Whether to upload the source code of the custom component, to share with the community.')]repo_idAnnotated[str | None, Option(help='The repository id to upload the demo to. If not provided, a space will be created with the same name as the package in the HuggingFace account corresponding to the hf_token.')]c stdd||| dt} |}|st|d|s&t|dddt|dD} dDd d } td d | D| d}|sHtdd}|r|rP|st dt d}t |t d}|rxtdrxt dttd}n|rt dt d}tj ddd}|r zddlm}ddlm}ddlm}Wnttfy}ztd|d}~ww|r|r|||d}n |r||d }ntd!z2t| fd"d| D}t d#d$|z|||Wn|y}ztd%|d}~wwWnty | Ynw|rH|sHt d&}t |t d'}|rHt d(}t |t d)td*d+}|pAttd*d+}t|}|ro| sot d,}t |t d-} | rot d.|ph|}t|}|rE|d/}zt|}|d0d1}Wnty|j d2dd3\}}Ynw|std4|j d2dd3\}}t!"}t#j$t|t|dd5t#%t|d6tt|d6| rt#j$t|tt|d7dt&d8t#%t|d9tt|d9t'|d:}|j(| p|d;dd|r|j+| t||jd;d?t d@|j,| d|dAt dB| dCWddS1s>wYdSdS)ENpublish)rHrBrRzQ does not exist. Run `gradio cc build` to create a wheel and source distribution.z is not a directorycSsg|] }|jdvr|qS)>.gz.whl)suffixresolve.0r:r!r!r" ~sz_publish..*pathrcSs>t|jdd}t|}|durtd|d|j|S)N-r'zCould not parse version z from )r/r*splitr5r2)r_ version_strr<r!r!r"_get_wheel_versions z$_publish.._get_wheel_versioncss|] }|jdkr|VqdS)rXN)rYr[r!r!r" sz_publish..)keyzgA wheel file was not found in the distribution directory. Run `gradio cc build` to create a wheel file.zIt is recommended to upload your component to pypi so that [bold][magenta]anyone[/][/] can install it with [bold][magenta]pip install[/][/]. A PyPi account is needed. If you do not have an account, register account here: [blue]z[/]z:snake: Upload to pypi?z.pypircz<:closed_lock_with_key: Found .pypirc file in home directory.zx:light_bulb: If you have Two Factor Authentication enabled, the username is __token__ and your password is your API key.z*:laptop_computer: Enter your pypi usernamez/:closed_lock_with_key: Enter your pypi passwordT)passwordr)upload)InvalidDistribution)Settingsz^The twine library must be installed to publish to pypi.Install it with pip, pip install twine.)usernamerf) config_filezBNo pypi username or password provided and no ~/.pypirc file found.cs"g|] }r |jvrt|qSr!)r*r/r[ max_versionr!r"r]s zUploading files: ,zqInvalid distribution when uploading to pypi. Try upgrading 'pkginfo' with python -m pip install pkginfo --upgradezIt is recommended you upload a demo of your component to [blue]https://huggingface.co/spaces[/] so that anyone can try it from their browser.z:hugging_face: Upload demo?aXPlease provide the path to the [magenta]demo directory[/] for your custom component. This directory should contain [magenta]all the files[/] it needs to run successfully. Please make sure the gradio app is in an [magenta]space.py[/] file. If you need additional python requirements, add a [magenta]requirements.txt[/] file to this directory.zR:roller_coaster: Please enter the path to the demo directory. Leave blank to use: r>demozwIt is recommended that you share your [magenta]source code[/] so that others can learn from and improve your component.z1:books: Would you like to share your source code?zQ:file_folder: Please enter the path to the source directory. Leave blank to use: zpyproject.tomlprojectr*r`zdemo_dir must be set) dirs_exist_okz .gitignoresrc)rrignorez README.md)tokenspacegradio)rT repo_typeexist_ok space_sdk)rT folder_pathrx)rTpath_or_fileobj path_in_reporx )rTfactory_rebootruz/Demo uploaded to https://huggingface.co/spaces/z !)r_r)-rrrZexistsr2is_dirrglobr8r PYPI_REGISTER_URLrr askhomer/r twine.commands.uploadrgtwine.exceptionsrhtwine.settingsri ImportErrorModuleNotFoundErrorr=joinr4print_exceptionr read_textr*ratempfileTemporaryDirectoryshutilcopytreecopyfiler#r create_reporT upload_folder upload_file restart_space) r@rBrDrFrHrJrLrNrPrRrTconsoler6rc wheel_filerkpanel twine_uploadrhrietwine_settings twine_files demo_dir_ source_dir_pyproject_toml_pathpyproject_toml package_name_tempdirapirepo_urlr!rlr"_publishEsX)                     $rcCs@|}|dr|dS|dr|dStd|d)Nzspace.pyzapp.pyz*Could not find "space.py" or "app.py" in "z".)rZrFileNotFoundError)rJ _demo_dirr!r!r" resolve_demo+s   r)r$rr%r&)r.r/r%r0)r6r7r%r&)r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQrRrSrTrU)rJrr%r)' __future__rr(rrpathlibrtypingrr1huggingface_hubrpackaging.versionrr3richr rich.consoler rich.panelr rich.promptr r tomlkitr typerr rgradio.analyticsrcolorsrr#r-r5r=rrr!r!r!r"sH              g