Update app.py

app.py

@@ -432,30 +432,38 @@ if __name__ == "__main__":
         "summary": f"Smart template created for {repo_name}"
     }
 
-def clone(repo_git, repo_hf, sdk_type, skip_lfs, enable_smart_generation, hf_token=None):
+def clone(repo_git, repo_hf, sdk_type, skip_lfs, enable_smart_generation, profile: gr.OAuthProfile = None, oauth_token: gr.OAuthToken = None):
     folder = str(uuid.uuid4())
     
-    # HF Token 처리 - 환경변수보다 입력값 우선
-    if not hf_token:
-        hf_token = os.getenv("HF_TOKEN")
-    
-    if not hf_token:
-        yield "❌ Error: HuggingFace token not provided. Please enter your token or use OAuth login."
+    # OAuth 인증 우선, 없으면 환경변수 사용
+    if oauth_token:
+        hf_token = oauth_token.token
+        username = profile.username if profile else None
+        yield f"✅ Authenticated via OAuth as: @{username}"
+    elif profile:
+        # Profile은 있지만 token이 없는 경우 (권한 부족)
+        yield "❌ Error: Insufficient permissions. Please logout and login again with write access."
         return
+    else:
+        # OAuth 없을 때 환경변수 체크
+        hf_token = os.getenv("HF_TOKEN")
+        if not hf_token:
+            yield "❌ Error: Please login with HuggingFace using the 'Sign in with Hugging Face' button at the top."
+            return
+        username = None
     
     try:
         yield "🔄 Starting clone process..."
         
-        # HuggingFace 로그인 및 인증
+        # HuggingFace API 인증
         try:
-            login(token=hf_token)
             api = HfApi(token=hf_token)
-            user_info = api.whoami()
-            username = user_info["name"]
-            yield f"✅ Authenticated as HuggingFace user: @{username}"
+            if not username:  # OAuth로 로그인하지 않은 경우 username 가져오기
+                user_info = api.whoami()
+                username = user_info["name"]
+                yield f"✅ Authenticated as HuggingFace user: @{username}"
         except Exception as e:
             yield f"❌ HuggingFace authentication failed: {str(e)}"
-            yield "Please check your token or try OAuth login"
             return
         
         # URL 정규화 및 검증
@@ -1184,60 +1192,36 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(
     </div>
     """)
     
-    with gr.Tabs(elem_classes="tabs"):
+with gr.Blocks(css=css, theme=gr.themes.Soft(
+    primary_hue="purple",
+    secondary_hue="purple",
+    neutral_hue="slate"
+)) as demo:
+    gr.HTML("""
+    <div class="main-header">
+        <h1 style="font-size: 2.5rem; margin: 0; font-weight: 700;">
+            🔄 Repository Converter Pro
+        </h1>
+        <p style="font-size: 1.1rem; margin-top: 10px; opacity: 0.95;">
+            Seamlessly transfer repositories between GitHub and HuggingFace
+        </p>
+    </div>
+    """)
+    
+    # OAuth 로그인 버튼을 상단에 배치
+    with gr.Row():
+        with gr.Column(scale=4):
+            gr.LoginButton()
+        with gr.Column(scale=8):
+            login_status = gr.HTML("""
+            <div style="padding: 10px; background: #fef2f2; border: 1px solid #fecaca; 
+                       border-radius: 8px; margin: 5px 0;">
+                <span style="color: #dc2626;">⚠️ Please login with HuggingFace to continue</span>
+            </div>
+            """)
+    
+    with gr.Tabs(elem_classes="tabs") as main_tabs:
         with gr.Tab("🚀 GitHub → HF Space", elem_id="tab1"):
-            # OAuth 로그인 섹션
-            with gr.Group(elem_classes="input-section", visible=True) as auth_section:
-                gr.HTML("""
-                <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
-                            padding: 20px; border-radius: 10px; color: white; margin-bottom: 20px;">
-                    <h3 style="margin: 0 0 10px 0;">🔐 HuggingFace Authentication Required</h3>
-                    <p style="margin: 5px 0;">Choose one of the following methods to authenticate:</p>
-                </div>
-                """)
-                
-                with gr.Row():
-                    with gr.Column():
-                        gr.Markdown("""
-                        ### Option 1: OAuth Login (Recommended)
-                        Click the button below to login with your HuggingFace account
-                        """)
-                        gr.HTML("""
-                        <a href="https://huggingface.co/login?next=%2Fsettings%2Ftokens" target="_blank">
-                            <button style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-                                         color: white; border: none; padding: 12px 24px;
-                                         border-radius: 8px; font-size: 16px; font-weight: 600;
-                                         cursor: pointer; width: 100%; margin: 10px 0;">
-                                🤗 Login with HuggingFace
-                            </button>
-                        </a>
-                        """)
-                        gr.Markdown("""
-                        After login:
-                        1. Go to [Settings → Access Tokens](https://huggingface.co/settings/tokens)
-                        2. Create a new token with **write** access
-                        3. Copy and paste it below
-                        """)
-                    
-                    with gr.Column():
-                        gr.Markdown("""
-                        ### Option 2: Direct Token Input
-                        If you already have a token, enter it here
-                        """)
-                        hf_token_input = gr.Textbox(
-                            label="HuggingFace Token",
-                            type="password",
-                            placeholder="hf_xxxxxxxxxxxxxxxxxxxxx",
-                            info="Your HF token with write access"
-                        )
-                        gr.Markdown("""
-                        **Need a token?**
-                        1. Visit [HuggingFace Tokens](https://huggingface.co/settings/tokens)
-                        2. Click "New token"
-                        3. Give it a name and select "write" role
-                        4. Copy the token here
-                        """)
-            
             with gr.Row():
                 with gr.Column(scale=5):
                     with gr.Group(elem_classes="input-section"):
@@ -1272,7 +1256,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(
                             enable_smart_generation = gr.Checkbox(
                                 label="🤖 AI-Powered App Generation",
                                 value=False,
-                                info="Generate app.py automatically"
+                                info="Generate app.py automatically (requires FIREWORKS_API_KEY)"
                             )
                         
                         btn_to_hf = gr.Button("🚀 Deploy to HuggingFace", 
@@ -1291,40 +1275,9 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(
                             show_copy_button=True
                         )
             
-            # Token 상태 표시
-            with gr.Row():
-                token_status = gr.HTML("""
-                <div id="token-status" style="background: #fef2f2; border: 1px solid #fecaca; 
-                                              border-radius: 8px; padding: 12px; margin-top: 10px;">
-                    <span style="color: #dc2626; font-weight: 600;">⚠️ Status:</span>
-                    <span style="color: #7f1d1d;"> No token provided. Please login or enter your token above.</span>
-                </div>
-                """)
-            
-            # Token 입력시 상태 업데이트
-            def update_token_status(token):
-                if token and token.strip():
-                    return """
-                    <div style="background: #f0fdf4; border: 1px solid #bbf7d0; 
-                               border-radius: 8px; padding: 12px; margin-top: 10px;">
-                        <span style="color: #16a34a; font-weight: 600;">✅ Status:</span>
-                        <span style="color: #14532d;"> Token provided. Ready to deploy!</span>
-                    </div>
-                    """
-                else:
-                    return """
-                    <div style="background: #fef2f2; border: 1px solid #fecaca; 
-                               border-radius: 8px; padding: 12px; margin-top: 10px;">
-                        <span style="color: #dc2626; font-weight: 600;">⚠️ Status:</span>
-                        <span style="color: #7f1d1d;"> No token provided. Please login or enter your token above.</span>
-                    </div>
-                    """
-            
-            hf_token_input.change(update_token_status, inputs=[hf_token_input], outputs=[token_status])
-            
             btn_to_hf.click(
                 fn=clone,
-                inputs=[repo_git, repo_hf, sdk_choices, skip_lfs, enable_smart_generation, hf_token_input],
+                inputs=[repo_git, repo_hf, sdk_choices, skip_lfs, enable_smart_generation],
                 outputs=output_to_hf
             )