import gradio as gr
from huggingface_hub import HfApi
import os

hf_api = HfApi()
CURRENT_MODEL_PATH = "./current_model"

HF_TOKEN = os.environ.get("HF_TOKEN", None)
REPO = os.getenv("REPO")
REVISION = os.getenv("REVISION")
TARGET = os.getenv("TARGET")

def func(hf_token, repo, revision, target_repo):
    hf_token = hf_token if len(hf_token)>0 else HF_TOKEN
    repo = repo if len(repo)>0 else REPO
    revision = revision if len(revision)>0 else REVISION
    target_repo = target_repo if len(target_repo)>0 else TARGET
    
    print('download the desired model locally ...')
    hf_api.snapshot_download(
        repo_id=repo,
        revision=revision,
        token=hf_token,
        local_dir=CURRENT_MODEL_PATH,
    )
    print(os.listdir(CURRENT_MODEL_PATH))
    
    return f"Using {hf_token}, the content of {repo}/{revision} has been copied to {target_repo}!" 

iface = gr.Interface(fn=func, inputs=["text","text","text"], outputs="text")
iface.launch()