@gradio/client 0.0.1 → 0.1.0

package/package.json

@@ -1,9 +1,32 @@
 {
   "name": "@gradio/client",
-  "version": "0.0.1",
+  "version": "0.1.0",
   "description": "Gradio UI packages",
   "type": "module",
-  "main": "src/index.ts",
+  "main": "dist/index.js",
   "author": "",
-  "license": "ISC"
+  "license": "ISC",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js"
+    },
+    "./package.json": "./package.json"
+  },
+  "dependencies": {
+    "bufferutil": "^4.0.7",
+    "semiver": "^1.1.0",
+    "ws": "^8.13.0"
+  },
+  "devDependencies": {
+    "@types/ws": "^8.5.4",
+    "esbuild": "^0.17.14"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "scripts": {
+    "bundle": "vite build --ssr",
+    "generate_types": "tsc",
+    "build": "pnpm bundle && pnpm generate_types"
+  }
 }

package/src/client.node-test.ts

@@ -0,0 +1,172 @@
+import { test, describe, assert } from "vitest";
+import { readFileSync } from "fs";
+import { join, dirname } from "path";
+import { fileURLToPath } from "url";
+import { Blob } from "node:buffer";
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const image_path = join(
+	__dirname,
+	"..",
+	"..",
+	"..",
+	"demo",
+	"kitchen_sink",
+	"files",
+	"lion.jpg"
+);
+
+import { walk_and_store_blobs, client, handle_blob } from "./client";
+
+describe.skip("extract blob parts", () => {
+	test("convert Buffer to Blob", async () => {
+		const image = readFileSync(image_path);
+		await client("gradio/hello_world_main");
+		const parts = walk_and_store_blobs({
+			data: {
+				image
+			}
+		});
+
+		assert.isTrue(parts[0].blob instanceof Blob);
+	});
+
+	test("leave node Blob as Blob", async () => {
+		const image = new Blob([readFileSync(image_path)]);
+
+		await client("gradio/hello_world_main");
+		const parts = walk_and_store_blobs({
+			data: {
+				image
+			}
+		});
+
+		assert.isTrue(parts[0].blob instanceof Blob);
+	});
+
+	test("handle deep structures", async () => {
+		const image = new Blob([readFileSync(image_path)]);
+
+		await client("gradio/hello_world_main");
+		const parts = walk_and_store_blobs({
+			a: {
+				b: {
+					data: {
+						image
+					}
+				}
+			}
+		});
+
+		assert.isTrue(parts[0].blob instanceof Blob);
+	});
+
+	test("handle deep structures with arrays", async () => {
+		const image = new Blob([readFileSync(image_path)]);
+
+		await client("gradio/hello_world_main");
+		const parts = walk_and_store_blobs({
+			a: [
+				{
+					b: [
+						{
+							data: [
+								{
+									image
+								}
+							]
+						}
+					]
+				}
+			]
+		});
+
+		assert.isTrue(parts[0].blob instanceof Blob);
+	});
+
+	test("handle deep structures with arrays 2", async () => {
+		const image = new Blob([readFileSync(image_path)]);
+
+		await client("gradio/hello_world_main");
+		const obj = {
+			a: [
+				{
+					b: [
+						{
+							data: [[image], image, [image, [image]]]
+						}
+					]
+				}
+			]
+		};
+		const parts = walk_and_store_blobs(obj);
+
+		function map_path(
+			obj: Record<string, any>,
+			parts: { path: string[]; blob: any }[]
+		) {
+			const { path, blob } = parts[parts.length - 1];
+			let ref = obj;
+			path.forEach((p) => (ref = ref[p]));
+
+			return ref === blob;
+		}
+
+		assert.isTrue(parts[0].blob instanceof Blob);
+		// assert.isTrue(map_path(obj, parts));
+	});
+});
+
+describe("handle_blob", () => {
+	test("handle blobs", async () => {
+		const image = new Blob([readFileSync(image_path)]);
+
+		const app = await client("gradio/hello_world_main");
+		const obj = [
+			{
+				a: [
+					{
+						b: [
+							{
+								data: [[image], image, [image, [image]]]
+							}
+						]
+					}
+				]
+			}
+		];
+
+		const parts = await handle_blob(app.config.root, obj, undefined);
+		//@ts-ignore
+		// assert.isString(parts.data[0].a[0].b[0].data[0][0]);
+	});
+});
+
+describe.skip("private space", () => {
+	test("can access a private space", async () => {
+		const image = new Blob([readFileSync(image_path)]);
+
+		const app = await client("pngwn/hello_world", {
+			hf_token: "hf_"
+		});
+
+		console.log(app);
+		const obj = [
+			{
+				a: [
+					{
+						b: [
+							{
+								data: [[image], image, [image, [image]]]
+							}
+						]
+					}
+				]
+			}
+		];
+
+		const parts = await handle_blob(app.config.root, obj, "hf_");
+		//@ts-ignore
+		assert.isString(parts.data[0].a[0].b[0].data[0][0]);
+	});
+});