Upload 6 files

README.md

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

app.py

@@ -1,46 +1,34 @@
-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.7.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()
+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="4.44.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)

t2i_space.py

@@ -1,45 +1,37 @@
 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:
-        token = ""
-    return token
-
-
-def is_repo_exists(repo_id: str):
-    api = HfApi(token=get_token())
+def is_repo_exists(repo_id):
+    from huggingface_hub import HfApi
+    api = HfApi()
     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}")
+        print(f"Error: Failed to connect {repo_id}.")
         return True # for safe
 
 
-def is_repo_t2i(repo_id: str):
-    api = HfApi(token=get_token())
+def is_repo_t2i(repo_id):
+    from huggingface_hub import HfApi
+    api = HfApi()
     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}")
+        print(f"Error: Failed to connect {repo_id}.")
         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):
+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)
         gr.Info(f"Error: Invalid repo ID: {repo_id}.")
         return []
     if not private_ok and not is_repo_exists(repo_id):
@@ -47,44 +39,32 @@ 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]
-    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
+    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
 
-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}")
+"""
+    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 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
+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/")

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()
+demo = gr.load("{repo_id}", src="models", hf_token=os.environ.get("HF_TOKEN"), examples=None).launch(ssr_mode=False)