@gradio/client 0.1.3 → 0.2.1

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;
 	}
 }
 
@@ -23,7 +21,7 @@ export interface Config {
 	theme: string;
 	title: string;
 	version: string;
-	is_space: boolean;
+	space_id: string | null;
 	is_colab: boolean;
 	show_api: boolean;
 	stylesheets: string[];

package/src/index.ts

@@ -1,2 +1,8 @@
-export { client, post_data, upload_files, duplicate } from "./client.js";
+export {
+	client,
+	post_data,
+	upload_files,
+	duplicate,
+	api_factory
+} from "./client.js";
 export type { SpaceStatus } from "./types.js";

package/src/types.ts

@@ -13,7 +13,7 @@ export interface Config {
 	theme: string;
 	title: string;
 	version: string;
-	is_space: boolean;
+	space_id: string | null;
 	is_colab: boolean;
 	show_api: boolean;
 	stylesheets: string[];
@@ -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}`;