prodia.ts: inpainting + sdxl

prodialabs · Oct 10, 2023 · 52926ad · 52926ad
1 parent 0cafc44
commit 52926ad
Showing 1 changed file with 61 additions and 14 deletions.
diff --git a/prodia.ts b/prodia.ts
@@ -30,8 +30,9 @@ export type ProdiaGenerateRequest = {
 	aspect_ratio?: "square" | "portrait" | "landscape";
 };
 
-export type ProdiaTransformRequest = {
-	imageUrl: string;
+type ImageInput = { imageUrl: string } | { imageData: string };
+
+export type ProdiaTransformRequest = ImageInput & {
 	prompt: string;
 	model?: string;
 	denoising_strength?: number;
@@ -43,8 +44,7 @@ export type ProdiaTransformRequest = {
 	sampler?: string;
 };
 
-export type ProdiaControlnetRequest = {
-	imageUrl: string;
+export type ProdiaControlnetRequest = ImageInput & {
 	controlnet_model: string;
 	controlnet_module?: string;
 	threshold_a?: number;
@@ -61,6 +61,36 @@ export type ProdiaControlnetRequest = {
 	height?: number;
 };
 
+type MaskInput = { maskUrl: string } | { maskData: string };
+
+export type ProdiaInpaintingRequest = ImageInput &
+	MaskInput & {
+		prompt: string;
+		model?: string;
+		denoising_strength?: number;
+		negative_prompt?: string;
+		steps?: number;
+		cfg_scale?: number;
+		seed?: number;
+		upscale?: boolean;
+		mask_blur: number;
+		inpainting_fill: number;
+		inpainting_mask_invert: number;
+		inpainting_full_res: string;
+		sampler?: string;
+	};
+
+export type ProdiaXlGenerateRequest = {
+	prompt: string;
+	model?: string;
+	negative_prompt?: string;
+	steps?: number;
+	cfg_scale?: number;
+	seed?: number;
+	upscale?: boolean;
+	sampler?: string;
+};
+
 /* Constructor Definions */
 
 export type Prodia = ReturnType<typeof createProdia>;
@@ -74,17 +104,17 @@ export const createProdia = ({ apiKey, base: _base }: CreateProdiaOptions) => {
 	const base = _base || "https://api.prodia.com/v1";
 
 	const headers = {
-		"X-Prodia-Key": apiKey,
+		"X-Prodia-Key": apiKey
 	};
 
 	const generate = async (params: ProdiaGenerateRequest) => {
 		const response = await fetch(`${base}/sd/generate`, {
 			method: "POST",
 			headers: {
 				...headers,
-				"Content-Type": "application/json",
+				"Content-Type": "application/json"
 			},
-			body: JSON.stringify(params),
+			body: JSON.stringify(params)
 		});
 
 		if (response.status !== 200) {
@@ -99,9 +129,9 @@ export const createProdia = ({ apiKey, base: _base }: CreateProdiaOptions) => {
 			method: "POST",
 			headers: {
 				...headers,
-				"Content-Type": "application/json",
+				"Content-Type": "application/json"
 			},
-			body: JSON.stringify(params),
+			body: JSON.stringify(params)
 		});
 
 		if (response.status !== 200) {
@@ -116,9 +146,26 @@ export const createProdia = ({ apiKey, base: _base }: CreateProdiaOptions) => {
 			method: "POST",
 			headers: {
 				...headers,
-				"Content-Type": "application/json",
+				"Content-Type": "application/json"
+			},
+			body: JSON.stringify(params)
+		});
+
+		if (response.status !== 200) {
+			throw new Error(`Bad Prodia Response: ${response.status}`);
+		}
+
+		return (await response.json()) as ProdiaJobQueued;
+	};
+
+	const inpainting = async (params: ProdiaInpaintingRequest) => {
+		const response = await fetch(`${base}/sd/inpainting`, {
+			method: "POST",
+			headers: {
+				...headers,
+				"Content-Type": "application/json"
 			},
-			body: JSON.stringify(params),
+			body: JSON.stringify(params)
 		});
 
 		if (response.status !== 200) {
@@ -130,7 +177,7 @@ export const createProdia = ({ apiKey, base: _base }: CreateProdiaOptions) => {
 
 	const getJob = async (jobId: string) => {
 		const response = await fetch(`${base}/job/${jobId}`, {
-			headers,
+			headers
 		});
 
 		if (response.status !== 200) {
@@ -157,7 +204,7 @@ export const createProdia = ({ apiKey, base: _base }: CreateProdiaOptions) => {
 
 	const listModels = async () => {
 		const response = await fetch(`${base}/models/list`, {
-			headers,
+			headers
 		});
 
 		if (response.status !== 200) {
@@ -173,6 +220,6 @@ export const createProdia = ({ apiKey, base: _base }: CreateProdiaOptions) => {
 		controlnet,
 		wait,
 		getJob,
-		listModels,
+		listModels
 	};
 };