deploy.py

"""
🚀 지니젠AI Hugging Face Space 자동 배포 스크립트

사용법:
    1. pip install huggingface_hub
    2. 환경변수에 토큰 설정 (보안!):
       export HF_TOKEN=your_token_here
    3. python deploy.py

⚠️ 보안 경고:
    - 토큰을 이 파일에 직접 작성하지 마세요!
    - 환경변수 또는 `huggingface-cli login` 사용 권장
    - 토큰이 노출되었다면 즉시 https://huggingface.co/settings/tokens 에서 revoke
"""

import os
import sys
from pathlib import Path

try:
    from huggingface_hub import HfApi, create_repo
except ImportError:
    print("❌ huggingface_hub 설치 필요: pip install huggingface_hub")
    sys.exit(1)


# -------------------------------------------------------------
# 설정
# -------------------------------------------------------------
SPACE_NAME = "geniegen-ai"  # 원하는 Space 이름으로 변경
SPACE_SDK = "docker"        # FastAPI + HTML을 위해 docker SDK 사용

# 업로드할 파일 목록 (이 스크립트와 같은 디렉토리에 있어야 함)
FILES_TO_UPLOAD = [
    "app.py",
    "index.html",
    "requirements.txt",
    "README.md",
    "Dockerfile",
]


def get_token():
    """환경변수 또는 CLI 로그인에서 토큰 가져오기"""
    token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
    if not token:
        print("⚠️  HF_TOKEN 환경변수가 설정되지 않았습니다.")
        print("   다음 중 하나를 수행하세요:")
        print("   1) export HF_TOKEN=hf_xxx  (리눅스/맥)")
        print("   2) set HF_TOKEN=hf_xxx      (윈도우 CMD)")
        print("   3) huggingface-cli login    (대화형 로그인)")
        sys.exit(1)
    return token


def get_username(api: HfApi):
    """로그인된 사용자명 조회"""
    info = api.whoami()
    return info["name"]


def deploy():
    print("=" * 60)
    print("🚀 지니젠AI Hugging Face Space 배포 시작")
    print("=" * 60)

    # 1. 토큰 & API 초기화
    token = get_token()
    api = HfApi(token=token)

    try:
        username = get_username(api)
        print(f"✅ 로그인 확인: {username}")
    except Exception as e:
        print(f"❌ 로그인 실패: {e}")
        print("   토큰이 유효한지 확인하세요.")
        sys.exit(1)

    repo_id = f"{username}/{SPACE_NAME}"
    print(f"📦 대상 Space: {repo_id}")

    # 2. Space 생성 (이미 있으면 스킵)
    try:
        create_repo(
            repo_id=repo_id,
            repo_type="space",
            space_sdk=SPACE_SDK,
            token=token,
            exist_ok=True,
            private=False,
        )
        print(f"✅ Space 준비 완료 (SDK: {SPACE_SDK})")
    except Exception as e:
        print(f"❌ Space 생성 실패: {e}")
        sys.exit(1)

    # 3. 파일 존재 확인
    base_dir = Path(__file__).parent
    missing = [f for f in FILES_TO_UPLOAD if not (base_dir / f).exists()]
    if missing:
        print(f"❌ 누락된 파일: {missing}")
        sys.exit(1)

    # 4. 파일 업로드
    print(f"\n📤 파일 {len(FILES_TO_UPLOAD)}개 업로드 중...")
    for filename in FILES_TO_UPLOAD:
        filepath = base_dir / filename
        try:
            api.upload_file(
                path_or_fileobj=str(filepath),
                path_in_repo=filename,
                repo_id=repo_id,
                repo_type="space",
                token=token,
                commit_message=f"Add {filename}",
            )
            size_kb = filepath.stat().st_size / 1024
            print(f"   ✓ {filename:<20} ({size_kb:.1f} KB)")
        except Exception as e:
            print(f"   ✗ {filename} 업로드 실패: {e}")

    # 5. 완료 안내
    space_url = f"https://huggingface.co/spaces/{repo_id}"
    print("\n" + "=" * 60)
    print("🎉 배포 완료!")
    print("=" * 60)
    print(f"🔗 Space URL: {space_url}")
    print(f"📊 빌드 상태: {space_url} 페이지에서 로그 확인 가능")
    print(f"⏱  Docker 빌드는 2~5분 소요됩니다")
    print("=" * 60)


if __name__ == "__main__":
    deploy()