|
import gradio as gr |
|
from huggingface_hub import HfApi |
|
from git import Repo |
|
import uuid |
|
from slugify import slugify |
|
|
|
def clone(profile: gr.OAuthProfile, oauth_token: gr.OAuthToken, repo_git, repo_hf, sdk_type): |
|
folder = str(uuid.uuid4()) |
|
cloned_repo = Repo.clone_from(repo_git, folder) |
|
cloned_repo.git.submodule('update', '--init', '--recursive') |
|
|
|
|
|
api = HfApi(token=oauth_token.token) |
|
api.create_repo( |
|
f"{profile.username}/{slugify(repo_hf)}", |
|
repo_type="space", |
|
space_sdk=sdk_type |
|
) |
|
api.upload_folder( |
|
folder_path=folder, |
|
repo_id=f"{profile.username}/{slugify(repo_hf)}", |
|
repo_type="space", |
|
) |
|
return f"https://huggingface.co/spaces/{profile.username}/{slugify(repo_hf)}" |
|
|
|
|
|
with gr.Blocks() as demo: |
|
gr.LoginButton() |
|
with gr.Row(): |
|
with gr.Column(): |
|
repo_git = gr.Textbox(label="GitHub Repository") |
|
repo_hf = gr.Textbox(label="Hugging Face Space name") |
|
sdk_choices = gr.Radio(["gradio", "streamlit", "docker", "static"], label="SDK Choices") |
|
with gr.Column(): |
|
output = gr.Textbox(label="Output repo") |
|
btn = gr.Button("Bring over!") |
|
btn.click(fn=clone, inputs=[repo_git, repo_hf, sdk_choices], outputs=output) |
|
|
|
if __name__ == "__main__": |
|
demo.launch() |
|
|
