Upload folder using huggingface_hub

README.md

@@ -1,12 +1,6 @@
 ---
 title: Snail
-emoji: 🌍
-colorFrom: gray
-colorTo: yellow
-sdk: gradio
-sdk_version: 4.19.1
 app_file: app.py
-pinned: false
+sdk: gradio
+sdk_version: 4.18.0
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,80 @@
+import base64
+import hashlib
+import json
+import time
+import gradio as gr
+import os
+import queue
+import zstandard as zstd
+
+q = queue.Queue()
+
+"""
+Format:
+{
+    "status": "queue" | "progress" | "done"
+    "prompt": "Some ducks..."
+    "id": "abc123"
+}
+"""
+models = []
+
+WORKERS = []
+
+def enqueue(prompt: str):
+    tm = time.time()
+    hsh =  hashlib.sha256(prompt.encode("utf-8")).hexdigest()
+    md = {
+        "status": "queue",
+        "prompt": prompt,
+        "id": f"{hsh}.{tm}"
+    }
+    models.append(md)
+    q.put(json.dumps(md))
+    
+def dequeue():
+    if not q.empty():
+        pr = json.loads(q.get_nowait())
+        return json.dumps({
+            "status": "ok",
+            "prompt": pr["prompt"],
+            "id": pr["id"]
+        })
+    return json.dumps({
+        "status": "empty"
+    })
+    
+def complete(data):
+    jsn = json.loads(data)
+    for i in range(len(models)):
+        if models[i]["id"] == jsn["_id"]:
+            models[i]["status"] = "done"
+            for fl in jsn["files"]:
+                rd = zstd.decompress(base64.b64decode(fl["data"]))
+                os.makedirs(f"files/{fl['path']}", exist_ok=True)
+                os.rmdir(f"files/{fl['path']}")
+                with open(f"files/{fl['path']}", "wb") as f:
+                    f.write(rd)
+                    f.flush()
+                    f.close()
+            break
+    return json.dumps({"status": "ok"})
+    
+with gr.Blocks() as bl:
+    with gr.Row("panel"):
+        gr.Label("Enter a prompt to generate an image. This is a work-in-progress. Release 1.")
+    with gr.Row("panel"):
+        prompt_input = gr.Textbox(placeholder="Enter a prompt here")
+    with gr.Row("panel"):
+        submit_button = gr.Button(value="Generate")
+    with gr.Row("panel"):
+        image_output = gr.Image(label="Generated Image")
+    with gr.Row("none"):
+        shad_out = gr.Textbox(visible=False)
+        shad_dequeue = gr.Button(value="Dequeue", visible=False)
+        shad_in = gr.Textbox(visible=False)
+        shad_complete = gr.Button(value="Complete", visible=False)
+    submit_button.click(enqueue, inputs=[prompt_input], api_name="enqueue")
+    shad_dequeue.click(dequeue, outputs=[shad_out], api_name="dequeue")
+    shad_complete.click(complete, inputs=[shad_in], outputs=[shad_out], api_name="complete")
+    bl.launch(share=False, debug=True)

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+zstandard