Add e621 API test app

app.py

@@ -0,0 +1,47 @@
+import requests
+import gradio as gr
+import json
+
+USER_AGENT = "e621-hf-test/1.0 (by lain-studio on e621)"
+
+def test_e621(rating: str, tags: str, count: int):
+    query_tags = f"rating:{rating}"
+    if tags.strip():
+        query_tags += "+" + tags.strip().replace(" ", "+")
+    query_tags += "+order:random"
+
+    url = "https://e621.net/posts.json"
+    params = {"limit": count, "tags": query_tags}
+    headers = {"User-Agent": USER_AGENT}
+
+    try:
+        resp = requests.get(url, params=params, headers=headers, timeout=10)
+        resp.raise_for_status()
+        posts = resp.json().get("posts", [])
+        results = []
+        for p in posts:
+            f = p["file"]
+            results.append({
+                "id": p["id"],
+                "rating": p["rating"],
+                "md5": f["md5"],
+                "url": f["url"],
+                "ext": f["ext"],
+            })
+        return json.dumps({"status": "ok", "ip_used": "space", "count": len(results), "posts": results}, indent=2)
+    except Exception as e:
+        return json.dumps({"status": "error", "error": str(e)})
+
+demo = gr.Interface(
+    fn=test_e621,
+    inputs=[
+        gr.Textbox(value="s", label="Rating (s/q/e)"),
+        gr.Textbox(value="", label="Extra tags (optional)"),
+        gr.Slider(1, 5, value=2, step=1, label="Count"),
+    ],
+    outputs=gr.Textbox(label="Result", lines=20),
+    title="e621 API Test",
+)
+
+if __name__ == "__main__":
+    demo.launch()