@gradio/client 0.1.4 → 0.2.1

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @gradio/client
 
+## 0.2.1
+
+### Features
+
+- [#5173](https://github.com/gradio-app/gradio/pull/5173) [`730f0c1d`](https://github.com/gradio-app/gradio/commit/730f0c1d54792eb11359e40c9f2326e8a6e39203) - Ensure gradio client works as expected for functions that return nothing.  Thanks [@raymondtri](https://github.com/raymondtri)!
+
+## 0.2.0
+
+### Features
+
+- [#5133](https://github.com/gradio-app/gradio/pull/5133) [`61129052`](https://github.com/gradio-app/gradio/commit/61129052ed1391a75c825c891d57fa0ad6c09fc8) - Update dependency esbuild to ^0.19.0. Thanks [@renovate](https://github.com/apps/renovate)!
+- [#5035](https://github.com/gradio-app/gradio/pull/5035) [`8b4eb8ca`](https://github.com/gradio-app/gradio/commit/8b4eb8cac9ea07bde31b44e2006ca2b7b5f4de36) - JS Client: Fixes cannot read properties of null (reading 'is_file'). Thanks [@raymondtri](https://github.com/raymondtri)!
+
+### Fixes
+
+- [#5075](https://github.com/gradio-app/gradio/pull/5075) [`67265a58`](https://github.com/gradio-app/gradio/commit/67265a58027ef1f9e4c0eb849a532f72eaebde48) - Allow supporting >1000 files in `gr.File()` and `gr.UploadButton()`. Thanks [@abidlabs](https://github.com/abidlabs)!
+
 ## 0.1.4
 
 ### Patch Changes
@@ -40,4 +57,4 @@
 
 - [#3692](https://github.com/gradio-app/gradio/pull/3692) [`48e8b113`](https://github.com/gradio-app/gradio/commit/48e8b113f4b55e461d9da4f153bf72aeb4adf0f1) Thanks [@pngwn](https://github.com/pngwn)! - Ensure client works in node, create ESM bundle and generate typescript declaration files.
 
-- [#3605](https://github.com/gradio-app/gradio/pull/3605) [`ae4277a9`](https://github.com/gradio-app/gradio/commit/ae4277a9a83d49bdadfe523b0739ba988128e73b) Thanks [@pngwn](https://github.com/pngwn)! - Update readme.
+- [#3605](https://github.com/gradio-app/gradio/pull/3605) [`ae4277a9`](https://github.com/gradio-app/gradio/commit/ae4277a9a83d49bdadfe523b0739ba988128e73b) Thanks [@pngwn](https://github.com/pngwn)! - Update readme.

package/README.md

@@ -65,8 +65,8 @@ Applications hosted on Hugging Face spaces can be in a number of different state
 import { client, type SpaceStatus } from "@gradio/client";
 
 const app = await client("user/space-name", {
-  // The space_status parameter does not need to be manually annotated, this is just for illustration.
-  space_status: (space_status: SpaceStatus) => console.log(space_status),
+	// The space_status parameter does not need to be manually annotated, this is just for illustration.
+	space_status: (space_status: SpaceStatus) => console.log(space_status)
 });
 ```
 
@@ -157,22 +157,22 @@ The status payload look like this:
 
 ```ts
 interface Status {
-  queue: boolean;
-  code?: string;
-  success?: boolean;
-  stage: "pending" | "error" | "complete" | "generating";
-  size?: number;
-  position?: number;
-  eta?: number;
-  message?: string;
-  progress_data?: Array<{
-    progress: number | null;
-    index: number | null;
-    length: number | null;
-    unit: string | null;
-    desc: string | null;
-  }>;
-  time?: Date;
+	queue: boolean;
+	code?: string;
+	success?: boolean;
+	stage: "pending" | "error" | "complete" | "generating";
+	size?: number;
+	position?: number;
+	eta?: number;
+	message?: string;
+	progress_data?: Array<{
+		progress: number | null;
+		index: number | null;
+		length: number | null;
+		unit: string | null;
+		desc: string | null;
+	}>;
+	time?: Date;
 }
 ```
 
@@ -183,9 +183,9 @@ import { client } from "@gradio/client";
 
 const app = await client("user/space-name");
 const submission = app
-  .submit("/predict", payload)
-  .on("data", (data) => console.log(data))
-  .on("status", (status: Status) => console.log(status));
+	.submit("/predict", payload)
+	.on("data", (data) => console.log(data))
+	.on("status", (status: Status) => console.log(status));
 ```
 
 ##### `off`
@@ -229,8 +229,8 @@ import { client } from "@gradio/client";
 
 const app = await client("user/space-name");
 const submission = app
-  .submit("/predict", payload)
-  .on("data", (data) => console.log(data));
+	.submit("/predict", payload)
+	.on("data", (data) => console.log(data));
 
 // later
 
@@ -271,7 +271,7 @@ The duplicate function will attempt to duplicate the space that is referenced an
 import { duplicate } from "@gradio/client";
 
 const app = await duplicate("user/space-name", {
-  hf_token: "hf_...",
+	hf_token: "hf_..."
 });
 ```
 
@@ -295,8 +295,8 @@ This is an optional property specific to `duplicate`'s options object and will d
 import { duplicate } from "@gradio/client";
 
 const app = await duplicate("user/space-name", {
-  hf_token: "hf_...",
-  private: true,
+	hf_token: "hf_...",
+	private: true
 });
 ```
 
@@ -308,9 +308,9 @@ This is an optional property specific to `duplicate`'s options object and will s
 import { duplicate } from "@gradio/client";
 
 const app = await duplicate("user/space-name", {
-  hf_token: "hf_...",
-  private: true,
-  timeout: 5,
+	hf_token: "hf_...",
+	private: true,
+	timeout: 5
 });
 ```
 
@@ -332,8 +332,8 @@ Possible hardware options are:
 import { duplicate } from "@gradio/client";
 
 const app = await duplicate("user/space-name", {
-  hf_token: "hf_...",
-  private: true,
-  hardware: "a10g-small",
+	hf_token: "hf_...",
+	private: true,
+	hardware: "a10g-small"
 });
 ```

package/dist/client.d.ts

@@ -20,27 +20,25 @@ export declare function duplicate(app_reference: string, options: {
     hf_token: `hf_${string}`;
     private?: boolean;
     status_callback: SpaceStatusCallback;
-    hardware?: typeof hardware_types[number];
+    hardware?: (typeof hardware_types)[number];
     timeout?: number;
 }): Promise<client_return>;
-/**
- * We need to inject a customized fetch implementation for the Wasm version.
- */
-export declare function api_factory(fetch_implementation: typeof fetch): {
+interface Client {
     post_data: (url: string, body: unknown, token?: `hf_${string}`) => Promise<[PostResponse, number]>;
-    upload_files: (root: string, files: Array<File>, token?: `hf_${string}`) => Promise<UploadResponse>;
-    client: (app_reference: string, options?: {
+    upload_files: (root: string, files: File[], token?: `hf_${string}`) => Promise<UploadResponse>;
+    client: (app_reference: string, options: {
         hf_token?: `hf_${string}`;
         status_callback?: SpaceStatusCallback;
         normalise_files?: boolean;
     }) => Promise<client_return>;
-    handle_blob: (endpoint: string, data: unknown[], api_info: any, token?: `hf_${string}`) => Promise<unknown[]>;
-};
-export declare const post_data: (url: string, body: unknown, token?: `hf_${string}`) => Promise<[PostResponse, number]>, upload_files: (root: string, files: Array<File>, token?: `hf_${string}`) => Promise<UploadResponse>, client: (app_reference: string, options?: {
+    handle_blob: (endpoint: string, data: unknown[], api_info: ApiInfo<JsApiData>, token?: `hf_${string}`) => Promise<unknown[]>;
+}
+export declare function api_factory(fetch_implementation: typeof fetch): Client;
+export declare const post_data: (url: string, body: unknown, token?: `hf_${string}`) => Promise<[PostResponse, number]>, upload_files: (root: string, files: File[], token?: `hf_${string}`) => Promise<UploadResponse>, client: (app_reference: string, options: {
     hf_token?: `hf_${string}`;
     status_callback?: SpaceStatusCallback;
     normalise_files?: boolean;
-}) => Promise<client_return>, handle_blob: (endpoint: string, data: unknown[], api_info: any, token?: `hf_${string}`) => Promise<unknown[]>;
+}) => Promise<client_return>, handle_blob: (endpoint: string, data: unknown[], api_info: ApiInfo<JsApiData>, token?: `hf_${string}`) => Promise<unknown[]>;
 interface ApiData {
     label: string;
     type: {
@@ -68,6 +66,11 @@ interface ApiInfo<T extends ApiData | JsApiData> {
         [key: string]: EndpointInfo<T>;
     };
 }
-export declare function walk_and_store_blobs(param: any, type?: any, path?: any[], root?: boolean, api_info?: any): Promise<any[]>;
+export declare function walk_and_store_blobs(param: any, type?: any, path?: any[], root?: boolean, api_info?: any): Promise<{
+    path: string[];
+    data: string | false;
+    type: string;
+    blob: Blob | false;
+}[]>;
 export {};
 //# sourceMappingURL=client.d.ts.map

package/dist/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAEA,OAAO,EAQN,cAAc,EACd,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EACX,SAAS,EACT,aAAa,EAIb,YAAY,EACZ,cAAc,EAGd,mBAAmB,EAEnB,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAEzC,KAAK,KAAK,GAAG,CAAC,CAAC,SAAS,SAAS,EAChC,SAAS,EAAE,CAAC,EACZ,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,KACtB,YAAY,CAAC;AAClB,KAAK,OAAO,GAAG,CACd,QAAQ,EAAE,MAAM,GAAG,MAAM,EACzB,IAAI,CAAC,EAAE,OAAO,EAAE,EAChB,UAAU,CAAC,EAAE,OAAO,KAChB,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB,KAAK,aAAa,GAAG;IACpB,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,CACP,QAAQ,EAAE,MAAM,GAAG,MAAM,EACzB,IAAI,CAAC,EAAE,OAAO,EAAE,EAChB,UAAU,CAAC,EAAE,OAAO,KAChB,YAAY,CAAC;IAClB,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;CACtD,CAAC;AAEF,KAAK,YAAY,GAAG;IACnB,EAAE,EAAE,KAAK,CAAC;IACV,GAAG,EAAE,KAAK,CAAC;IACX,MAAM,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,OAAO,EAAE,MAAM,IAAI,CAAC;CACpB,CAAC;AAKF,eAAO,IAAI,QAAQ,KAAA,CAAC;AAEpB,wBAAsB,SAAS,CAC9B,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE;IACR,QAAQ,EAAE,MAAM,MAAM,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,eAAe,EAAE,mBAAmB,CAAC;IACrC,QAAQ,CAAC,EAAE,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;IACzC,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB,0BAyED;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,oBAAoB,EAAE,OAAO,KAAK;qBAIvD,MAAM,QACL,OAAO,UACL,MAAM,MAAM,EAAE,KACpB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;yBAsB3B,MAAM,SACL,MAAM,IAAI,CAAC,UACV,MAAM,MAAM,EAAE,KACpB,QAAQ,cAAc,CAAC;4BA0BV,MAAM,YACZ;QACR,QAAQ,CAAC,EAAE,MAAM,MAAM,EAAE,CAAC;QAC1B,eAAe,CAAC,EAAE,mBAAmB,CAAC;QACtC,eAAe,CAAC,EAAE,OAAO,CAAC;KAC1B,KACC,QAAQ,aAAa,CAAC;4BA8ed,MAAM,QACV,OAAO,EAAE,yBAEP,MAAM,MAAM,EAAE,KACpB,QAAQ,OAAO,EAAE,CAAC;EAuCrB;AAED,eAAO,MAAQ,SAAS,QAvlBjB,MAAM,QACL,OAAO,UACL,MAAM,MAAM,EAAE,KACpB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,EAolBT,YAAY,SA9jB9B,MAAM,SACL,MAAM,IAAI,CAAC,UACV,MAAM,MAAM,EAAE,KACpB,QAAQ,cAAc,CAAC,EA2jBa,MAAM,kBAjiB7B,MAAM,YACZ;IACR,QAAQ,CAAC,EAAE,MAAM,MAAM,EAAE,CAAC;IAC1B,eAAe,CAAC,EAAE,mBAAmB,CAAC;IACtC,eAAe,CAAC,EAAE,OAAO,CAAC;CAC1B,KACC,QAAQ,aAAa,CAAC,EA2hBsB,WAAW,aA7C/C,MAAM,QACV,OAAO,EAAE,yBAEP,MAAM,MAAM,EAAE,KACpB,QAAQ,OAAO,EAAE,CA0CF,CAAC;AAyEpB,UAAU,OAAO;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE;QACL,IAAI,EAAE,GAAG,CAAC;QACV,WAAW,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,GAAG,CAAC;CACpB;AAED,UAAU,SAAS;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,GAAG,CAAC;CACnB;AAED,UAAU,YAAY,CAAC,CAAC,SAAS,OAAO,GAAG,SAAS;IACnD,UAAU,EAAE,CAAC,EAAE,CAAC;IAChB,OAAO,EAAE,CAAC,EAAE,CAAC;CACb;AACD,UAAU,OAAO,CAAC,CAAC,SAAS,OAAO,GAAG,SAAS;IAC9C,eAAe,EAAE;QAChB,CAAC,GAAG,EAAE,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;KAC/B,CAAC;IACF,iBAAiB,EAAE;QAClB,CAAC,GAAG,EAAE,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;KAC/B,CAAC;CACF;AAmID,wBAAsB,oBAAoB,CACzC,KAAK,KAAA,EACL,IAAI,MAAY,EAChB,IAAI,QAAK,EACT,IAAI,UAAQ,EACZ,QAAQ,MAAY,kBAyEpB"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAEA,OAAO,EAQN,cAAc,EACd,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EACX,SAAS,EACT,aAAa,EAIb,YAAY,EACZ,cAAc,EAGd,mBAAmB,EAEnB,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAEzC,KAAK,KAAK,GAAG,CAAC,CAAC,SAAS,SAAS,EAChC,SAAS,EAAE,CAAC,EACZ,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,KACtB,YAAY,CAAC;AAClB,KAAK,OAAO,GAAG,CACd,QAAQ,EAAE,MAAM,GAAG,MAAM,EACzB,IAAI,CAAC,EAAE,OAAO,EAAE,EAChB,UAAU,CAAC,EAAE,OAAO,KAChB,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB,KAAK,aAAa,GAAG;IACpB,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,CACP,QAAQ,EAAE,MAAM,GAAG,MAAM,EACzB,IAAI,CAAC,EAAE,OAAO,EAAE,EAChB,UAAU,CAAC,EAAE,OAAO,KAChB,YAAY,CAAC;IAClB,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;CACtD,CAAC;AAEF,KAAK,YAAY,GAAG;IACnB,EAAE,EAAE,KAAK,CAAC;IACV,GAAG,EAAE,KAAK,CAAC;IACX,MAAM,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,OAAO,EAAE,MAAM,IAAI,CAAC;CACpB,CAAC;AAKF,eAAO,IAAI,QAAQ,KAAA,CAAC;AAEpB,wBAAsB,SAAS,CAC9B,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE;IACR,QAAQ,EAAE,MAAM,MAAM,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,eAAe,EAAE,mBAAmB,CAAC;IACrC,QAAQ,CAAC,EAAE,CAAC,OAAO,cAAc,CAAC,CAAC,MAAM,CAAC,CAAC;IAC3C,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB,GACC,OAAO,CAAC,aAAa,CAAC,CAmExB;AAED,UAAU,MAAM;IACf,SAAS,EAAE,CACV,GAAG,EAAE,MAAM,EACX,IAAI,EAAE,OAAO,EACb,KAAK,CAAC,EAAE,MAAM,MAAM,EAAE,KAClB,OAAO,CAAC,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC;IACrC,YAAY,EAAE,CACb,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,IAAI,EAAE,EACb,KAAK,CAAC,EAAE,MAAM,MAAM,EAAE,KAClB,OAAO,CAAC,cAAc,CAAC,CAAC;IAC7B,MAAM,EAAE,CACP,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE;QACR,QAAQ,CAAC,EAAE,MAAM,MAAM,EAAE,CAAC;QAC1B,eAAe,CAAC,EAAE,mBAAmB,CAAC;QACtC,eAAe,CAAC,EAAE,OAAO,CAAC;KAC1B,KACG,OAAO,CAAC,aAAa,CAAC,CAAC;IAC5B,WAAW,EAAE,CACZ,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,EAAE,EACf,QAAQ,EAAE,OAAO,CAAC,SAAS,CAAC,EAC5B,KAAK,CAAC,EAAE,MAAM,MAAM,EAAE,KAClB,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;CACxB;AAED,wBAAgB,WAAW,CAAC,oBAAoB,EAAE,OAAO,KAAK,GAAG,MAAM,CAmnBtE;AAED,eAAO,MAAQ,SAAS,QA9oBjB,MAAM,QACL,OAAO,UACL,MAAM,MAAM,EAAE,KAClB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,EA2oBX,YAAY,SAzoB9B,MAAM,SACL,IAAI,EAAE,UACL,MAAM,MAAM,EAAE,KAClB,QAAQ,cAAc,CAAC,EAsoBW,MAAM,kBApoB7B,MAAM,WACZ;IACR,QAAQ,CAAC,EAAE,MAAM,MAAM,EAAE,CAAC;IAC1B,eAAe,CAAC,EAAE,mBAAmB,CAAC;IACtC,eAAe,CAAC,EAAE,OAAO,CAAC;CAC1B,KACG,QAAQ,aAAa,CAAC,EA8nBoB,WAAW,aA5nB/C,MAAM,QACV,OAAO,EAAE,YACL,QAAQ,SAAS,CAAC,UACpB,MAAM,MAAM,EAAE,KAClB,QAAQ,OAAO,EAAE,CAynBJ,CAAC;AAoEpB,UAAU,OAAO;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE;QACL,IAAI,EAAE,GAAG,CAAC;QACV,WAAW,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,GAAG,CAAC;CACpB;AAED,UAAU,SAAS;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,GAAG,CAAC;CACnB;AAED,UAAU,YAAY,CAAC,CAAC,SAAS,OAAO,GAAG,SAAS;IACnD,UAAU,EAAE,CAAC,EAAE,CAAC;IAChB,OAAO,EAAE,CAAC,EAAE,CAAC;CACb;AACD,UAAU,OAAO,CAAC,CAAC,SAAS,OAAO,GAAG,SAAS;IAC9C,eAAe,EAAE;QAChB,CAAC,GAAG,EAAE,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;KAC/B,CAAC;IACF,iBAAiB,EAAE;QAClB,CAAC,GAAG,EAAE,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;KAC/B,CAAC;CACF;AAiID,wBAAsB,oBAAoB,CACzC,KAAK,KAAA,EACL,IAAI,MAAY,EAChB,IAAI,QAAK,EACT,IAAI,UAAQ,EACZ,QAAQ,MAAY,GAClB,OAAO,CACT;IACC,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,IAAI,EAAE,MAAM,GAAG,KAAK,CAAC;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,IAAI,GAAG,KAAK,CAAC;CACnB,EAAE,CACH,CAsEA"}

package/dist/index.js

@@ -13,13 +13,12 @@ function determine_protocol(endpoint) {
         host,
         http_protocol: protocol
       };
-    } else {
-      return {
-        ws_protocol: protocol === "https:" ? "wss" : "ws",
-        http_protocol: protocol,
-        host
-      };
     }
+    return {
+      ws_protocol: protocol === "https:" ? "wss" : "ws",
+      http_protocol: protocol,
+      host
+    };
   }
   return {
     ws_protocol: "wss",
@@ -86,8 +85,7 @@ async function discussions_enabled(space_id) {
     const error = r.headers.get("x-error-message");
     if (error && RE_DISABLED_DISCUSSION.test(error))
       return false;
-    else
-      return true;
+    return true;
   } catch (e) {
     return false;
   }
@@ -193,25 +191,20 @@ async function duplicate(app_reference, options) {
     );
     if (response.status === 409) {
       return client(`${user}/${space_name}`, options);
-    } else {
-      const duplicated_space = await response.json();
-      let original_hardware;
-      if (!hardware) {
-        original_hardware = await get_space_hardware(app_reference, hf_token);
-      }
-      const requested_hardware = hardware || original_hardware || "cpu-basic";
-      await set_space_hardware(
-        `${user}/${space_name}`,
-        requested_hardware,
-        hf_token
-      );
-      await set_space_timeout(
-        `${user}/${space_name}`,
-        timeout || 300,
-        hf_token
-      );
-      return client(duplicated_space.url, options);
     }
+    const duplicated_space = await response.json();
+    let original_hardware;
+    if (!hardware) {
+      original_hardware = await get_space_hardware(app_reference, hf_token);
+    }
+    const requested_hardware = hardware || original_hardware || "cpu-basic";
+    await set_space_hardware(
+      `${user}/${space_name}`,
+      requested_hardware,
+      hf_token
+    );
+    await set_space_timeout(`${user}/${space_name}`, timeout || 300, hf_token);
+    return client(duplicated_space.url, options);
   } catch (e) {
     throw new Error(e);
   }
@@ -240,21 +233,27 @@ function api_factory(fetch_implementation) {
     if (token) {
       headers.Authorization = `Bearer ${token}`;
     }
-    const formData = new FormData();
-    files.forEach((file) => {
-      formData.append("files", file);
-    });
-    try {
-      var response = await fetch_implementation(`${root}/upload`, {
-        method: "POST",
-        body: formData,
-        headers
+    const chunkSize = 1e3;
+    const uploadResponses = [];
+    for (let i = 0; i < files.length; i += chunkSize) {
+      const chunk = files.slice(i, i + chunkSize);
+      const formData = new FormData();
+      chunk.forEach((file) => {
+        formData.append("files", file);
       });
-    } catch (e) {
-      return { error: BROKEN_CONNECTION_MSG };
+      try {
+        var response = await fetch_implementation(`${root}/upload`, {
+          method: "POST",
+          body: formData,
+          headers
+        });
+      } catch (e) {
+        return { error: BROKEN_CONNECTION_MSG };
+      }
+      const output = await response.json();
+      uploadResponses.push(...output);
     }
-    const output = await response.json();
-    return { files: output };
+    return { files: uploadResponses };
   }
   async function client2(app_reference, options = { normalise_files: true }) {
     return new Promise(async (res) => {
@@ -347,22 +346,37 @@ function api_factory(fetch_implementation) {
       function predict(endpoint, data, event_data) {
         let data_returned = false;
         let status_complete = false;
+        let dependency;
+        if (typeof endpoint === "number") {
+          dependency = config.dependencies[endpoint];
+        } else {
+          const trimmed_endpoint = endpoint.replace(/^\//, "");
+          dependency = config.dependencies[api_map[trimmed_endpoint]];
+        }
+        if (dependency.types.continuous) {
+          throw new Error(
+            "Cannot call predict on this function as it may run forever. Use submit instead"
+          );
+        }
         return new Promise((res2, rej) => {
           const app = submit(endpoint, data, event_data);
+          let result;
           app.on("data", (d) => {
-            data_returned = true;
             if (status_complete) {
               app.destroy();
+              res2(d);
             }
-            res2(d);
+            data_returned = true;
+            result = d;
           }).on("status", (status) => {
             if (status.stage === "error")
               rej(status);
-            if (status.stage === "complete" && data_returned) {
-              app.destroy();
-            }
             if (status.stage === "complete") {
               status_complete = true;
+              app.destroy();
+              if (data_returned) {
+                res2(result);
+              }
             }
           });
         });
@@ -477,6 +491,7 @@ function api_factory(fetch_implementation) {
                 fire_event({
                   type: "status",
                   stage: "error",
+                  broken: true,
                   message: BROKEN_CONNECTION_MSG,
                   queue: true,
                   endpoint: _endpoint,
@@ -509,6 +524,14 @@ function api_factory(fetch_implementation) {
                 websocket.send(JSON.stringify({ ...payload, session_hash }));
               } else if (type === "complete") {
                 complete = status;
+              } else if (type === "log") {
+                fire_event({
+                  type: "log",
+                  log: data2.log,
+                  level: data2.level,
+                  endpoint: _endpoint,
+                  fn_index
+                });
               } else if (type === "generating") {
                 fire_event({
                   type: "status",
@@ -676,9 +699,8 @@ function api_factory(fetch_implementation) {
         if (blob) {
           const file_url = (await upload_files2(endpoint, [blob], token)).files[0];
           return { path, file_url, type };
-        } else {
-          return { path, base64: data2, type };
         }
+        return { path, base64: data2, type };
       })
     ).then((r) => {
       r.forEach(({ path, file_url, base64, type }) => {
@@ -704,17 +726,16 @@ const { post_data, upload_files, client, handle_blob } = api_factory(fetch);
 function transform_output(data, api_info, root_url, remote_url) {
   return data.map((d, i) => {
     var _a, _b, _c, _d;
-    if (((_b = (_a = api_info.returns) == null ? void 0 : _a[i]) == null ? void 0 : _b.component) === "File") {
+    if (((_b = (_a = api_info == null ? void 0 : api_info.returns) == null ? void 0 : _a[i]) == null ? void 0 : _b.component) === "File") {
       return normalise_file(d, root_url, remote_url);
-    } else if (((_d = (_c = api_info.returns) == null ? void 0 : _c[i]) == null ? void 0 : _d.component) === "Gallery") {
+    } else if (((_d = (_c = api_info == null ? void 0 : api_info.returns) == null ? void 0 : _c[i]) == null ? void 0 : _d.component) === "Gallery") {
       return d.map((img) => {
         return Array.isArray(img) ? [normalise_file(img[0], root_url, remote_url), img[1]] : [normalise_file(img, root_url, remote_url), null];
       });
-    } else if (typeof d === "object" && d.is_file) {
+    } else if (typeof d === "object" && (d == null ? void 0 : d.is_file)) {
       return normalise_file(d, root_url, remote_url);
-    } else {
-      return d;
     }
+    return d;
   });
 }
 function normalise_file(file, root, root_url) {
@@ -762,9 +783,8 @@ function get_type(type, component, serializer, signature_type) {
   } else if (serializer === "FileSerializable") {
     if ((type == null ? void 0 : type.type) === "array") {
       return signature_type === "parameter" ? "(Blob | File | Buffer)[]" : `{ name: string; data: string; size?: number; is_file?: boolean; orig_name?: string}[]`;
-    } else {
-      return signature_type === "parameter" ? "Blob | File | Buffer" : `{ name: string; data: string; size?: number; is_file?: boolean; orig_name?: string}`;
     }
+    return signature_type === "parameter" ? "Blob | File | Buffer" : `{ name: string; data: string; size?: number; is_file?: boolean; orig_name?: string}`;
   } else if (serializer === "GallerySerializable") {
     return signature_type === "parameter" ? "[(Blob | File | Buffer), (string | null)][]" : `[{ name: string; data: string; size?: number; is_file?: boolean; orig_name?: string}, (string | null))][]`;
   }
@@ -776,9 +796,8 @@ function get_description(type, serializer) {
     return "array of strings";
   } else if (serializer === "FileSerializable") {
     return "array of files or single file";
-  } else {
-    return type.description;
   }
+  return type.description;
 }
 function transform_api_info(api_info, config, api_map) {
   const new_data = {
@@ -872,10 +891,9 @@ async function walk_and_store_blobs(param, type = void 0, path = [], root = fals
         const buffer = await param.arrayBuffer();
         data = Buffer.from(buffer).toString("base64");
       }
-      return [{ path, data, type }];
-    } else {
-      return [{ path, blob: param, type }];
+      return [{ path, data, type, blob: false }];
     }
+    return [{ path, blob: param, type, data: false }];
   } else if (typeof param === "object") {
     let blob_refs = [];
     for (let key in param) {
@@ -894,9 +912,8 @@ async function walk_and_store_blobs(param, type = void 0, path = [], root = fals
       }
     }
     return blob_refs;
-  } else {
-    return [];
   }
+  return [];
 }
 function image_to_data_uri(blob) {
   return new Promise((resolve, _) => {
@@ -928,9 +945,8 @@ async function resolve_config(fetch_implementation, endpoint, token) {
       config.path = config.path ?? "";
       config.root = endpoint;
       return config;
-    } else {
-      throw new Error("Could not get config.");
     }
+    throw new Error("Could not get config.");
   }
   throw new Error("No config or app endpoint found");
 }
@@ -1055,6 +1071,8 @@ function handle_message(data, last_status) {
           success: data.success
         }
       };
+    case "log":
+      return { type: "log", data };
     case "process_generating":
       return {
         type: "generating",
@@ -1080,20 +1098,19 @@ function handle_message(data, last_status) {
             success: data.success
           }
         };
-      } else {
-        return {
-          type: "complete",
-          status: {
-            queue,
-            message: !data.success ? data.output.error : void 0,
-            stage: data.success ? "complete" : "error",
-            code: data.code,
-            progress_data: data.progress_data,
-            eta: data.output.average_duration
-          },
-          data: data.success ? data.output : null
-        };
       }
+      return {
+        type: "complete",
+        status: {
+          queue,
+          message: !data.success ? data.output.error : void 0,
+          stage: data.success ? "complete" : "error",
+          code: data.code,
+          progress_data: data.progress_data,
+          eta: data.output.average_duration
+        },
+        data: data.success ? data.output : null
+      };
     case "process_starts":
       return {
         type: "update",

package/dist/types.d.ts

@@ -20,7 +20,7 @@ export interface Config {
     path: string;
 }
 export interface Payload {
-    data: Array<unknown>;
+    data: unknown[];
     fn_index?: number;
     event_data?: unknown;
     time?: Date;
@@ -31,26 +31,31 @@ export interface PostResponse {
 }
 export interface UploadResponse {
     error?: string;
-    files?: Array<string>;
+    files?: string[];
 }
 export interface Status {
     queue: boolean;
     code?: string;
     success?: boolean;
     stage: "pending" | "error" | "complete" | "generating";
+    broken?: boolean;
     size?: number;
     position?: number;
     eta?: number;
     message?: string;
-    progress_data?: Array<{
+    progress_data?: {
         progress: number | null;
         index: number | null;
         length: number | null;
         unit: string | null;
         desc: string | null;
-    }>;
+    }[];
     time?: Date;
 }
+export interface LogMessage {
+    log: string;
+    level: "warning" | "info";
+}
 export interface SpaceStatusNormal {
     status: "sleeping" | "running" | "building" | "error" | "stopped";
     detail: "SLEEPING" | "RUNNING" | "RUNNING_BUILDING" | "BUILDING" | "NOT_FOUND";
@@ -67,10 +72,11 @@ export interface SpaceStatusError {
 export type SpaceStatus = SpaceStatusNormal | SpaceStatusError;
 export type status_callback_function = (a: Status) => void;
 export type SpaceStatusCallback = (a: SpaceStatus) => void;
-export type EventType = "data" | "status";
+export type EventType = "data" | "status" | "log";
 export interface EventMap {
     data: Payload;
     status: Status;
+    log: LogMessage;
 }
 export type Event<K extends EventType> = {
     [P in K]: EventMap[P] & {

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,MAAM;IACtB,aAAa,EAAE,OAAO,GAAG,SAAS,CAAC;IACnC,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,GAAG,EAAE,CAAC;IAClB,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,YAAY,EAAE,GAAG,EAAE,CAAC;IACpB,QAAQ,EAAE,OAAO,CAAC;IAClB,YAAY,EAAE,OAAO,CAAC;IACtB,MAAM,EAAE,GAAG,CAAC;IACZ,IAAI,EAAE,QAAQ,GAAG,WAAW,CAAC;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,OAAO;IACvB,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,IAAI,CAAC,EAAE,IAAI,CAAC;CACZ;AAED,MAAM,WAAW,YAAY;IAC5B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,CAAC,EAAE,MAAM,GAAG,GAAG,CAAC;CACjB;AACD,MAAM,WAAW,cAAc;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;CACtB;AAED,MAAM,WAAW,MAAM;IACtB,KAAK,EAAE,OAAO,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,EAAE,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;IACvD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,KAAK,CAAC;QACrB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QACxB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;QACrB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;QACpB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;KACpB,CAAC,CAAC;IACH,IAAI,CAAC,EAAE,IAAI,CAAC;CACZ;AAED,MAAM,WAAW,iBAAiB;IACjC,MAAM,EAAE,UAAU,GAAG,SAAS,GAAG,UAAU,GAAG,OAAO,GAAG,SAAS,CAAC;IAClE,MAAM,EACH,UAAU,GACV,SAAS,GACT,kBAAkB,GAClB,UAAU,GACV,WAAW,CAAC;IACf,WAAW,EAAE,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;IAC7D,OAAO,EAAE,MAAM,CAAC;CAChB;AACD,MAAM,WAAW,gBAAgB;IAChC,MAAM,EAAE,aAAa,GAAG,QAAQ,CAAC;IACjC,MAAM,EACH,aAAa,GACb,cAAc,GACd,aAAa,GACb,eAAe,GACf,QAAQ,CAAC;IACZ,WAAW,EAAE,OAAO,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,OAAO,CAAC;CAC7B;AACD,MAAM,MAAM,WAAW,GAAG,iBAAiB,GAAG,gBAAgB,CAAC;AAE/D,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;AAC3D,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,EAAE,WAAW,KAAK,IAAI,CAAC;AAE3D,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,QAAQ,CAAC;AAE1C,MAAM,WAAW,QAAQ;IACxB,IAAI,EAAE,OAAO,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,KAAK,CAAC,CAAC,SAAS,SAAS,IAAI;KACvC,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,GAAG;QAAE,IAAI,EAAE,CAAC,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE;CACvE,CAAC,CAAC,CAAC,CAAC;AACL,MAAM,MAAM,aAAa,CAAC,CAAC,SAAS,SAAS,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;AAC3E,MAAM,MAAM,WAAW,CAAC,CAAC,SAAS,SAAS,IAAI;KAC7C,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE;CAC7B,CAAC;AACF,MAAM,WAAW,QAAQ;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,IAAI,CAAC;IACZ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,MAAM;IACtB,aAAa,EAAE,OAAO,GAAG,SAAS,CAAC;IACnC,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,GAAG,EAAE,CAAC;IAClB,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,YAAY,EAAE,GAAG,EAAE,CAAC;IACpB,QAAQ,EAAE,OAAO,CAAC;IAClB,YAAY,EAAE,OAAO,CAAC;IACtB,MAAM,EAAE,GAAG,CAAC;IACZ,IAAI,EAAE,QAAQ,GAAG,WAAW,CAAC;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,OAAO;IACvB,IAAI,EAAE,OAAO,EAAE,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,IAAI,CAAC,EAAE,IAAI,CAAC;CACZ;AAED,MAAM,WAAW,YAAY;IAC5B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,CAAC,EAAE,MAAM,GAAG,GAAG,CAAC;CACjB;AACD,MAAM,WAAW,cAAc;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,MAAM;IACtB,KAAK,EAAE,OAAO,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,EAAE,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;IACvD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE;QACf,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QACxB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;QACrB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;QACpB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;KACpB,EAAE,CAAC;IACJ,IAAI,CAAC,EAAE,IAAI,CAAC;CACZ;AAED,MAAM,WAAW,UAAU;IAC1B,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,SAAS,GAAG,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,iBAAiB;IACjC,MAAM,EAAE,UAAU,GAAG,SAAS,GAAG,UAAU,GAAG,OAAO,GAAG,SAAS,CAAC;IAClE,MAAM,EACH,UAAU,GACV,SAAS,GACT,kBAAkB,GAClB,UAAU,GACV,WAAW,CAAC;IACf,WAAW,EAAE,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;IAC7D,OAAO,EAAE,MAAM,CAAC;CAChB;AACD,MAAM,WAAW,gBAAgB;IAChC,MAAM,EAAE,aAAa,GAAG,QAAQ,CAAC;IACjC,MAAM,EACH,aAAa,GACb,cAAc,GACd,aAAa,GACb,eAAe,GACf,QAAQ,CAAC;IACZ,WAAW,EAAE,OAAO,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,OAAO,CAAC;CAC7B;AACD,MAAM,MAAM,WAAW,GAAG,iBAAiB,GAAG,gBAAgB,CAAC;AAE/D,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;AAC3D,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,EAAE,WAAW,KAAK,IAAI,CAAC;AAE3D,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,QAAQ,GAAG,KAAK,CAAC;AAElD,MAAM,WAAW,QAAQ;IACxB,IAAI,EAAE,OAAO,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,UAAU,CAAC;CAChB;AAED,MAAM,MAAM,KAAK,CAAC,CAAC,SAAS,SAAS,IAAI;KACvC,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,GAAG;QAAE,IAAI,EAAE,CAAC,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE;CACvE,CAAC,CAAC,CAAC,CAAC;AACL,MAAM,MAAM,aAAa,CAAC,CAAC,SAAS,SAAS,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;AAC3E,MAAM,MAAM,WAAW,CAAC,CAAC,SAAS,SAAS,IAAI;KAC7C,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE;CAC7B,CAAC;AACF,MAAM,WAAW,QAAQ;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,IAAI,CAAC;IACZ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB"}

package/dist/utils.d.ts

@@ -14,8 +14,8 @@ export declare function process_endpoint(app_reference: string, token?: `hf_${st
 }>;
 export declare function map_names_to_ids(fns: Config["dependencies"]): Record<string, number>;
 export declare function discussions_enabled(space_id: string): Promise<boolean>;
-export declare function get_space_hardware(space_id: string, token: `hf_${string}`): Promise<any>;
-export declare function set_space_hardware(space_id: string, new_hardware: typeof hardware_types[number], token: `hf_${string}`): Promise<any>;
-export declare function set_space_timeout(space_id: string, timeout: number, token: `hf_${string}`): Promise<any>;
+export declare function get_space_hardware(space_id: string, token: `hf_${string}`): Promise<(typeof hardware_types)[number]>;
+export declare function set_space_hardware(space_id: string, new_hardware: (typeof hardware_types)[number], token: `hf_${string}`): Promise<(typeof hardware_types)[number]>;
+export declare function set_space_timeout(space_id: string, timeout: number, token: `hf_${string}`): Promise<number>;
 export declare const hardware_types: readonly ["cpu-basic", "cpu-upgrade", "t4-small", "t4-medium", "a10g-small", "a10g-large", "a100-large"];
 //# sourceMappingURL=utils.d.ts.map

package/dist/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAEzC,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG;IACrD,WAAW,EAAE,IAAI,GAAG,KAAK,CAAC;IAC1B,aAAa,EAAE,OAAO,GAAG,QAAQ,CAAC;IAClC,IAAI,EAAE,MAAM,CAAC;CACb,CAyBA;AAED,eAAO,MAAM,aAAa,QAAqB,CAAC;AAChD,eAAO,MAAM,eAAe,QAAwB,CAAC;AACrD,wBAAsB,gBAAgB,CACrC,aAAa,EAAE,MAAM,EACrB,KAAK,CAAC,EAAE,MAAM,MAAM,EAAE,GACpB,OAAO,CAAC;IACV,QAAQ,EAAE,MAAM,GAAG,KAAK,CAAC;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,IAAI,GAAG,KAAK,CAAC;IAC1B,aAAa,EAAE,OAAO,GAAG,QAAQ,CAAC;CAClC,CAAC,CA4CD;AAED,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,MAAM,CAAC,cAAc,CAAC,0BAQ3D;AAID,wBAAsB,mBAAmB,CAAC,QAAQ,EAAE,MAAM,oBAezD;AAED,wBAAsB,kBAAkB,CACvC,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,MAAM,EAAE,gBAsBrB;AAED,wBAAsB,kBAAkB,CACvC,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,OAAO,cAAc,CAAC,MAAM,CAAC,EAC3C,KAAK,EAAE,MAAM,MAAM,EAAE,gBAwBrB;AAED,wBAAsB,iBAAiB,CACtC,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,MAAM,EAAE,gBAwBrB;AAED,eAAO,MAAM,cAAc,0GAQjB,CAAC"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAEzC,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG;IACrD,WAAW,EAAE,IAAI,GAAG,KAAK,CAAC;IAC1B,aAAa,EAAE,OAAO,GAAG,QAAQ,CAAC;IAClC,IAAI,EAAE,MAAM,CAAC;CACb,CAwBA;AAED,eAAO,MAAM,aAAa,QAAqB,CAAC;AAChD,eAAO,MAAM,eAAe,QAAwB,CAAC;AACrD,wBAAsB,gBAAgB,CACrC,aAAa,EAAE,MAAM,EACrB,KAAK,CAAC,EAAE,MAAM,MAAM,EAAE,GACpB,OAAO,CAAC;IACV,QAAQ,EAAE,MAAM,GAAG,KAAK,CAAC;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,IAAI,GAAG,KAAK,CAAC;IAC1B,aAAa,EAAE,OAAO,GAAG,QAAQ,CAAC;CAClC,CAAC,CA4CD;AAED,wBAAgB,gBAAgB,CAC/B,GAAG,EAAE,MAAM,CAAC,cAAc,CAAC,GACzB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAQxB;AAID,wBAAsB,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAe5E;AAED,wBAAsB,kBAAkB,CACvC,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,MAAM,EAAE,GACnB,OAAO,CAAC,CAAC,OAAO,cAAc,CAAC,CAAC,MAAM,CAAC,CAAC,CAqB1C;AAED,wBAAsB,kBAAkB,CACvC,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,CAAC,OAAO,cAAc,CAAC,CAAC,MAAM,CAAC,EAC7C,KAAK,EAAE,MAAM,MAAM,EAAE,GACnB,OAAO,CAAC,CAAC,OAAO,cAAc,CAAC,CAAC,MAAM,CAAC,CAAC,CAuB1C;AAED,wBAAsB,iBAAiB,CACtC,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,MAAM,EAAE,GACnB,OAAO,CAAC,MAAM,CAAC,CAuBjB;AAED,eAAO,MAAM,cAAc,0GAQjB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gradio/client",
-  "version": "0.1.4",
+  "version": "0.2.1",
   "description": "Gradio API client",
   "type": "module",
   "main": "dist/index.js",
@@ -19,11 +19,12 @@
   },
   "devDependencies": {
     "@types/ws": "^8.5.4",
-    "esbuild": "^0.18.0"
+    "esbuild": "^0.19.0"
   },
   "engines": {
     "node": ">=18.0.0"
   },
+  "main_changeset": true,
   "scripts": {
     "bundle": "vite build --ssr",
     "generate_types": "tsc",

package/src/client.ts

@@ -66,10 +66,10 @@ export async function duplicate(
 		hf_token: `hf_${string}`;
 		private?: boolean;
 		status_callback: SpaceStatusCallback;
-		hardware?: typeof hardware_types[number];
+		hardware?: (typeof hardware_types)[number];
 		timeout?: number;
 	}
-) {
+): Promise<client_return> {
 	const { hf_token, private: _private, hardware, timeout } = options;
 
 	if (hardware && !hardware_types.includes(hardware)) {
@@ -115,38 +115,57 @@ export async function duplicate(
 
 		if (response.status === 409) {
 			return client(`${user}/${space_name}`, options);
-		} else {
-			const duplicated_space = await response.json();
+		}
+		const duplicated_space = await response.json();
 
-			let original_hardware;
+		let original_hardware;
 
-			if (!hardware) {
-				original_hardware = await get_space_hardware(app_reference, hf_token);
-			}
+		if (!hardware) {
+			original_hardware = await get_space_hardware(app_reference, hf_token);
+		}
 
-			const requested_hardware = hardware || original_hardware || "cpu-basic";
-			await set_space_hardware(
-				`${user}/${space_name}`,
-				requested_hardware,
-				hf_token
-			);
+		const requested_hardware = hardware || original_hardware || "cpu-basic";
+		await set_space_hardware(
+			`${user}/${space_name}`,
+			requested_hardware,
+			hf_token
+		);
 
-			await set_space_timeout(
-				`${user}/${space_name}`,
-				timeout || 300,
-				hf_token
-			);
-			return client(duplicated_space.url, options);
-		}
+		await set_space_timeout(`${user}/${space_name}`, timeout || 300, hf_token);
+		return client(duplicated_space.url, options);
 	} catch (e: any) {
 		throw new Error(e);
 	}
 }
 
-/**
- * We need to inject a customized fetch implementation for the Wasm version.
- */
-export function api_factory(fetch_implementation: typeof fetch) {
+interface Client {
+	post_data: (
+		url: string,
+		body: unknown,
+		token?: `hf_${string}`
+	) => Promise<[PostResponse, number]>;
+	upload_files: (
+		root: string,
+		files: File[],
+		token?: `hf_${string}`
+	) => Promise<UploadResponse>;
+	client: (
+		app_reference: string,
+		options: {
+			hf_token?: `hf_${string}`;
+			status_callback?: SpaceStatusCallback;
+			normalise_files?: boolean;
+		}
+	) => Promise<client_return>;
+	handle_blob: (
+		endpoint: string,
+		data: unknown[],
+		api_info: ApiInfo<JsApiData>,
+		token?: `hf_${string}`
+	) => Promise<unknown[]>;
+}
+
+export function api_factory(fetch_implementation: typeof fetch): Client {
 	return { post_data, upload_files, client, handle_blob };
 
 	async function post_data(
@@ -176,7 +195,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 
 	async function upload_files(
 		root: string,
-		files: Array<File>,
+		files: (Blob | File)[],
 		token?: `hf_${string}`
 	): Promise<UploadResponse> {
 		const headers: {
@@ -185,22 +204,27 @@ export function api_factory(fetch_implementation: typeof fetch) {
 		if (token) {
 			headers.Authorization = `Bearer ${token}`;
 		}
-
-		const formData = new FormData();
-		files.forEach((file) => {
-			formData.append("files", file);
-		});
-		try {
-			var response = await fetch_implementation(`${root}/upload`, {
-				method: "POST",
-				body: formData,
-				headers
+		const chunkSize = 1000;
+		const uploadResponses = [];
+		for (let i = 0; i < files.length; i += chunkSize) {
+			const chunk = files.slice(i, i + chunkSize);
+			const formData = new FormData();
+			chunk.forEach((file) => {
+				formData.append("files", file);
 			});
-		} catch (e) {
-			return { error: BROKEN_CONNECTION_MSG };
+			try {
+				var response = await fetch_implementation(`${root}/upload`, {
+					method: "POST",
+					body: formData,
+					headers
+				});
+			} catch (e) {
+				return { error: BROKEN_CONNECTION_MSG };
+			}
+			const output: UploadResponse["files"] = await response.json();
+			uploadResponses.push(...output);
 		}
-		const output: UploadResponse["files"] = await response.json();
-		return { files: output };
+		return { files: uploadResponses };
 	}
 
 	async function client(
@@ -242,7 +266,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 				jwt = await get_jwt(space_id, hf_token);
 			}
 
-			async function config_success(_config: Config) {
+			async function config_success(_config: Config): Promise<client_return> {
 				config = _config;
 				api_map = map_names_to_ids(_config?.dependencies || []);
 				try {
@@ -257,7 +281,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 				};
 			}
 			let api: ApiInfo<JsApiData>;
-			async function handle_space_sucess(status: SpaceStatus) {
+			async function handle_space_sucess(status: SpaceStatus): Promise<void> {
 				if (status_callback) status_callback(status);
 				if (status.status === "running")
 					try {
@@ -310,37 +334,50 @@ export function api_factory(fetch_implementation: typeof fetch) {
 				}
 			}
 
-			/**
-			 * Run a prediction.
-			 * @param endpoint - The prediction endpoint to use.
-			 * @param status_callback - A function that is called with the current status of the prediction immediately and every time it updates.
-			 * @return Returns the data for the prediction or an error message.
-			 */
 			function predict(
 				endpoint: string,
 				data: unknown[],
 				event_data?: unknown
-			) {
+			): Promise<unknown> {
 				let data_returned = false;
 				let status_complete = false;
+				let dependency;
+				if (typeof endpoint === "number") {
+					dependency = config.dependencies[endpoint];
+				} else {
+					const trimmed_endpoint = endpoint.replace(/^\//, "");
+					dependency = config.dependencies[api_map[trimmed_endpoint]];
+				}
+
+				if (dependency.types.continuous) {
+					throw new Error(
+						"Cannot call predict on this function as it may run forever. Use submit instead"
+					);
+				}
+
 				return new Promise((res, rej) => {
 					const app = submit(endpoint, data, event_data);
+					let result;
 
 					app
 						.on("data", (d) => {
-							data_returned = true;
+							// if complete message comes before data, resolve here
 							if (status_complete) {
 								app.destroy();
+								res(d);
 							}
-							res(d);
+							data_returned = true;
+							result = d;
 						})
 						.on("status", (status) => {
 							if (status.stage === "error") rej(status);
-							if (status.stage === "complete" && data_returned) {
-								app.destroy();
-							}
 							if (status.stage === "complete") {
 								status_complete = true;
+								app.destroy();
+								// if complete message comes after data, resolve here
+								if (data_returned) {
+									res(result);
+								}
 							}
 						});
 				});
@@ -478,6 +515,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 								fire_event({
 									type: "status",
 									stage: "error",
+									broken: true,
 									message: BROKEN_CONNECTION_MSG,
 									queue: true,
 									endpoint: _endpoint,
@@ -513,6 +551,14 @@ export function api_factory(fetch_implementation: typeof fetch) {
 								websocket.send(JSON.stringify({ ...payload, session_hash }));
 							} else if (type === "complete") {
 								complete = status;
+							} else if (type === "log") {
+								fire_event({
+									type: "log",
+									log: data.log,
+									level: data.level,
+									endpoint: _endpoint,
+									fn_index
+								});
 							} else if (type === "generating") {
 								fire_event({
 									type: "status",
@@ -565,7 +611,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 					}
 				});
 
-				function fire_event<K extends EventType>(event: Event<K>) {
+				function fire_event<K extends EventType>(event: Event<K>): void {
 					const narrowed_listener_map: ListenerMap<K> = listener_map;
 					const listeners = narrowed_listener_map[event.type] || [];
 					listeners?.forEach((l) => l(event));
@@ -574,7 +620,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 				function on<K extends EventType>(
 					eventType: K,
 					listener: EventListener<K>
-				) {
+				): SubmitReturn {
 					const narrowed_listener_map: ListenerMap<K> = listener_map;
 					const listeners = narrowed_listener_map[eventType] || [];
 					narrowed_listener_map[eventType] = listeners;
@@ -586,7 +632,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 				function off<K extends EventType>(
 					eventType: K,
 					listener: EventListener<K>
-				) {
+				): SubmitReturn {
 					const narrowed_listener_map: ListenerMap<K> = listener_map;
 					let listeners = narrowed_listener_map[eventType] || [];
 					listeners = listeners?.filter((l) => l !== listener);
@@ -595,7 +641,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 					return { on, off, cancel, destroy };
 				}
 
-				async function cancel() {
+				async function cancel(): Promise<void> {
 					const _status: Status = {
 						stage: "complete",
 						queue: false,
@@ -633,7 +679,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 					}
 				}
 
-				function destroy() {
+				function destroy(): void {
 					for (const event_type in listener_map) {
 						listener_map[event_type as "data" | "status"].forEach((fn) => {
 							off(event_type as "data" | "status", fn);
@@ -706,7 +752,7 @@ export function api_factory(fetch_implementation: typeof fetch) {
 	async function handle_blob(
 		endpoint: string,
 		data: unknown[],
-		api_info,
+		api_info: ApiInfo<JsApiData>,
 		token?: `hf_${string}`
 	): Promise<unknown[]> {
 		const blob_refs = await walk_and_store_blobs(
@@ -723,9 +769,8 @@ export function api_factory(fetch_implementation: typeof fetch) {
 					const file_url = (await upload_files(endpoint, [blob], token))
 						.files[0];
 					return { path, file_url, type };
-				} else {
-					return { path, base64: data, type };
 				}
+				return { path, base64: data, type };
 			})
 		).then((r) => {
 			r.forEach(({ path, file_url, base64, type }) => {
@@ -759,27 +804,26 @@ function transform_output(
 	remote_url?: string
 ): unknown[] {
 	return data.map((d, i) => {
-		if (api_info.returns?.[i]?.component === "File") {
+		if (api_info?.returns?.[i]?.component === "File") {
 			return normalise_file(d, root_url, remote_url);
-		} else if (api_info.returns?.[i]?.component === "Gallery") {
+		} else if (api_info?.returns?.[i]?.component === "Gallery") {
 			return d.map((img) => {
 				return Array.isArray(img)
 					? [normalise_file(img[0], root_url, remote_url), img[1]]
 					: [normalise_file(img, root_url, remote_url), null];
 			});
-		} else if (typeof d === "object" && d.is_file) {
+		} else if (typeof d === "object" && d?.is_file) {
 			return normalise_file(d, root_url, remote_url);
-		} else {
-			return d;
 		}
+		return d;
 	});
 }
 
 function normalise_file(
-	file: Array<FileData>,
+	file: FileData[],
 	root: string,
 	root_url: string | null
-): Array<FileData>;
+): FileData[];
 function normalise_file(
 	file: FileData | string,
 	root: string,
@@ -790,11 +834,7 @@ function normalise_file(
 	root: string,
 	root_url: string | null
 ): null;
-function normalise_file(
-	file,
-	root,
-	root_url
-): Array<FileData> | FileData | null {
+function normalise_file(file, root, root_url): FileData[] | FileData | null {
 	if (file == null) return null;
 	if (typeof file === "string") {
 		return {
@@ -802,7 +842,7 @@ function normalise_file(
 			data: file
 		};
 	} else if (Array.isArray(file)) {
-		const normalized_file: Array<FileData | null> = [];
+		const normalized_file: (FileData | null)[] = [];
 
 		for (const x of file) {
 			if (x === null) {
@@ -812,7 +852,7 @@ function normalise_file(
 			}
 		}
 
-		return normalized_file as Array<FileData>;
+		return normalized_file as FileData[];
 	} else if (file.is_file) {
 		if (!root_url) {
 			file.data = root + "/file=" + file.name;
@@ -858,7 +898,7 @@ function get_type(
 	component: string,
 	serializer: string,
 	signature_type: "return" | "parameter"
-) {
+): string {
 	switch (type.type) {
 		case "string":
 			return "string";
@@ -882,11 +922,10 @@ function get_type(
 			return signature_type === "parameter"
 				? "(Blob | File | Buffer)[]"
 				: `{ name: string; data: string; size?: number; is_file?: boolean; orig_name?: string}[]`;
-		} else {
-			return signature_type === "parameter"
-				? "Blob | File | Buffer"
-				: `{ name: string; data: string; size?: number; is_file?: boolean; orig_name?: string}`;
 		}
+		return signature_type === "parameter"
+			? "Blob | File | Buffer"
+			: `{ name: string; data: string; size?: number; is_file?: boolean; orig_name?: string}`;
 	} else if (serializer === "GallerySerializable") {
 		return signature_type === "parameter"
 			? "[(Blob | File | Buffer), (string | null)][]"
@@ -897,16 +936,15 @@ function get_type(
 function get_description(
 	type: { type: any; description: string },
 	serializer: string
-) {
+): string {
 	if (serializer === "GallerySerializable") {
 		return "array of [file, label] tuples";
 	} else if (serializer === "ListStringSerializable") {
 		return "array of strings";
 	} else if (serializer === "FileSerializable") {
 		return "array of files or single file";
-	} else {
-		return type.description;
 	}
+	return type.description;
 }
 
 function transform_api_info(
@@ -974,7 +1012,7 @@ async function get_jwt(
 	}
 }
 
-function update_object(object, newValue, stack) {
+function update_object(object, newValue, stack): void {
 	while (stack.length > 1) {
 		object = object[stack.shift()];
 	}
@@ -988,7 +1026,14 @@ export async function walk_and_store_blobs(
 	path = [],
 	root = false,
 	api_info = undefined
-) {
+): Promise<
+	{
+		path: string[];
+		data: string | false;
+		type: string;
+		blob: Blob | false;
+	}[]
+> {
 	if (Array.isArray(param)) {
 		let blob_refs = [];
 
@@ -1035,10 +1080,9 @@ export async function walk_and_store_blobs(
 				data = Buffer.from(buffer).toString("base64");
 			}
 
-			return [{ path, data, type }];
-		} else {
-			return [{ path: path, blob: param, type }];
+			return [{ path, data, type, blob: false }];
 		}
+		return [{ path: path, blob: param, type, data: false }];
 	} else if (typeof param === "object") {
 		let blob_refs = [];
 		for (let key in param) {
@@ -1057,12 +1101,11 @@ export async function walk_and_store_blobs(
 			}
 		}
 		return blob_refs;
-	} else {
-		return [];
 	}
+	return [];
 }
 
-function image_to_data_uri(blob: Blob) {
+function image_to_data_uri(blob: Blob): Promise<string | ArrayBuffer> {
 	return new Promise((resolve, _) => {
 		const reader = new FileReader();
 		reader.onloadend = () => resolve(reader.result);
@@ -1070,7 +1113,7 @@ function image_to_data_uri(blob: Blob) {
 	});
 }
 
-function skip_queue(id: number, config: Config) {
+function skip_queue(id: number, config: Config): boolean {
 	return (
 		!(config?.dependencies?.[id]?.queue === null
 			? config.enable_queue
@@ -1106,9 +1149,8 @@ async function resolve_config(
 			config.path = config.path ?? "";
 			config.root = endpoint;
 			return config;
-		} else {
-			throw new Error("Could not get config.");
 		}
+		throw new Error("Could not get config.");
 	}
 
 	throw new Error("No config or app endpoint found");
@@ -1118,7 +1160,7 @@ async function check_space_status(
 	id: string,
 	type: "subdomain" | "space_name",
 	status_callback: SpaceStatusCallback
-) {
+): Promise<void> {
 	let endpoint =
 		type === "subdomain"
 			? `https://huggingface.co/api/spaces/by-subdomain/${id}`
@@ -1211,7 +1253,7 @@ function handle_message(
 	data: any,
 	last_status: Status["stage"]
 ): {
-	type: "hash" | "data" | "update" | "complete" | "generating" | "none";
+	type: "hash" | "data" | "update" | "complete" | "generating" | "log" | "none";
 	data?: any;
 	status?: Status;
 } {
@@ -1256,6 +1298,8 @@ function handle_message(
 					success: data.success
 				}
 			};
+		case "log":
+			return { type: "log", data: data };
 		case "process_generating":
 			return {
 				type: "generating",
@@ -1281,20 +1325,19 @@ function handle_message(
 						success: data.success
 					}
 				};
-			} else {
-				return {
-					type: "complete",
-					status: {
-						queue,
-						message: !data.success ? data.output.error : undefined,
-						stage: data.success ? "complete" : "error",
-						code: data.code,
-						progress_data: data.progress_data,
-						eta: data.output.average_duration
-					},
-					data: data.success ? data.output : null
-				};
 			}
+			return {
+				type: "complete",
+				status: {
+					queue,
+					message: !data.success ? data.output.error : undefined,
+					stage: data.success ? "complete" : "error",
+					code: data.code,
+					progress_data: data.progress_data,
+					eta: data.output.average_duration
+				},
+				data: data.success ? data.output : null
+			};
 
 		case "process_starts":
 			return {

package/src/globals.d.ts

@@ -1,11 +1,9 @@
 declare global {
 	interface Window {
 		__gradio_mode__: "app" | "website";
-		launchGradio: Function;
-		launchGradioFromSpaces: Function;
 		gradio_config: Config;
-		scoped_css_attach: (link: HTMLLinkElement) => void;
 		__is_colab__: boolean;
+		__gradio_space__: string | null;
 	}
 }
 

package/src/types.ts

@@ -21,7 +21,7 @@ export interface Config {
 }
 
 export interface Payload {
-	data: Array<unknown>;
+	data: unknown[];
 	fn_index?: number;
 	event_data?: unknown;
 	time?: Date;
@@ -33,7 +33,7 @@ export interface PostResponse {
 }
 export interface UploadResponse {
 	error?: string;
-	files?: Array<string>;
+	files?: string[];
 }
 
 export interface Status {
@@ -41,20 +41,26 @@ export interface Status {
 	code?: string;
 	success?: boolean;
 	stage: "pending" | "error" | "complete" | "generating";
+	broken?: boolean;
 	size?: number;
 	position?: number;
 	eta?: number;
 	message?: string;
-	progress_data?: Array<{
+	progress_data?: {
 		progress: number | null;
 		index: number | null;
 		length: number | null;
 		unit: string | null;
 		desc: string | null;
-	}>;
+	}[];
 	time?: Date;
 }
 
+export interface LogMessage {
+	log: string;
+	level: "warning" | "info";
+}
+
 export interface SpaceStatusNormal {
 	status: "sleeping" | "running" | "building" | "error" | "stopped";
 	detail:
@@ -83,11 +89,12 @@ export type SpaceStatus = SpaceStatusNormal | SpaceStatusError;
 export type status_callback_function = (a: Status) => void;
 export type SpaceStatusCallback = (a: SpaceStatus) => void;
 
-export type EventType = "data" | "status";
+export type EventType = "data" | "status" | "log";
 
 export interface EventMap {
 	data: Payload;
 	status: Status;
+	log: LogMessage;
 }
 
 export type Event<K extends EventType> = {

package/src/utils.ts

@@ -14,13 +14,12 @@ export function determine_protocol(endpoint: string): {
 				host: host,
 				http_protocol: protocol as "http:" | "https:"
 			};
-		} else {
-			return {
-				ws_protocol: protocol === "https:" ? "wss" : "ws",
-				http_protocol: protocol as "http:" | "https:",
-				host
-			};
 		}
+		return {
+			ws_protocol: protocol === "https:" ? "wss" : "ws",
+			http_protocol: protocol as "http:" | "https:",
+			host
+		};
 	}
 
 	// default to secure if no protocol is provided
@@ -87,7 +86,9 @@ export async function process_endpoint(
 	};
 }
 
-export function map_names_to_ids(fns: Config["dependencies"]) {
+export function map_names_to_ids(
+	fns: Config["dependencies"]
+): Record<string, number> {
 	let apis: Record<string, number> = {};
 
 	fns.forEach(({ api_name }, i) => {
@@ -99,7 +100,7 @@ export function map_names_to_ids(fns: Config["dependencies"]) {
 
 const RE_DISABLED_DISCUSSION =
 	/^(?=[^]*\b[dD]iscussions{0,1}\b)(?=[^]*\b[dD]isabled\b)[^]*$/;
-export async function discussions_enabled(space_id: string) {
+export async function discussions_enabled(space_id: string): Promise<boolean> {
 	try {
 		const r = await fetch(
 			`https://huggingface.co/api/spaces/${space_id}/discussions`,
@@ -110,7 +111,7 @@ export async function discussions_enabled(space_id: string) {
 		const error = r.headers.get("x-error-message");
 
 		if (error && RE_DISABLED_DISCUSSION.test(error)) return false;
-		else return true;
+		return true;
 	} catch (e) {
 		return false;
 	}
@@ -119,7 +120,7 @@ export async function discussions_enabled(space_id: string) {
 export async function get_space_hardware(
 	space_id: string,
 	token: `hf_${string}`
-) {
+): Promise<(typeof hardware_types)[number]> {
 	const headers: { Authorization?: string } = {};
 	if (token) {
 		headers.Authorization = `Bearer ${token}`;
@@ -144,9 +145,9 @@ export async function get_space_hardware(
 
 export async function set_space_hardware(
 	space_id: string,
-	new_hardware: typeof hardware_types[number],
+	new_hardware: (typeof hardware_types)[number],
 	token: `hf_${string}`
-) {
+): Promise<(typeof hardware_types)[number]> {
 	const headers: { Authorization?: string } = {};
 	if (token) {
 		headers.Authorization = `Bearer ${token}`;
@@ -175,7 +176,7 @@ export async function set_space_timeout(
 	space_id: string,
 	timeout: number,
 	token: `hf_${string}`
-) {
+): Promise<number> {
 	const headers: { Authorization?: string } = {};
 	if (token) {
 		headers.Authorization = `Bearer ${token}`;