from huggingface_hub import HfApi
import os

api = HfApi(token=os.getenv("HF_TOKEN"))

def download():
    api.hf_hub_download(
        repo_type="dataset",
        repo_id="xcx0902/database",
        filename="database.db",
        local_dir="."
    )

def upload():
    api.upload_folder(
        folder_path=".",
        repo_type="dataset",
        repo_id="xcx0902/database",
        allow_patterns="database.db"
    )