Update app.py

app.py

@@ -7,16 +7,30 @@ from pathlib import Path
 sys.path.insert(0, str(Path(__file__).parent))
 from repo_to_space_auto import deploy
 
-def launch_deploy(repo_url, private, request: gr.Request):
-    """HF OAuth 토큰으로 Space 배포"""
-    # OAuth로 받은 토큰 확인
-    hf_token = request.headers.get("x-oauth-identity")
-    
-    if not hf_token:
+# 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
+    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 토큰을 입력해주세요."""
+    
+    return _deploy_space(repo_url, hf_token, private)
+
+def _deploy_space(repo_url, hf_token, private):
+    """공통 배포 로직"""
     # 기존 환경변수 백업 및 임시 설정
     old_token = os.environ.get("HF_TOKEN")
     os.environ["HF_TOKEN"] = hf_token
@@ -64,47 +78,108 @@ with gr.Blocks(
     공개 Git 저장소를 Hugging Face Gradio Space로 자동 변환하고 배포합니다.
     
     ## 사용 방법
-    1. **Hugging Face 로그인** (상단 우측 버튼)
-    2. Repository URL 입력
-    3. Deploy 버튼 클릭
+    1. **인증**: OAuth 로그인 또는 토큰 입력
+    2. **Repository URL 입력**
+    3. **Deploy 버튼 클릭**
     
     ---
     """)
     
     with gr.Column(elem_classes="main-container"):
-        # 로그인 상태 표시
-        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로 생성됩니다"
-        )
+        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
+            )
         
-        deploy_btn = gr.Button(
-            "🚀 Deploy to Space",
-            variant="primary",
-            size="lg"
-        )
-        
-        output_status = gr.Markdown(
-            elem_classes="status-box",
-            visible=False
-        )
-        
-        # 예시 및 도움말
+        # 공통 예시 및 도움말
         gr.Markdown("""
         ### 💡 예시
         - `https://github.com/gradio-app/gradio`
@@ -115,16 +190,6 @@ with gr.Blocks(
         - Space 이름은 repository 이름을 기반으로 자동 생성됩니다
         - GPU가 필요한 경우 Space 생성 후 수동 설정이 필요합니다
         """)
-    
-    # 이벤트 핸들러
-    deploy_btn.click(
-        fn=launch_deploy,
-        inputs=[repo_input, private_checkbox],
-        outputs=output_status
-    ).then(
-        fn=lambda: gr.update(visible=True),
-        outputs=output_status
-    )
 
 if __name__ == "__main__":
     demo.launch(