Update app.py

app.py

@@ -2,91 +2,102 @@ import gradio as gr
 import os
 import shutil
 import tempfile
-from huggingface_hub import HfApi, create_repo, create_repo_readme
+from huggingface_hub import HfApi
 from git import Repo
 
 api = HfApi()
 
-def clone_and_create(hf_token, github_url, repo_type, private, space_id, port, title, description):
-    if not github_url.strip():
-        return "GitHub URLを入力してください。"
-    
+def clone_and_create(hf_token, git_url, repo_type, private, space_id, port, title, description):
+    if not git_url.strip():
+        return "Git リポジトリのURLを入力してください。"
+
     temp_dir = tempfile.mkdtemp()
-    
+
     try:
-        # GitHubからクローン
-        Repo.clone_from(github_url, temp_dir)
+        # リポジトリをクローン（github 以外も可）
+        Repo.clone_from(git_url, temp_dir)
 
-        # 自動生成ID
+        # 空欄ならIDをリポジトリ名から自動生成
         if not space_id.strip():
-            space_id = github_url.strip().rstrip('/').split('/')[-1]
+            space_id = git_url.rstrip("/").split("/")[-1].replace(".git", "")
+
+        user_info = api.whoami(token=hf_token)
+        repo_id = f"{user_info['name']}/{space_id}"
+
+        # Hugging Face 上でリポジトリ作成
+        api.create_repo(
+            repo_id=repo_id,
+            repo_type=repo_type,
+            private=private,
+            token=hf_token,
+            exist_ok=False
+        )
 
-        # README.mdの処理（Space用）
-        readme_path = os.path.join(temp_dir, "README.md")
+        # README.md の作成（Space のときだけ）
         if repo_type == "space":
-            lines = []
+            readme_path = os.path.join(temp_dir, "README.md")
+            readme_lines = []
+
             if title.strip():
-                lines.append(f"# {title.strip()}")
+                readme_lines.append(f"# {title.strip()}")
             if description.strip():
-                lines.append(f"\n{description.strip()}")
+                readme_lines.append(f"\n{description.strip()}")
             if port.strip():
-                lines.append(f"\napp_port: {port.strip()}")
-            readme_text = "\n".join(lines)
-            with open(readme_path, "w", encoding="utf-8") as f:
-                f.write(readme_text)
-
-        # huggingface に作成
-        repo_url = create_repo(
-            name=space_id,
-            token=hf_token,
+                readme_lines.append(f"\napp_port: {port.strip()}")
+
+            if readme_lines:
+                with open(readme_path, "w", encoding="utf-8") as f:
+                    f.write("\n".join(readme_lines))
+
+        # クローンしたファイルをアップロード
+        api.upload_folder(
+            folder_path=temp_dir,
+            repo_id=repo_id,
             repo_type=repo_type,
-            private=private,
-            exist_ok=False
+            token=hf_token
         )
 
-        # リポジトリをpush
-        local_repo = Repo(temp_dir)
-        local_repo.create_remote("hf", repo_url)
-        local_repo.git.push("hf", "HEAD:main")
-
-        return f"{repo_type} を作成しました：\n{repo_url}"
+        return f"✅ {repo_type} を作成しました！\nhttps://huggingface.co/{repo_type}s/{space_id}"
 
     except Exception as e:
-        return f"エラー: {str(e)}"
+        return f"❌ エラー: {str(e)}"
+
     finally:
         shutil.rmtree(temp_dir)
 
-with gr.Blocks(title="Hugging Face リポジトリ作成") as demo:
-    gr.Markdown("## Hugging Face に新しい Space / Dataset / Model を作成")
 
-    with gr.Row():
-        login = gr.LoginButton()
+with gr.Blocks(title="Hugging Face リポジトリ作成ツール") as demo:
+    gr.Markdown("## 🛠️ Hugging Face Space / Dataset / Model 作成ツール")
+
+    login = gr.LoginButton()
 
     with gr.Column():
-        token = gr.Textbox(label="Hugging Face Token（ログイン後自動取得）", type="password")
-        github_url = gr.Textbox(label="GitHub リポジトリURL", placeholder="https://github.com/your/repo")
+        token = gr.Textbox(label="Hugging Face トークン", type="password")
+        git_url = gr.Textbox(label="Git リポジトリのURL（GitHub / GitLab など）", placeholder="https://...")
         repo_type = gr.Radio(label="作成タイプ", choices=["space", "dataset", "model"], value="space")
         private = gr.Checkbox(label="非公開にする", value=False)
 
-        with gr.Group(visible=False) as space_options:
-            space_id = gr.Textbox(label="Space ID（空欄で自動生成）", placeholder="例: my-awesome-space")
-            port = gr.Textbox(label="起動ポート（README.mdに追加）", placeholder="例: 7860")
-            title = gr.Textbox(label="スペース名（READMEのタイトル）", placeholder="例: My Space")
-            description = gr.Textbox(label="説明（READMEのshort_description）", placeholder="例: これは○○なスペースです")
+        with gr.Group(visible=True) as space_options:
+            space_id = gr.Textbox(label="Space ID（空欄で自動生成）", placeholder="例: my-space")
+            port = gr.Textbox(label="起動ポート（READMEに記載）", placeholder="例: 7860")
+            title = gr.Textbox(label="スペース名（READMEタイトル）", placeholder="例: My App")
+            description = gr.Textbox(label="説明文（READMEに記載）", placeholder="例: これは〇〇なスペースです")
 
-        repo_type.change(lambda x: gr.update(visible=(x == "space")), inputs=repo_type, outputs=space_options)
+        repo_type.change(lambda t: gr.update(visible=(t == "space")), inputs=repo_type, outputs=space_options)
 
         submit_btn = gr.Button("作成")
-        output = gr.Textbox(label="出力メッセージ", lines=5)
+        output = gr.Textbox(label="出力", lines=5)
 
-    def auto_token(login_data):
+    # ログイン成功時にトークンを自動取得
+    def on_login(login_data):
         return login_data["access_token"]
 
-    login.login(auto_token, outputs=token)
+    login.login(on_login, outputs=token)
 
+    # 実行
     submit_btn.click(
-        clone_and_create,
-        inputs=[token, github_url, repo_type, private, space_id, port, title, description],
+        fn=clone_and_create,
+        inputs=[token, git_url, repo_type, private, space_id, port, title, description],
         outputs=output
     )