Update app.py

app.py

@@ -5,6 +5,7 @@ from slugify import slugify
 import uuid
 import os
 import re
+import subprocess
 
 # Function to extract metadata from README.md file
 def extract_from_readme(readme_path):
@@ -25,34 +26,19 @@ def extract_from_readme(readme_path):
                 port = match_port.group(1).strip()
     return title, description, port
 
+
 # Main function to clone a Git repo and create a Hugging Face repo
 def clone_and_create(profile: gr.OAuthProfile, token: gr.OAuthToken,
                      repo_url, repo_type, private, space_sdk, sdk_version,
-                     space_id, space_title, space_description):
-
-    # README.md から情報を抽出する関数
-    def extract_from_readme(readme_path):
-        import re
-        title = None
-        description = None
-        port = None
-        if os.path.exists(readme_path):
-            with open(readme_path, encoding="utf-8") as f:
-                content = f.read()
-                match_title = re.search(r'title:\s*(.*)', content)
-                match_desc = re.search(r'short_description:\s*(.*)', content)
-                match_port = re.search(r'app_port:\s*(\d+)', content)
-                if match_title:
-                    title = match_title.group(1).strip()
-                if match_desc:
-                    description = match_desc.group(1).strip()
-                if match_port:
-                    port = match_port.group(1).strip()
-        return title, description, port
+                     space_id, space_title, space_description, include_submodules):
 
     folder = str(uuid.uuid4())  # ユニークなフォルダ名作成
     Repo.clone_from(repo_url, folder)  # Gitリポジトリをクローン
 
+    # サブモジュールを含める場合
+    if include_submodules:
+        subprocess.run(["git", "submodule", "update", "--init", "--recursive"], cwd=folder, check=True)
+
     hf_api = HfApi(token=token.token)
 
     # README.md から自動情報取得
@@ -116,6 +102,7 @@ with gr.Blocks() as demo:
             repo_url = gr.Textbox(label="Git Repository URL (e.g., GitHub)")
             repo_type = gr.Dropdown(choices=["space", "dataset", "model"], value="space", label="Hugging Face repo type")
             private = gr.Checkbox(label="Make Private", value=False)
+            include_submodules = gr.Checkbox(label="Include submodules", value=False)  # 追加
             space_sdk = gr.Dropdown(choices=["gradio", "streamlit", "docker", "static"], label="Space SDK", visible=True)
             sdk_version = gr.Textbox(label="SDK Version (optional)", visible=True)
             space_id = gr.Textbox(label="Hugging Face ID (auto if blank)")
@@ -141,7 +128,7 @@ with gr.Blocks() as demo:
     btn = gr.Button("Clone and Create!")
     btn.click(
         fn=clone_and_create,
-        inputs=[repo_url, repo_type, private, space_sdk, sdk_version, space_id, space_title, space_description],
+        inputs=[repo_url, repo_type, private, space_sdk, sdk_version, space_id, space_title, space_description, include_submodules],
         outputs=output
     )