Upload 6 files

README.md

@@ -1,12 +1,12 @@
----
-title: Gradio Demo Space creation helper
-emoji: 🐶
-colorFrom: yellow
-colorTo: red
-sdk: gradio
-sdk_version: 5.11.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Gradio Demo Space creation helper V2
+emoji: 🐶
+colorFrom: yellow
+colorTo: red
+sdk: gradio
+sdk_version: 5.11.0
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,34 +1,46 @@
-import gradio as gr
-from t2i_space import get_t2i_space_contents
-
-css = """"""
-
-with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
-    gr.Markdown("# Gradio Demo Space creation helper")
-    gr.Markdown(
-        f"""
-**The steps are the following**:
-- Input model Repo ID which you want to use. e.g. 'user/model'.
-- Click "Submit" and download generated README.md and app.py.
-- [Create your new Gradio space](https://huggingface.co/new-space) with blank.
-- Upload README.md and app.py to your space.
-- If you got 500 / 504 error, it happens often, just restart space.
-- If it does not work no matter how many times you try, there is often a problem with the settings of the original repo itself, or there is no generation function.
-            """
-    )
-    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="5.11.0", max_lines=1)
-            private_ok = gr.Checkbox(label="Allow private repo", value=True)
-        run_button = gr.Button(value="Submit")
-        space_file = gr.Files(label="Output", interactive=False)
-
-    gr.on(
-        triggers=[repo_id.submit, run_button.click],
-        fn=get_t2i_space_contents,
-        inputs=[repo_id, gradio_version, private_ok],
-        outputs=[space_file],
-    )
-
-demo.queue().launch(ssr_mode=False)
+import gradio as gr
+from t2i_space import get_t2i_space_contents
+
+css = """
+.title { text-align: center; }
+"""
+
+with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css) as demo:
+    gr.Markdown("# Gradio Demo Space creation helper V2", elem_classes="title")
+    gr.Markdown(
+        f"""
+**The steps are the following**:
+- Input model Repo ID which you want to use. e.g. 'user/model'.
+- Click "Submit" and download generated README.md and app.py.
+- [Create your new Gradio space](https://huggingface.co/new-space) with blank.
+- Upload README.md and app.py to your space.
+- If you got 500 / 504 error, it happens often, just restart space.
+- If it does not work no matter how many times you try, there is often a problem with the settings of the original repo itself, or there is no generation function.
+            """
+    )
+    with gr.Column():
+        repo_id = gr.Textbox(label="Model repo ID", placeholder="username/modelname", value="", max_lines=1)
+        with gr.Accordion("Advanced", open=False):
+            with gr.Row():
+                gradio_version = gr.Textbox(label="Gradio version", placeholder="username/modelname", value="5.11.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, hf_user, hf_repo, is_private, is_setkey, hf_token],
+        outputs=[space_file, output_md],
+    )
+
+demo.queue()
+demo.launch()

t2i_space.py

@@ -1,37 +1,45 @@
 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 is_repo_exists(repo_id):
-    from huggingface_hub import HfApi
-    api = HfApi()
+def get_token():
+    try:
+        token = HfFolder.get_token()
+    except Exception:
+        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}.")
+        print(f"Error: Failed to connect {repo_id}. {e}")
         return True # for safe
 
 
-def is_repo_t2i(repo_id):
-    from huggingface_hub import HfApi
-    api = HfApi()
+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}.")
+        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):
-    if not is_repo_name(repo_id): #  or not is_repo_t2i(repo_id)
+
+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):
@@ -39,32 +47,44 @@ def save_space_contents(repo_id: str, gradio_version: str, private_ok: bool, dir
         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
----
+    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
 
-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: 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):
+            if is_setkey: create_repo(repo_id, repo_type="space", space_sdk="gradio", token=token, private=is_private,
+                                      space_secrets=[{"key": "HF_TOKEN", "value": token}])
+            else: create_repo(repo_id, repo_type="space", space_sdk="gradio", token=token, private=is_private)
+            
+        # 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):
-    return save_space_contents(repo_id, gradio_version, private_ok, "./temp/")
+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

template/app.py

@@ -1,3 +1,3 @@
 import gradio as gr
 import os
-demo = gr.load("{repo_id}", src="models", hf_token=os.environ.get("HF_TOKEN"), examples=None).launch(ssr_mode=False)
+demo = gr.load("{repo_id}", src="models", hf_token=os.environ.get("HF_TOKEN"), examples=None).launch()