Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
 from huggingface_hub import HfApi, create_repo
 import os
-from github import Github
+from git import Repo
+import git
+from urllib.parse import urlparse
 
-# 翻訳辞書（簡単な日本語と英語の例）
+# 翻訳辞書（日本語と英語）
 translations = {
     "en": {
         "title": "Hugging Face Space Creation Form",
@@ -37,123 +39,132 @@ translations = {
     }
 }
 
-# Hugging Faceスペースの作成
+# Hugging Face API クライアント
+hf_api = HfApi()
+
+# GitHub リポジトリ情報を取得
+def get_github_repo_info(github_repo_url, github_token=None):
+    try:
+        parsed_url = urlparse(github_repo_url)
+        repo_path = parsed_url.path.strip('/')
+        
+        # 一時ディレクトリにクローンして情報を取得
+        temp_dir = "temp_repo_clone"
+        if os.path.exists(temp_dir):
+            os.system(f"rm -rf {temp_dir}")
+        
+        if github_token:
+            # 認証付きURLでクローン
+            auth_url = f"https://{github_token}@github.com/{repo_path}.git"
+            repo = Repo.clone_from(auth_url, temp_dir)
+        else:
+            repo = Repo.clone_from(github_repo_url, temp_dir)
+        
+        repo_name = os.path.basename(repo_path)
+        description = "GitHub リポジトリの説明"  # 必要に応じて `repo.description` を使用
+        
+        # 一時ディレクトリを削除
+        os.system(f"rm -rf {temp_dir}")
+        
+        return type('obj', (object,), {
+            'name': repo_name,
+            'description': description
+        })
+    except git.exc.GitCommandError as e:
+        raise Exception(f"GitHub リポジトリの取得に失敗: {str(e)}")
+
+# Hugging Face スペースを作成
 def create_huggingface_space(space_name, github_repo_url, github_token, visibility, sdk, description, license_type, port, name, emoji, pinned):
     try:
-        # GitHubリポジトリの情報取得
+        # GitHub リポジトリ情報を取得
         repo = get_github_repo_info(github_repo_url, github_token)
         
-        # Hugging Faceスペースを作成
-        create_repo(space_name, private=(visibility == 'Private'))
-
-        # README.mdの作成
-        readme_content = f"# {repo.name}\n\n{description}\n\n## GitHub Repository\n{github_repo_url}\n"
-        readme_content += f"\n**License**: {license_type}\n**Port**: {port}\n**Name**: {name}\n**Emoji**: {emoji}"
-
-        # README.mdファイルを作成しアップロード
+        # Hugging Face スペースを作成
+        create_repo(
+            repo_id=space_name,
+            repo_type="space",
+            private=(visibility == 'Private'),
+            space_sdk=sdk.lower()  # "gradio", "streamlit", etc.
+        )
+        
+        # README.md の内容を生成
+        readme_content = f"""# {repo.name}\n\n{description}\n\n## GitHub Repository\n{github_repo_url}\n\n**License**: {license_type}\n**Port**: {port}\n"""
+        
+        # README.md をアップロード
         hf_api.upload_file(
-            path_or_fileobj=readme_content,
+            path_or_fileobj=readme_content.encode(),
             path_in_repo="README.md",
-            repo_id=space_name
+            repo_id=space_name,
+            repo_type="space"
         )
         
-        # スペースの公開/非公開設定
-        hf_api.set_repo_visibility(space_name, private=(visibility == 'Private'))
-
-        # スペース設定の詳細（SDKやその他の設定など）
-        hf_api.set_space_sdk(space_name, sdk)
-        hf_api.set_space_emoji(space_name, emoji)
-        hf_api.set_space_pinned(space_name, pinned)
-
-        return f"スペース {space_name} が作成されました！"
+        # スペース設定を更新
+        hf_api.set_space_sdk(space_name, sdk.lower())
+        if emoji:
+            hf_api.set_space_emoji(space_name, emoji)
+        if pinned:
+            hf_api.set_space_pinned(space_name, pinned)
+        
+        return f"✅ スペース '{space_name}' が正常に作成されました！"
     except Exception as e:
-        return f"エラーが発生しました: {str(e)}"
-# Gradio UIの作成
+        return f"❌ エラーが発生しました: {str(e)}"
+
+# Gradio UI の構築
 def gradio_ui():
     with gr.Blocks() as demo:
-        # JavaScriptでブラウザ言語を取得し、UIを動的に翻訳
+        # ブラウザの言語に基づいてUIを翻訳
         js_code = """
         <script>
-        // ブラウザの言語を取得
         const userLang = navigator.language || navigator.userLanguage;
-        let lang = userLang.split('-')[0];  // ja or en
-
-        // Gradioに必要な翻訳データを設定
-        const translations = {
-            "en": {
-                "title": "Hugging Face Space Creation Form",
-                "space_name": "Space Name",
-                "repo_url": "GitHub Repository URL",
-                "github_token": "GitHub Token (Only for private repos)",
-                "visibility": "Visibility (Public or Private)",
-                "sdk": "SDK",
-                "description": "Description",
-                "license": "License",
-                "port": "Port",
-                "name": "Name",
-                "emoji": "Emoji",
-                "pinned": "Pin this space",
-                "submit": "Create Space"
-            },
-            "ja": {
-                "title": "Hugging Face スペース作成フォーム",
-                "space_name": "新規作成するスペース名",
-                "repo_url": "GitHubリポジトリのURL",
-                "github_token": "GitHubトークン（プライベートリポジトリの場合のみ）",
-                "visibility": "公開/非公開",
-                "sdk": "SDK",
-                "description": "説明",
-                "license": "ライセンス",
-                "port": "ポート",
-                "name": "名前",
-                "emoji": "絵文字",
-                "pinned": "固定するかどうか",
-                "submit": "スペースを作成"
-            }
-        };
-
-        // 言語に基づいてUIのテキストを変更
-        const translateUI = (lang) => {
-            document.querySelectorAll('[data-translate]').forEach((element) => {
-                const key = element.getAttribute('data-translate');
-                element.innerText = translations[lang][key] || translations["en"][key];
+        const lang = userLang.startsWith('ja') ? 'ja' : 'en';
+        const translations = """ + str(translations) + """;
+        
+        function translateUI() {
+            document.querySelectorAll('[data-translate]').forEach(el => {
+                const key = el.getAttribute('data-translate');
+                if (translations[lang] && translations[lang][key]) {
+                    el.textContent = translations[lang][key];
+                }
             });
-        };
-
-        // 言語をセットしUIを更新
-        translateUI(lang);
+        }
+        translateUI();
         </script>
         """
-        
-        gr.HTML(js_code)  # JavaScriptコードを埋め込む
+        gr.HTML(js_code)
 
-        # UIコンポーネント
+        # UI コンポーネント
+        gr.Markdown("## <span data-translate='title'>Hugging Face スペース作成フォーム</span>")
+        
         with gr.Row():
-            gr.Markdown("<h3 data-translate='title'>Hugging Face スペース作成フォーム</h3>")
-
+            space_name = gr.Textbox(label="Space Name", data-translate="space_name")
+            github_repo_url = gr.Textbox(label="GitHub Repository URL", data-translate="repo_url")
+        
         with gr.Row():
-            space_name = gr.Textbox(label="新規作成するスペース名", placeholder="スペース名", elem_id="space_name", data_translate="space_name")
-            github_repo_url = gr.Textbox(label="GitHubリポジトリのURL", placeholder="GitHubリポジトリURL", elem_id="repo_url", data_translate="repo_url")
-            github_token = gr.Textbox(label="GitHubトークン", placeholder="GitHubトークン（プライベートリポジトリの場合のみ）", type="password", elem_id="github_token", data_translate="github_token")
-            visibility = gr.Dropdown(label="公開/非公開", choices=["Public", "Private"], value="Public", elem_id="visibility", data_translate="visibility")
-            sdk = gr.Dropdown(label="SDK", choices=["Gradio", "Static", "Streamlit", "Docker"], value="Gradio", elem_id="sdk", data_translate="sdk")
-            description = gr.Textbox(label="説明", placeholder="スペースの説明", elem_id="description", data_translate="description")
-            license_type = gr.Textbox(label="ライセンス", placeholder="ライセンス", elem_id="license", data_translate="license")
-            port = gr.Number(label="ポート", value=7860, elem_id="port", data_translate="port")
-            name = gr.Textbox(label="名前", placeholder="名前", elem_id="name", data_translate="name")
-            emoji = gr.Textbox(label="絵文字", placeholder="絵文字", elem_id="emoji", data_translate="emoji")
-            pinned = gr.Checkbox(label="固定するかどうか", value=False, elem_id="pinned", data_translate="pinned")
-
-        submit_btn = gr.Button("スペースを作成", elem_id="submit", data_translate="submit")
-
+            github_token = gr.Textbox(label="GitHub Token", type="password", data-translate="github_token")
+            visibility = gr.Dropdown(choices=["Public", "Private"], value="Public", data-translate="visibility")
+            sdk = gr.Dropdown(choices=["Gradio", "Streamlit", "Static", "Docker"], value="Gradio", data-translate="sdk")
+        
+        description = gr.Textbox(label="Description", lines=3, data-translate="description")
+        license_type = gr.Textbox(label="License", value="MIT", data-translate="license")
+        
+        with gr.Row():
+            port = gr.Number(label="Port", value=7860, data-translate="port")
+            name = gr.Textbox(label="Name", data-translate="name")
+            emoji = gr.Textbox(label="Emoji", placeholder="🚀", data-translate="emoji")
+            pinned = gr.Checkbox(label="Pin this space", data-translate="pinned")
+        
+        submit_btn = gr.Button("Create Space", data-translate="submit")
+        output = gr.Textbox(label="Result")
+        
         submit_btn.click(
             create_huggingface_space,
             inputs=[space_name, github_repo_url, github_token, visibility, sdk, description, license_type, port, name, emoji, pinned],
-            outputs="text"
+            outputs=output
         )
-
+    
     return demo
 
-# Gradio UIの起動
+# アプリ起動
 if __name__ == "__main__":
-    gradio_ui().launch()
+    app = gradio_ui()
+    app.launch()