Upload 4 files

README.md

@@ -1,12 +1,12 @@
----
-title: T2i Demo Helper
-emoji: 😻
-colorFrom: indigo
-colorTo: red
-sdk: gradio
-sdk_version: 4.39.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Gradio Text-to-Image Demo Space creation helper
+emoji: 🐶
+colorFrom: yellow
+colorTo: red
+sdk: gradio
+sdk_version: 4.38.1
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,32 @@
+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 Text-to-Image Demo Space creation helper")
+    gr.Markdown(
+        f"""
+**The steps are the following**:
+- Input text-to-image 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 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)
+        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, space_file],
+        outputs=[space_file],
+    )
+
+demo.queue()
+demo.launch()

requirements.txt

@@ -0,0 +1 @@
+huggingface_hub

t2i_space.py

@@ -0,0 +1,67 @@
+from pathlib import Path
+
+
+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()
+    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}. ")
+        return True # for safe
+
+
+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}. ")
+        return True # for safe
+    
+
+def save_space_contents(repo_id: str, dir: str):
+    if not is_repo_name(repo_id) or not is_repo_exists(repo_id) or not is_repo_t2i(repo_id):
+        print(f"Error: Invalid repo ID: {repo_id}. ")
+        return []
+    model_name = repo_id.split("/")[-1]
+    app_py = f"""
+import gradio as gr
+demo = gr.load("{repo_id}", src="models").launch()
+demo.launch()
+"""
+    readme_md = f"""---
+title: {model_name} Text-to-Image
+emoji: 🖼
+colorFrom: purple
+colorTo: red
+sdk: gradio
+sdk_version: 4.38.1
+app_file: app.py
+pinned: false
+short_description: {repo_id} | Text-to-Image
+---
+
+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 get_t2i_space_contents(repo_id: str):
+    return save_space_contents(repo_id, "./temp/")