Update repo_to_space_auto.py

repo_to_space_auto.py

@@ -3,19 +3,16 @@
 repo_to_space_auto.py
 ---------------------
 공개 Git 레포를 Hugging Face Gradio Space로 자동 배포합니다.
-
 필요 환경변수
 -------------
 BAPI_TOKEN      : Brave Search API Key
 FRIENDLI_TOKEN  : Friendli Dedicated Endpoint Token
-
 CLI 인자
 --------
 --repo_url <URL>  : 원본 Git 레포 URL (필수*¹)
 --hf_token <TOK>  : 쓰기 권한 Hugging Face Access Token (필수*¹)
 --private         : 비공개 Space로 생성
 --hardware <tier> : 예) 't4-medium'  GPU 인스턴스 지정
-
 *¹ 인자를 생략하면 동명 환경변수(REPO_URL, HF_TOKEN)가 대신 사용됩니다.
 """
 
@@ -68,8 +65,23 @@ def deploy(repo_url: str, hf_token: str, private=False, hardware=None) -> str:
     api   = HfApi(token=hf_token)
     user  = api.whoami()["name"]
     space = f"{user}/{Path(repo_url).stem.lower().replace('.', '-')}"
-    api.create_repo(space, repo_type="space", space_sdk="gradio",
-                    private=private, exist_ok=True, hardware=hardware)
+    
+    # create_repo 호출 (hardware 파라미터 제거)
+    api.create_repo(
+        repo_id=space, 
+        repo_type="space", 
+        space_sdk="gradio",
+        private=private, 
+        exist_ok=True
+    )
+    
+    # hardware 설정이 필요한 경우 별도로 update_repo_settings 호출
+    if hardware:
+        try:
+            # Space 설정 업데이트 (hardware)
+            api.request_space_hardware(repo_id=space, hardware=hardware)
+        except Exception as e:
+            print(f"⚠️  Hardware 설정 실패 (Space는 생성됨): {e}")
 
     with tempfile.TemporaryDirectory() as work:
         # 1) Brave 메타데이터
@@ -83,16 +95,14 @@ def deploy(repo_url: str, hf_token: str, private=False, hardware=None) -> str:
         if (src / "README.md").exists():
             readme = (src / "README.md").read_text(encoding="utf-8", errors="ignore")[:4000]
 
-        tree_out = subprocess.run(["bash", "-lc", f"tree -L 2 {src} | head -n 40"],
+        tree_out = subprocess.run(["bash", "-lc", f"tree -L 2 {src} 2>/dev/null || find {src} -maxdepth 2 -type f | head -n 40"],
                                   text=True, capture_output=True).stdout
 
         context = textwrap.dedent(f"""
         ## Brave meta
         {json.dumps(brave_meta, ensure_ascii=False, indent=2)}
-
         ## Repository tree (depth 2)
         {tree_out}
-
         ## README (first 4 kB)
         {readme}
         """)
@@ -102,17 +112,51 @@ def deploy(repo_url: str, hf_token: str, private=False, hardware=None) -> str:
 
         # 4) Space 레포 클론 & 파일 쓰기
         dst = Path(work) / "space"
-        api.clone_repo(space, local_dir=dst)
+        space_repo = git.Repo.clone_from(
+            f"https://huggingface.co/spaces/{space}",
+            dst,
+            env={"HF_TOKEN": hf_token}
+        )
 
         (dst / "app.py").write_text(scaffold["app_py"], encoding="utf-8")
         (dst / "requirements.txt").write_text(scaffold["requirements_txt"], encoding="utf-8")
         if scaffold.get("need_docker"):
             (dst / "Dockerfile").write_text(scaffold["dockerfile"], encoding="utf-8")
-        (dst / "README.md").write_text(scaffold["summary"], encoding="utf-8")
+        
+        # README.md에 원본 레포 정보 추가
+        readme_content = f"""---
+title: {Path(repo_url).stem}
+emoji: 🚀
+colorFrom: blue
+colorTo: green
+sdk: gradio
+sdk_version: 4.44.1
+app_file: app.py
+pinned: false
+---
+
+# {Path(repo_url).stem}
+
+Automatically deployed from: {repo_url}
+
+## Summary
+{scaffold["summary"]}
+
+---
+*Created by HF Space Auto-Deployer*
+"""
+        (dst / "README.md").write_text(readme_content, encoding="utf-8")
 
-        subprocess.run(["git", "-C", dst, "add", "."], check=True)
-        subprocess.run(["git", "-C", dst, "commit", "-m", "Initial auto-deploy"], check=True)
-        subprocess.run(["git", "-C", dst, "push"], check=True)
+        # Git 커밋 및 푸시
+        space_repo.index.add(["app.py", "requirements.txt", "README.md"])
+        if scaffold.get("need_docker"):
+            space_repo.index.add(["Dockerfile"])
+        
+        space_repo.index.commit("Initial auto-deploy")
+        
+        # Push with authentication
+        origin = space_repo.remote("origin")
+        origin.push(env={"HF_TOKEN": hf_token})
 
     return f"https://huggingface.co/spaces/{space}"
 
@@ -138,4 +182,4 @@ def main():
         sys.exit(1)
 
 if __name__ == "__main__":
-    main()
+    main()