app.py

import gradio as gr
import os
import sys
from pathlib import Path

# repo_to_space_auto 모듈 동적 임포트
sys.path.insert(0, str(Path(__file__).parent))
from repo_to_space_auto import deploy

# OAuth 사용 가능 여부 확인
OAUTH_ENABLED = os.getenv("OAUTH_CLIENT_ID") is not None

def launch_deploy_oauth(repo_url, private, profile: gr.OAuthProfile | None):
    """OAuth 인증 버전"""
    if not profile:
        return """### ❌ 로그인 필요
        
상단의 'Sign in with Hugging Face' 버튼을 클릭해 로그인해주세요."""
    
    hf_token = profile.token
    
    # 토큰 정리 (OAuth에서도 안전하게)
    if hf_token:
        hf_token = hf_token.strip()
    
    return _deploy_space(repo_url, hf_token, private)

def launch_deploy_manual(repo_url, hf_token, private):
    """수동 토큰 입력 버전"""
    if not hf_token:
        return """### ❌ 토큰 필요
        
Hugging Face Write 토큰을 입력해주세요."""
    
    # 토큰 정리 (공백, 줄바꿈 제거)
    hf_token = hf_token.strip()
    
    # 토큰 형식 검증
    if not hf_token.startswith("hf_"):
        return """### ❌ 잘못된 토큰 형식
        
토큰은 'hf_'로 시작해야 합니다.
https://huggingface.co/settings/tokens 에서 Write 권한 토큰을 생성해주세요."""
    
    return _deploy_space(repo_url, hf_token, private)

def _deploy_space(repo_url, hf_token, private):
    """공통 배포 로직"""
    # URL 정리
    repo_url = repo_url.strip()
    
    # URL 검증
    if not repo_url.startswith(("https://github.com/", "http://github.com/", "git@github.com:")):
        return """### ❌ 잘못된 Repository URL
        
GitHub URL 형식이 올바르지 않습니다.
예시: https://github.com/username/repository"""
    
    # 환경변수 대신 직접 토큰 전달
    try:
        link = deploy(repo_url, hf_token, private=private)
        return f"""### ✅ Space 생성 완료

🎉 성공적으로 배포되었습니다!

**Space URL:** [{link}]({link})

---
*배포된 Space가 제대로 작동하는지 확인해보세요.*"""
    except Exception as e:
        error_msg = str(e)
        
        # 특정 오류에 대한 상세 안내
        if "BAPI_TOKEN" in error_msg:
            return """### ❌ Brave Search API 토큰 필요

Space Settings > Variables and secrets에서 `BAPI_TOKEN`을 설정해주세요.
[Brave Search API](https://brave.com/search/api/) 에서 무료 키를 발급받을 수 있습니다."""
        elif "FRIENDLI_TOKEN" in error_msg:
            return """### ❌ Friendli AI 토큰 필요

Space Settings > Variables and secrets에서 `FRIENDLI_TOKEN`을 설정해주세요.
[Friendli AI](https://friendli.ai/) 에서 API 키를 발급받을 수 있습니다."""
        elif "Invalid header value" in error_msg:
            return """### ❌ 토큰 오류

토큰에 잘못된 문자가 포함되어 있습니다.
토큰을 다시 복사하여 공백이나 줄바꿈 없이 붙여넣어주세요."""
        elif "401" in error_msg or "403" in error_msg:
            return """### ❌ 인증 실패

토큰이 유효하지 않거나 Write 권한이 없습니다.
https://huggingface.co/settings/tokens 에서 Write 권한이 있는 토큰인지 확인해주세요."""
        else:
            return f"""### ❌ 배포 오류

오류 내용: `{error_msg}`

**가능한 원인:**
- 잘못된 Repository URL
- 권한 부족 (write 권한 필요)
- 네트워크 문제"""

# Gradio 앱 구성
with gr.Blocks(
    title="HF Space Auto-Deployer",
    theme=gr.themes.Soft(),
    css="""
    .main-container { max-width: 800px; margin: 0 auto; }
    .gr-button-primary { background-color: #2563eb !important; }
    .status-box { padding: 20px; border-radius: 8px; margin-top: 20px; }
    """
) as demo:
    gr.Markdown("""
    # 🚀 Git Repository → Hugging Face Space 자동 배포
    
    공개 Git 저장소를 Hugging Face Gradio Space로 자동 변환하고 배포합니다.
    
    ## 사용 방법
    1. **인증**: OAuth 로그인 또는 토큰 입력
    2. **Repository URL 입력**
    3. **Deploy 버튼 클릭**
    
    ---
    """)
    
    with gr.Column(elem_classes="main-container"):
        if OAUTH_ENABLED:
            # OAuth 버전
            gr.LoginButton(
                value="Sign in with Hugging Face 🤗",
                variant="primary",
                size="lg"
            )
            
            gr.Markdown("### 📝 배포 설정")
            
            repo_input = gr.Textbox(
                label="Git Repository URL",
                placeholder="https://github.com/username/repository",
                info="공개 저장소 URL을 입력하세요"
            )
            
            private_checkbox = gr.Checkbox(
                label="🔒 Private Space로 생성",
                value=False,
                info="체크하면 비공개 Space로 생성됩니다"
            )
            
            deploy_btn = gr.Button(
                "🚀 Deploy to Space",
                variant="primary",
                size="lg"
            )
            
            output_status = gr.Markdown(
                elem_classes="status-box",
                visible=False
            )
            
            # OAuth 이벤트 핸들러
            deploy_btn.click(
                fn=launch_deploy_oauth,
                inputs=[repo_input, private_checkbox],
                outputs=output_status
            ).then(
                fn=lambda: gr.update(visible=True),
                outputs=output_status
            )
        else:
            # 수동 토큰 입력 버전
            gr.Markdown("""
            ### 🔐 인증 정보
            
            > **참고**: OAuth가 설정되지 않은 Space입니다. 
            > Space 설정에서 `hf_oauth: true`를 추가하면 자동 로그인을 사용할 수 있습니다.
            """)
            
            token_input = gr.Textbox(
                label="Hugging Face Write Token",
                type="password",
                placeholder="hf_xxxxxxxxxxxxxxxxxxxxx",
                info="https://huggingface.co/settings/tokens 에서 Write 권한 토큰 생성"
            )
            
            gr.Markdown("### 📝 배포 설정")
            
            repo_input = gr.Textbox(
                label="Git Repository URL",
                placeholder="https://github.com/username/repository",
                info="공개 저장소 URL을 입력하세요"
            )
            
            private_checkbox = gr.Checkbox(
                label="🔒 Private Space로 생성",
                value=False,
                info="체크하면 비공개 Space로 생성됩니다"
            )
            
            deploy_btn = gr.Button(
                "🚀 Deploy to Space",
                variant="primary",
                size="lg"
            )
            
            output_status = gr.Markdown(
                elem_classes="status-box",
                visible=False
            )
            
            # 수동 입력 이벤트 핸들러
            deploy_btn.click(
                fn=launch_deploy_manual,
                inputs=[repo_input, token_input, private_checkbox],
                outputs=output_status
            ).then(
                fn=lambda: gr.update(visible=True),
                outputs=output_status
            )
        
        # 공통 예시 및 도움말
        gr.Markdown("""
        ### 💡 예시
        - `https://github.com/gradio-app/gradio`
        - `https://github.com/huggingface/transformers`
        
        ### ⚠️ 주의사항
        - **환경변수 필요**: `BAPI_TOKEN` (Brave Search API), `FRIENDLI_TOKEN` (Friendli AI)
        - Space 이름은 repository 이름을 기반으로 자동 생성됩니다
        - GPU가 필요한 경우 Space 생성 후 수동 설정이 필요합니다
        """)

if __name__ == "__main__":
    # 시작 시 환경 체크
    print("\n" + "="*50)
    print("🚀 HF Space Auto-Deployer 시작")
    print("="*50)
    
    # 필수 환경변수 체크
    missing_vars = []
    if not os.getenv("BAPI_TOKEN"):
        missing_vars.append("BAPI_TOKEN (Brave Search API)")
    if not os.getenv("FRIENDLI_TOKEN"):
        missing_vars.append("FRIENDLI_TOKEN (Friendli AI)")
    
    if missing_vars:
        print("\n⚠️  다음 환경변수가 설정되지 않았습니다:")
        for var in missing_vars:
            print(f"   - {var}")
        print("\nSpace Settings > Variables and secrets에서 설정해주세요.")
    else:
        print("✅ 모든 필수 환경변수가 설정되었습니다.")
    
    # OAuth 상태 확인
    if OAUTH_ENABLED:
        print("✅ OAuth가 활성화되었습니다.")
    else:
        print("ℹ️  OAuth가 비활성화되었습니다. (수동 토큰 입력 모드)")
    
    print("="*50 + "\n")
    
    demo.launch(
        ssr_mode=False,  # SSR 비활성화로 경고 제거
        show_error=True,
        quiet=True  # Gradio 기본 출력 최소화
    )