abidlabs/trackio-123
0
1o
2 mj7h� � @ s` d dl Z d dlZd dlmZ d dlmZ d dlZd dlZ d dededB dedB fdd�Z dS )3� N)�files)�Path�title�persistent_dataset�persistent_dataset_dirc C s� t �d�dkr d S td�}t�� }d }d}z|�� }|d d d dkr&d }W n ty2 d }Y nw |rCtd4� tjdd � |�� }tj | d d
d d�j5}|| ksSJ �t t |d�d��$}|�
� } | �dtj�} t�| �d��}6|j|7d|d
d� W d � n1 s�w Y tj�� |j|d
|dgd� t j�d�} | d ur�t�|d| � |d ur�t�|d|� |d ur�t�|d|� d S d S )NZSYSTEM�spaces�trackioF�auth� accessToken�role�writeTz2Need 'write' access token to create a Spaces repo.)Zadd_to_git_credential�gradio�space)� space_sdk� repo_type�exist_okz README.md�rz{GRADIO_VERSION}zutf-8)�path_or_fileobj� path_in_repo�repo_idr )r r � folder_path�ignore_patterns�HF_TOKENZPERSIST_TO_DATASETZPERSIST_TO_DATASET_DIR) �os�getenvr �huggingface_hub�HfApi�whoami�OSError�print�login� create_repor �openr �read�replacer
� __version__�io�BytesIO�encode� upload_file�utils�disable_progress_bars�
upload_folder�environ�get�add_space_secret�add_space_variable) r r r Z trackio_path�hf_apir r �space_id�fZreadme_contentZ
readme_bufferr � r4 �-/Users/abidlabs/dev/trackio/trackio/deploy.py�deploy_as_space8 st �� � �� � �9 � ���r6 )NN)10r& r �importlib.resourcesr �pathlibr r
r �strr6 r4 r4 r4 r5 �<module> s ����