o ۷i+@sddlmZddlZddlZddlZddlZddlmZddlZddl m Z ddl m Z ddl ZeZejedZejejedZdd dZdddZddZddZ   dd ddZdS)!) annotationsN) Annotated)print)Optionz README.mdzdeploy_space_action.yamltitle str | Noneapp_filereturndictc Csi}tjt}|durtd|dp|}t|}||kr&td|d||d<|durttD]?}tjt|}tj |rG| dsHq3t |ddd }| }d |vrc|} WdnWdn1smwYq3td |r}d |d nddp|}|rtj |std||d<d|d<tj|d<tt|tddddtjDdpd|d<i} tddkr td} | sn td | d} | | | <q| |d!<tjtd"} tj | std#dkr td$} | sn t | d%dd&}|| d'Wdn 1swYqtd(dkr}td)p(d*}tjtd+}tjtj|dd,t tdd& }| }Wdn 1sQwY|d-|}t |d.dd& }||Wdn 1stwYtd/|S)0NzEnter Spaces app title [z]: z Formatted to z. rz.pyutf-8ignore)encodingerrorsz import gradiozEnter Gradio app file []z: zFailed to find Gradio app file.rgradiosdk sdk_versionzEnter Spaces hardware (z, css|]}|jVqdS)N)value).0hardwarerV/home/ubuntu/vllm_env/lib/python3.10/site-packages/gradio/cli/commands/deploy_space.py >sz.add_configuration_to_readme..z) [cpu-basic]: z cpu-basicrzAny Spaces secrets (y/n) [n]: yTz(Enter secret name (leave blank to end): zEnter secret value for secretsrequirements.txtz)Create requirements.txt file? (y/n) [n]: z)Enter a dependency (leave blank to end): ar  zGCreate Github Action to automatically update Space on 'git push'? [n]: zEnter branch to track [main]: mainz".github/workflows/update_space.yml)exist_okz$branchwzGithub Action created. Add your Hugging Face write token (from https://huggingface.co/settings/tokens) as an Actions Secret named 'hf_token' to your GitHub repository. This can be set in your repository's settings page.)ospathbasenamerepo_directoryinput format_titlerlistdirjoinisfileendswithopenreadexistsFileNotFoundErrorgr __version__huggingface_hub metadata_save readme_file SpaceHardwarelowerwritemakedirsdirnamegithub_action_templatereplace)rr configurationdir_nameformatted_titlefile file_pathfcontentr secret_name secret_valuerequirements_file requirement track_branchgithub_action_filegithub_action_contentrrradd_configuration_to_readmes         rLstrcCsL|dd}tdd|}tdd|}|dr$|dd}|ds|S) N _z[^a-zA-Z0-9\-._]rz-+-.)r=resub startswith)rrrrr)qs    r)c Csxzatjgddddd}|js"tdtdtjddgddtjgd dddd}|jsFtd td }|sCtd Wd S|WStd|jdtd|jd|jWStjy}z@td|dtdz tjddgddWn!tjy}ztd|dWYd }~WYd }~dSd }~wwWYd }~dSd }~wtytdYdSw)zFCheck if user is logged in to Google Cloud and has a project selected.)gcloudauthlistz--filter=status:ACTIVETchecktextcapture_outputzA[bold yellow]You are not logged in to Google Cloud.[/bold yellow]z1Running 'gcloud init' to set up authentication...rVinit)rZ)rVconfigz get-valueprojectz?[bold yellow]No Google Cloud project is selected.[/bold yellow]z$Enter your Google Cloud project ID: z1[red]Project ID is required for deployment.[/red]Nu[green]✓ Authenticated as: z[/green]u[green]✓ Project: z5[bold red]Error checking Google Cloud configuration: z [/bold red]z0Running 'gcloud init' to set up configuration...z"[red]Failed to run 'gcloud init': [/red]FzK[bold red]gcloud CLI not found. Please install Google Cloud SDK.[/bold red]) subprocessrunstdoutstriprr(CalledProcessErrorr1) auth_resultproject_result project_ide init_errorrrrcheck_gcloud_authzs\      rkc Cstds tddSt}|stddStjds(tjds(tddSd}tj|smtd tj d  d krgt |d d d}| dtj dWdn1sYwYtdtj n[tddSt |d d }| }Wdn1swYd|vrtdtj d d krt |dd d}| dtj dWdn1swYtdtj dtdzgd}|r|d|gtj|ddddtdWdStjy}ztd |d!WYd}~dSd}~wty tdYdSw)"zODeploy a Gradio app to Google Cloud Run. Always uses app.py as the entry point.rVz[bold red]gcloud CLI is not installed.[/bold red] Please install the Google Cloud SDK from: [link]https://cloud.google.com/sdk/docs/install[/link]NzZ[bold red]Google Cloud configuration failed. Please run 'gcloud init' manually.[/bold red]app.pyzmain.pyz[bold red]Error:[/bold red] app.py and main.py not found. Google Cloud Run deployment requires app.py or main.py as the entry point.rzkA requirements.txt file is necessary for Google Cloud Run deployment. Create requirements.txt with gradio==z ? (y/n) [y]: nr#r rzgradio==r z&Created requirements.txt with gradio==z' [yellow]Deployment cancelled.[/yellow]rz Add gradio==z! to requirements.txt? (y/n) [y]: rzAdded gradio==z to requirements.txtz-[bold]Deploying to Google Cloud Run...[/bold])rVrbdeployz --source=.z--labels=created-by=gradioz --projectTFrYu'[green]✓ Deployment complete![/green]z[red]Deployment failed: r`)shutilwhichrrkr$r%r0r(r2r3r8r.r9r/extendrarbreKeyboardInterrupt)rhrGrCrequirements_contentdeploy_commandrirrrdeploy_to_gclouds     ruspaces6Annotated[str | None, Option(help='Spaces app title')]DAnnotated[str | None, Option(help='File containing the Gradio app')]providerLAnnotated[str | None, Option(help='Deployment provider (spaces or gcloud)')]c Cs|dkr|r|dkrtdtdS|dkr!td|ddStddkr*dSt}d}d}z|}|d d d d krCd }Wn tyOd }Ynw|r`tdtjdd|}d}tj t ryzt t }Wn t yxYnw|durtdtdt||}|j|dddd |ddj}|j|dtd|dr|dD] \}} |||| qtd|dS)NrVrlzm[yellow]Warning:[/yellow] --app-file is ignored for Google Cloud Run deployment. Using app.py as entry point.rvz[red]Unknown provider: z!. Use 'spaces' or 'gcloud'.[/red]SYSTEMFrW accessTokenroler9Tz2Need 'write' access token to create a Spaces repo.)add_to_git_credentialzCreating new Spaces Repo in 'z<'. Collecting metadata, press Enter to accept default value.rrspacer) space_sdk repo_typer"space_hardware)repo_idr folder_pathrz1Space available at https://huggingface.co/spaces/)rrur$getenvr4HfApiwhoamiOSErrorloginr%r0r6 metadata_load ValueErrorr'rL create_repogetr upload_folderitemsadd_space_secret) rrryhf_apirrr>space_idrErFrrrrnsx       rn)rrrrr r )rrM)NNrv)rrwrrxryrz) __future__rr$rSroratypingrr4richrtyperrrr2getcwdr'r%r+r6r;__file__r<rLr)rkrurnrrrrs.       [ 3P