temp.py

# import os
# from huggingface_hub import upload_folder

# # Hugging Face 토큰을 환경 변수로 설정
# hf_token = os.getenv("HF_TOKEN")

# # 현재 디렉토리 전체를 업로드
# folder_path = "."  # 현재 디렉토리

# # Hugging Face 리포지토리에 폴더 업로드
# upload_folder(
#     folder_path=folder_path,  # 업로드할 폴더 경로
#     path_in_repo="",          # 리포지토리 내 경로 (루트로 설정)
#     repo_id="yerang/txt2liveportrait",  # 리포지토리 ID
#     token=hf_token            # 환경 변수에서 토큰 사용
# )


import os
from huggingface_hub import upload_file
from tqdm import tqdm

# Hugging Face 토큰을 환경 변수로 설정
hf_token = os.getenv("HF_TOKEN")

# 현재 디렉토리 전체를 업로드
folder_path = "."  # 현재 디렉토리
repo_id = "yerang/txt2liveportrait"  # 리포지토리 ID

# 제외할 파일 크기 (단위: 바이트, 여기서는 100MB 이상인 파일을 제외)
max_file_size = 100 * 1024 * 1024  # 100MB

# 제외할 폴더 리스트
exclude_dirs = {".git", ".ipynb_checkpoints"}
exclude_files = {".gitignore", ".gitattributes", ".gitmodules"}

# 파일 목록 가져오기 (필터링 적용)
file_paths = []
for root, dirs, files in os.walk(folder_path):
    # 제외할 디렉토리 필터링
    dirs[:] = [d for d in dirs if d not in exclude_dirs]
    
    for file in files:
        if file not in exclude_files:
            full_path = os.path.join(root, file)
            # 파일 크기 확인 후, 설정한 크기 이하의 파일만 추가
            if os.path.getsize(full_path) < max_file_size:
                rel_path_in_repo = os.path.relpath(full_path, folder_path)
                file_paths.append((full_path, rel_path_in_repo))

# 파일 업로드 진행 상황 표시
for full_path, rel_path_in_repo in tqdm(file_paths, desc="Uploading files"):
    upload_file(
        path_or_fileobj=full_path,     # 실제 파일 경로
        path_in_repo=rel_path_in_repo,  # 리포지토리 내 경로
        repo_id=repo_id,               # 리포지토리 ID
        token=hf_token                 # Hugging Face 토큰
    )