Update app.py

app.py

@@ -747,6 +747,114 @@ Deployed from: {repo_git}
             shutil.rmtree(folder)
         yield f"❌ Error: {str(e)}"
 
+def space_to_github(hf_username, hf_space_name, github_repo_name, github_token):
+    """Clone HuggingFace Space and push to GitHub"""
+    tmp_dir = None
+    try:
+        # HF Space URL 생성
+        hf_repo_url = f"https://huggingface.co/spaces/{hf_username}/{hf_space_name}.git"
+        
+        # 임시 디렉토리 생성 및 클론
+        tmp_dir = tempfile.mkdtemp()
+        yield f"📥 Cloning HF Space: {hf_username}/{hf_space_name}..."
+        
+        # Git 클론
+        env = os.environ.copy()
+        env['GIT_LFS_SKIP_SMUDGE'] = '1'  # LFS 파일 스킵
+        
+        clone_cmd = ['git', 'clone', hf_repo_url, tmp_dir]
+        result = subprocess.run(clone_cmd, env=env, capture_output=True, text=True)
+        
+        if result.returncode != 0:
+            raise Exception(f"Clone failed: {result.stderr}")
+        
+        yield "✅ HF Space cloned successfully"
+        
+        # GitHub 인증 및 레포 생성
+        yield "🔐 Authenticating with GitHub..."
+        
+        # GitHub API 호출로 레포 생성
+        headers = {
+            "Authorization": f"token {github_token}",
+            "Accept": "application/vnd.github.v3+json"
+        }
+        
+        # 사용자 정보 가져오기
+        user_response = requests.get("https://api.github.com/user", headers=headers)
+        if user_response.status_code != 200:
+            raise Exception("GitHub authentication failed")
+        
+        github_username = user_response.json()["login"]
+        yield f"✅ Authenticated as GitHub user: {github_username}"
+        
+        # 레포지토리 생성
+        yield f"📦 Creating GitHub repository: {github_repo_name}..."
+        create_data = {
+            "name": github_repo_name,
+            "private": False,
+            "auto_init": False
+        }
+        
+        create_response = requests.post(
+            "https://api.github.com/user/repos",
+            headers=headers,
+            json=create_data
+        )
+        
+        if create_response.status_code == 201:
+            repo_info = create_response.json()
+            yield f"✅ GitHub repository created: {repo_info['html_url']}"
+        elif create_response.status_code == 422:
+            # 레포가 이미 존재함
+            repo_info = {"html_url": f"https://github.com/{github_username}/{github_repo_name}"}
+            yield f"ℹ️ Repository already exists: {repo_info['html_url']}"
+        else:
+            raise Exception(f"Failed to create repository: {create_response.json().get('message', 'Unknown error')}")
+        
+        # Git remote 변경 및 푸시
+        yield "📤 Pushing to GitHub..."
+        
+        # git 명령어로 remote 변경 및 푸시
+        os.chdir(tmp_dir)
+        
+        # 기존 origin 제거
+        subprocess.run(['git', 'remote', 'remove', 'origin'], capture_output=True)
+        
+        # GitHub remote 추가
+        github_remote_url = f"https://{github_username}:{github_token}@github.com/{github_username}/{github_repo_name}.git"
+        subprocess.run(['git', 'remote', 'add', 'origin', github_remote_url], check=True)
+        
+        # main 브랜치로 푸시
+        push_result = subprocess.run(
+            ['git', 'push', '-u', 'origin', 'HEAD:main'],
+            capture_output=True,
+            text=True
+        )
+        
+        if push_result.returncode != 0:
+            # master 브랜치로 재시도
+            push_result = subprocess.run(
+                ['git', 'push', '-u', 'origin', 'HEAD:master'],
+                capture_output=True,
+                text=True
+            )
+            
+            if push_result.returncode != 0:
+                raise Exception(f"Push failed: {push_result.stderr}")
+        
+        yield "✅ Successfully pushed to GitHub!"
+        yield f"\n🔗 GitHub Repository: https://github.com/{github_username}/{github_repo_name}"
+        yield f"\n✨ Clone command: `git clone https://github.com/{github_username}/{github_repo_name}.git`"
+        
+    except Exception as e:
+        yield f"❌ Error: {str(e)}"
+    finally:
+        # 임시 디렉토리 정리
+        if tmp_dir and os.path.exists(tmp_dir):
+            os.chdir("/")  # 디렉토리 밖으로 이동
+            shutil.rmtree(tmp_dir)
+            yield "🧹 Cleaned up temporary files"
+
 css = """
 .container {
     max-width: 900px;
@@ -761,57 +869,127 @@ css = """
     font-size: 14px;
     line-height: 1.5;
 }
+.tab-content {
+    margin-top: 20px;
+}
 """
 
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("# 🚀 GitHub to Hugging Face Space Cloner")
+with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🔄 GitHub ↔ HuggingFace Space Converter")
     
-    if not os.getenv("HF_TOKEN"):
-        gr.Markdown("❌ HF_TOKEN Required - Set it in Space settings")
-    else:
-        gr.Markdown("✅ HF_TOKEN Found")
-    
-    with gr.Row():
-        with gr.Column():
-            repo_git = gr.Textbox(
-                label="GitHub Repository URL",
-                placeholder="https://github.com/username/repository"
-            )
-            repo_hf = gr.Textbox(
-                label="Hugging Face Space Name",
-                placeholder="my-awesome-space"
-            )
-            sdk_choices = gr.Radio(
-                ["gradio", "streamlit", "docker", "static"],
-                label="Space SDK",
-                value="gradio"
-            )
-            skip_lfs = gr.Checkbox(
-                label="Skip Git LFS files",
-                value=True
-            )
-            enable_smart_generation = gr.Checkbox(
-                label="🤖 Enable Smart app.py Generation",
-                value=False,
-                info="Requires FIREWORKS_API_KEY in environment variables"
+    with gr.Tabs():
+        with gr.Tab("GitHub → HF Space"):
+            if not os.getenv("HF_TOKEN"):
+                gr.Markdown("❌ HF_TOKEN Required - Set it in Space settings")
+            else:
+                gr.Markdown("✅ HF_TOKEN Found")
+            
+            with gr.Row():
+                with gr.Column():
+                    repo_git = gr.Textbox(
+                        label="GitHub Repository URL",
+                        placeholder="https://github.com/username/repository"
+                    )
+                    repo_hf = gr.Textbox(
+                        label="Hugging Face Space Name",
+                        placeholder="my-awesome-space"
+                    )
+                    sdk_choices = gr.Radio(
+                        ["gradio", "streamlit", "docker", "static"],
+                        label="Space SDK",
+                        value="gradio"
+                    )
+                    skip_lfs = gr.Checkbox(
+                        label="Skip Git LFS files",
+                        value=True
+                    )
+                    enable_smart_generation = gr.Checkbox(
+                        label="🤖 Enable Smart app.py Generation",
+                        value=False,
+                        info="Requires FIREWORKS_API_KEY in environment variables"
+                    )
+                    
+                    btn_to_hf = gr.Button("🚀 Deploy to HF Space", variant="primary")
+                    
+                with gr.Column():
+                    output_to_hf = gr.Textbox(
+                        label="Progress",
+                        lines=15,
+                        elem_classes=["output-box"],
+                        interactive=False,
+                        show_copy_button=True
+                    )
+            
+            btn_to_hf.click(
+                fn=clone,
+                inputs=[repo_git, repo_hf, sdk_choices, skip_lfs, enable_smart_generation],
+                outputs=output_to_hf
             )
+        
+        with gr.Tab("HF Space → GitHub"):
+            gr.Markdown("""
+            📌 **Export your HuggingFace Space to GitHub**
             
-            btn = gr.Button("🎯 Clone Repository", variant="primary")
+            Create a GitHub repository from your existing HuggingFace Space.
+            """)
             
-        with gr.Column():
-            output = gr.Textbox(
-                label="Progress",
-                lines=15,
-                elem_classes=["output-box"],
-                interactive=False,
-                show_copy_button=True
+            with gr.Row():
+                with gr.Column():
+                    hf_username = gr.Textbox(
+                        label="HuggingFace Username",
+                        placeholder="e.g., john_doe"
+                    )
+                    hf_space_name = gr.Textbox(
+                        label="HuggingFace Space Name",
+                        placeholder="e.g., my-space"
+                    )
+                    github_repo_name = gr.Textbox(
+                        label="GitHub Repository Name",
+                        placeholder="e.g., my-new-repo"
+                    )
+                    github_token = gr.Textbox(
+                        label="GitHub Personal Access Token",
+                        type="password",
+                        placeholder="ghp_xxxxxxxxxxxx",
+                        info="Token needs 'repo' scope. Get it from: https://github.com/settings/tokens"
+                    )
+                    
+                    btn_to_github = gr.Button("📤 Export to GitHub", variant="primary")
+                    
+                with gr.Column():
+                    output_to_github = gr.Textbox(
+                        label="Progress",
+                        lines=15,
+                        elem_classes=["output-box"],
+                        interactive=False,
+                        show_copy_button=True
+                    )
+            
+            btn_to_github.click(
+                fn=space_to_github,
+                inputs=[hf_username, hf_space_name, github_repo_name, github_token],
+                outputs=output_to_github
             )
     
-    btn.click(
-        fn=clone,
-        inputs=[repo_git, repo_hf, sdk_choices, skip_lfs, enable_smart_generation],
-        outputs=output
-    )
+    gr.Markdown("""
+    ---
+    ### 📖 Instructions
+    
+    **GitHub → HF Space:**
+    - Clone any public GitHub repository to your HuggingFace account
+    - Automatically handles LFS issues
+    - Optional AI-powered app.py generation
+    
+    **HF Space → GitHub:**
+    - Export your HuggingFace Space to a new GitHub repository
+    - Requires GitHub Personal Access Token with 'repo' scope
+    - Preserves all files and structure
+    
+    ### 🔑 Required Tokens
+    - **HF_TOKEN**: Set in Space settings for GitHub → HF
+    - **GitHub Token**: Required for HF → GitHub (create at [GitHub Settings](https://github.com/settings/tokens))
+    - **FIREWORKS_API_KEY**: Optional, for smart app generation
+    """)
 
 if __name__ == "__main__":
     demo.launch()