Upload 6 files

README.md

@@ -1,10 +1,10 @@
 ---
-title: Gradio Demo Space creation helper
+title: Gradio Demo Space creation helper V2
 emoji: 🐶
 colorFrom: yellow
 colorTo: red
 sdk: gradio
-sdk_version: 4.41.0
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -3,7 +3,7 @@ from t2i_space import get_t2i_space_contents
 
 css = """"""
 
-with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
+with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css) as demo:
     gr.Markdown("# Gradio Demo Space creation helper")
     gr.Markdown(
         f"""
@@ -18,17 +18,26 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
     )
     with gr.Column():
         repo_id = gr.Textbox(label="Model repo ID", placeholder="username/modelname", value="", max_lines=1)
-        with gr.Row():
-            gradio_version = gr.Textbox(label="Gradio version", placeholder="username/modelname", value="4.44.0", max_lines=1)
-            private_ok = gr.Checkbox(label="Allow private repo", value=True)
+        with gr.Accordion("Advanced", open=False):
+            with gr.Row():
+                gradio_version = gr.Textbox(label="Gradio version", placeholder="username/modelname", value="4.44.0", max_lines=1)
+                private_ok = gr.Checkbox(label="Allow private repo", value=True)
+            with gr.Row():
+                hf_user = gr.Textbox(label="Your HF user ID (Optional)", placeholder="username", value="", max_lines=1)
+                hf_repo = gr.Textbox(label="New space name (Optional)", placeholder="spacename", info="If empty, auto-complete", value="", max_lines=1)
+            hf_token = gr.Textbox(label="Your HF write token (Optional)", placeholder="hf_...", value="", max_lines=1)
+            with gr.Row():
+                is_private = gr.Checkbox(label="Create private repo (Optional)", value=True)
+                is_setkey = gr.Checkbox(label="Set your token to space (Optional)", info="For private / gated models", value=False)
         run_button = gr.Button(value="Submit")
         space_file = gr.Files(label="Output", interactive=False)
+        output_md = gr.Markdown(label="Output")
 
     gr.on(
         triggers=[repo_id.submit, run_button.click],
         fn=get_t2i_space_contents,
-        inputs=[repo_id, gradio_version, private_ok],
-        outputs=[space_file],
+        inputs=[repo_id, gradio_version, private_ok, hf_user, hf_repo, is_private, is_setkey, hf_token],
+        outputs=[space_file, output_md],
     )
 
 demo.queue()

t2i_space.py

@@ -1,77 +1,88 @@
-from pathlib import Path
-import os
-import gradio as gr
-from huggingface_hub import HfApi, HfFolder, CommitOperationAdd, create_repo, Repository
-
-def is_repo_name(s):
-    import re
-    return re.fullmatch(r'^[^/,\s]+?/[^/,\s]+?$', s)
-
-def is_repo_exists(repo_id):
-    api = HfApi()
-    try:
-        return api.repo_exists(repo_id=repo_id, repo_type="space")
-    except Exception as e:
-        print(f"Error: Failed to connect {repo_id}.")
-        return True  # for safe handling
-
-def save_space_contents(repo_id: str, gradio_version: str, private_ok: bool, dir: str):
-    if not is_repo_name(repo_id):
-        gr.Info(f"Error: Invalid repo ID: {repo_id}.")
-        return []
-    if not private_ok and not is_repo_exists(repo_id):
-        gr.Info(f"Error: Repo doesn't exist: {repo_id}.")
-        return []
-    os.makedirs(dir, exist_ok=True)
-    model_name = repo_id.split("/")[-1]
-    app_py = f"""import gradio as gr
-import os
-demo = gr.load("{repo_id}", src="models", hf_token=os.environ.get("HF_TOKEN")).launch()
-"""
-    readme_md = f"""---
-title: {model_name} Demo
-emoji: 🖼
-colorFrom: purple
-colorTo: red
-sdk: gradio
-sdk_version: {gradio_version}
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
-"""
-    app_path = str(Path(dir, "app.py"))
-    with open(app_path, mode='w', encoding="utf-8") as f:
-        f.write(app_py)
-    readme_path = str(Path(dir, "README.md"))
-    with open(readme_path, mode='w', encoding="utf-8") as f:
-        f.write(readme_md)
-    return [app_path, readme_path]
-
-def upload_to_space(repo_id, app_path, readme_path):
-    api = HfApi()
-    token = HfFolder.get_token()
-    
-    # Create the repo if it does not exist
-    if not is_repo_exists(repo_id):
-        create_repo(repo_id, repo_type="space", token=token, private=False)
-
-    # Upload files to the repository
-    operations = [
-        CommitOperationAdd(path_in_repo="app.py", path_or_fileobj=app_path),
-        CommitOperationAdd(path_in_repo="README.md", path_or_fileobj=readme_path),
-    ]
-    api.create_commit(
-        repo_id=repo_id,
-        repo_type="space",
-        operations=operations,
-        commit_message="Add Gradio app and README",
-        token=token,
-    )
-    print(f"Files uploaded successfully to {repo_id}.")
-
-def get_t2i_space_contents(repo_id: str, gradio_version: str, private_ok: bool):
-    paths = save_space_contents(repo_id, gradio_version, private_ok, "./temp/")
-    if paths:
-        upload_to_space(repo_id, *paths)
+from pathlib import Path
+import os
+import gradio as gr
+from huggingface_hub import HfApi, HfFolder, CommitOperationAdd, create_repo
+
+
+def is_repo_name(s):
+    import re
+    return re.fullmatch(r'^[^/,\s]+?/[^/,\s]+?$', s)
+
+
+def get_token():
+    try:
+        token = HfFolder.get_token()
+    except Exception as e:
+        token = ""
+    return token
+
+
+def is_repo_exists(repo_id: str):
+    api = HfApi(token=get_token())
+    try:
+        if api.repo_exists(repo_id=repo_id, repo_type="model"): return True
+        else: return False
+    except Exception as e:
+        print(f"Error: Failed to connect {repo_id}. {e}")
+        return True # for safe
+
+
+def is_repo_t2i(repo_id: str):
+    api = HfApi(token=get_token())
+    try:
+        model_info = api.repo_info(repo_id=repo_id, repo_type="model")
+        if model_info.pipeline_tag == "text-to-image": return True
+        else: return False
+    except Exception as e:
+        print(f"Error: Failed to connect {repo_id}. {e}")
+        return True # for safe
+
+
+def save_space_contents(repo_id: str, gradio_version: str, private_ok: bool, dir: str, template_dir: str = "template"):
+    if not is_repo_name(repo_id):
+        gr.Info(f"Error: Invalid repo ID: {repo_id}.")
+        return []
+    if not private_ok and not is_repo_exists(repo_id):
+        gr.Info(f"Error: Repo doesn't exist: {repo_id}.")
+        return []
+    os.makedirs(dir, exist_ok=True)
+    model_name = repo_id.split("/")[-1]
+    files = []
+    for readpath in Path(template_dir).glob("*.*"):
+        with open(readpath, mode='r', encoding="utf-8") as f:
+            content = str(f.read())
+        content = content.format(repo_id=repo_id, model_name=model_name, gradio_version=gradio_version)
+        writepath = str(Path(dir, readpath.name))
+        with open(writepath, mode='w', encoding="utf-8") as f:
+            f.write(content)
+        files.append(writepath)
+    return files
+
+
+def upload_to_space(repo_id: str, paths: list[str], is_private: bool, is_setkey: bool):
+    token = get_token()
+    api = HfApi(token=token)
+    try:
+        # Create the repo if it does not exist
+        if not is_repo_exists(repo_id):
+            create_repo(repo_id, repo_type="space", space_sdk="gradio", token=token, private=is_private,
+                        space_secrets=[{"key": "HF_TOKEN", "value": token}] if is_setkey else [{"key": "HF_TOKEN", "value": ""}])
+        # Upload files to the repository
+        operations = [CommitOperationAdd(path_in_repo=Path(p).name, path_or_fileobj=p) for p in paths]
+        api.create_commit(repo_id=repo_id, repo_type="space", operations=operations,
+                          commit_message="Add Gradio app and README", token=token)
+        print(f"Files uploaded successfully to {repo_id}.")
+        return repo_id
+    except Exception as e:
+        raise gr.Error(f"Failed to upload files to {repo_id}. {e}")
+
+
+def get_t2i_space_contents(repo_id: str, gradio_version: str, private_ok: bool, hf_user: str, hf_repo: str, is_private: bool, is_setkey: bool, hf_token: str):
+    HfFolder.save_token(hf_token)
+    paths = save_space_contents(repo_id, gradio_version, private_ok, "./temp/", "./template")
+    if hf_repo == "": new_repo_id = f"{hf_user}/{repo_id.split('/')[-1]}" # model name
+    else: new_repo_id = f"{hf_user}/{hf_repo}"
+    if hf_token and hf_user and upload_to_space(new_repo_id, paths, is_private, is_setkey):
+        md = f"Your new repo:<br>https://huggingface.co/spaces/{new_repo_id}"
+    else: md = ""
+    return paths, md