Create new file

app.py

@@ -0,0 +1,46 @@
+import gradio as gr
+from pathlib import Path
+from huggingface_hub import list_repo_files, hf_hub_url
+from collections import defaultdict
+import requests
+
+
+repo_id = 'nateraw/stable-diffusion-gallery'
+all_files = list_repo_files(repo_id, repo_type='dataset')
+data = defaultdict(list)
+for file in all_files:
+    path = Path(file)
+    parent, name = path.parent, path.name
+    if path.name == '.gitattributes':
+        continue
+    if path.suffix in ['.png', '.jpg', '.jpeg']:
+        data[parent.name].append(file)
+
+def fn(run):
+    images = [(hf_hub_url(repo_id=repo_id, filename=img, repo_type='dataset'), f"Seed: {Path(img).stem}") for img in data[run]]
+    prompt_config_url = hf_hub_url(
+        repo_id=repo_id,
+        filename='9fb143fa450011ed95da3947aa01fb28/prompt_config.json',
+        repo_type='dataset'
+    )
+    prompt_config_json = requests.get(prompt_config_url).json()
+    print("IMAGES", images)
+    print("PROMPT CONFIG", prompt_config_json)
+    return prompt_config_json, images
+
+with gr.Blocks() as demo:
+    with gr.Column(variant="panel"):
+        with gr.Row(variant="compact"):
+            run = gr.Dropdown(list(data.keys()), interactive=True).style(container=False,)
+            btn = gr.Button("View images").style(full_width=False)
+
+        with gr.Row(variant="compact"):
+            data_json = gr.Json()
+
+        gallery = gr.Gallery(
+            label="Generated images", show_label=False, elem_id="gallery"
+        ).style(grid=[5], height="auto")
+
+    btn.click(fn, run, [data_json, gallery])
+
+demo.launch(debug=True)