@gradio/client 0.19.3 → 0.20.0

package/src/utils/predict.ts

@@ -15,10 +15,12 @@ export async function predict(
 	}
 
 	if (typeof endpoint === "number") {
-		dependency = this.config.dependencies[endpoint];
+		dependency = this.config.dependencies.find((dep) => dep.id == endpoint)!;
 	} else {
 		const trimmed_endpoint = endpoint.replace(/^\//, "");
-		dependency = this.config.dependencies[this.api_map[trimmed_endpoint]];
+		dependency = this.config.dependencies.find(
+			(dep) => dep.id == this.api_map[trimmed_endpoint]
+		)!;
 	}
 
 	if (dependency?.types.continuous) {

package/src/utils/submit.ts

@@ -162,6 +162,35 @@ export function submit(
 			}
 		}
 
+		const resolve_heartbeat = async (config: Config): Promise<void> => {
+			await this._resolve_hearbeat(config);
+		};
+
+		async function handle_render_config(render_config: any): Promise<void> {
+			if (!config) return;
+			let render_id: number = render_config.render_id;
+			config.components = [
+				...config.components.filter((c) => c.props.rendered_in !== render_id),
+				...render_config.components
+			];
+			config.dependencies = [
+				...config.dependencies.filter((d) => d.rendered_in !== render_id),
+				...render_config.dependencies
+			];
+			const any_state = config.components.some((c) => c.type === "state");
+			const any_unload = config.dependencies.some((d) =>
+				d.targets.some((t) => t[1] === "unload")
+			);
+			config.connect_heartbeat = any_state || any_unload;
+			await resolve_heartbeat(config);
+			fire_event({
+				type: "render",
+				data: render_config,
+				endpoint: _endpoint,
+				fn_index
+			});
+		}
+
 		this.handle_blob(config.root, resolved_data, endpoint_info).then(
 			async (_payload) => {
 				payload = {
@@ -201,6 +230,9 @@ export function submit(
 									event_data,
 									trigger_id
 								});
+								if (output.render_config) {
+									handle_render_config(output.render_config);
+								}
 
 								fire_event({
 									type: "status",
@@ -496,8 +528,12 @@ export function submit(
 					const origin = hostname.includes(".dev.")
 						? `https://moon-${hostname.split(".")[1]}.${hfhubdev}`
 						: `https://huggingface.co`;
+
+					const is_iframe =
+						typeof window !== "undefined" && window.parent != window;
+					const is_zerogpu_space = dependency.zerogpu && config.space_id;
 					const zerogpu_auth_promise =
-						dependency.zerogpu && window.parent != window && config.space_id
+						is_iframe && is_zerogpu_space
 							? post_message<Headers>("zerogpu-headers", origin)
 							: Promise.resolve(null);
 					const post_data_promise = zerogpu_auth_promise.then((headers) => {
@@ -602,12 +638,7 @@ export function submit(
 											fn_index
 										});
 										if (data.render_config) {
-											fire_event({
-												type: "render",
-												data: data.render_config,
-												endpoint: _endpoint,
-												fn_index
-											});
+											await handle_render_config(data.render_config);
 										}
 
 										if (complete) {
@@ -694,13 +725,15 @@ function get_endpoint_info(
 	if (typeof endpoint === "number") {
 		fn_index = endpoint;
 		endpoint_info = api_info.unnamed_endpoints[fn_index];
-		dependency = config.dependencies[endpoint];
+		dependency = config.dependencies.find((dep) => dep.id == endpoint)!;
 	} else {
 		const trimmed_endpoint = endpoint.replace(/^\//, "");
 
 		fn_index = api_map[trimmed_endpoint];
 		endpoint_info = api_info.named_endpoints[endpoint.trim()];
-		dependency = config.dependencies[api_map[trimmed_endpoint]];
+		dependency = config.dependencies.find(
+			(dep) => dep.id == api_map[trimmed_endpoint]
+		)!;
 	}
 
 	if (typeof fn_index !== "number") {

package/src/utils/upload_files.ts

@@ -33,7 +33,8 @@ export async function upload_files(
 			response = await this.fetch(upload_url, {
 				method: "POST",
 				body: formData,
-				headers
+				headers,
+				credentials: "include"
 			});
 		} catch (e) {
 			throw new Error(BROKEN_CONNECTION_MSG + (e as Error).message);

package/src/utils/view_api.ts

@@ -3,7 +3,7 @@ import semiver from "semiver";
 import { API_INFO_URL, BROKEN_CONNECTION_MSG } from "../constants";
 import { Client } from "../client";
 import { SPACE_FETCHER_URL } from "../constants";
-import { transform_api_info } from "../helpers/api_info";
+import { join_urls, transform_api_info } from "../helpers/api_info";
 
 export async function view_api(this: Client): Promise<any> {
 	if (this.api_info) return this.api_info;
@@ -34,11 +34,14 @@ export async function view_api(this: Client): Promise<any> {
 					serialize: false,
 					config: JSON.stringify(config)
 				}),
-				headers
+				headers,
+				credentials: "include"
 			});
 		} else {
-			response = await this.fetch(`${config?.root}/${API_INFO_URL}`, {
-				headers
+			const url = join_urls(config.root, API_INFO_URL);
+			response = await this.fetch(url, {
+				headers,
+				credentials: "include"
 			});
 		}