app.py: added api payload control.

app.py

@@ -355,6 +355,70 @@ def generate_image (
 	return image, gr.update(visible=True)
 
 
+def get_payload (
+	model_name,
+	lora_selection,
+	face_image_path,
+	pose_image_path,
+	prompt,
+	negative_prompt,
+	style_name,
+	num_steps,
+	identitynet_strength_ratio,
+	adapter_strength_ratio,
+	controlnet_strength_1, controlnet_strength_2, controlnet_strength_3, controlnet_strength_4,
+	controlnet_selection,
+	guidance_scale,
+	seed,
+	scheduler,
+):
+	prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
+
+	ref_image_path = pose_image_path if pose_image_path else face_image_path
+	ref_image = PIL.Image.open(ref_image_path)
+
+	width, height = ref_image.size
+	large_edge = max(width, height)
+	if large_edge < 1024:
+		scaling = 1024 / large_edge
+		width = int(width * scaling)
+		height = int(height * scaling)
+
+	(
+		CONTROLNET_DICT['pose'].strength,
+		CONTROLNET_DICT['canny'].strength,
+		CONTROLNET_DICT['depth'].strength,
+		CONTROLNET_DICT['lineart'].strength,
+	) = [controlnet_strength_1, controlnet_strength_2, controlnet_strength_3, controlnet_strength_4]
+
+	return {
+		'extra': {
+			'response_image_type': 'jpeg',
+		},
+		'model_name': f'{model_name}.safetensors',
+		'face_image_assets_ids': "[assets_ids of id image, please manually upload to novita.ai]",
+		'ref_image_assets_ids': "[assets_ids of reference image, please manually upload to novita.ai]",
+		'prompt': prompt,
+		'negative_prompt': negative_prompt,
+		'controlnet': {
+			'units': [CONTROLNET_DICT[name] for name in controlnet_selection if name in CONTROLNET_DICT],
+		},
+		'loras': [dict(
+			model_name=f'{name}.safetensors',
+			strength=1,
+		) for name in lora_selection],
+		'image_num': 1,
+		'steps': num_steps,
+		'seed': seed,
+		'guidance_scale': guidance_scale,
+		'sampler_name': scheduler,
+		'id_strength': identitynet_strength_ratio,
+		'adapter_strength': adapter_strength_ratio,
+		'width': width,
+		'height': height,
+	}
+
+
 # Description
 title = r'''
 <h1 align="center">InstantID: Zero-shot Identity-Preserving Generation in Seconds (via Novita)</h1>
@@ -543,6 +607,8 @@ with gr.Blocks(css=css) as demo:
 				label='InstantID Usage Tips', value=tips, visible=False
 			)
 
+			api_payload = gr.JSON(label="Novita API Payload, POST /v3/async/instant-id")
+
 		submit.click(
 			fn=remove_tips,
 			outputs=usage_tips,
@@ -552,6 +618,31 @@ with gr.Blocks(css=css) as demo:
 			outputs=seed,
 			queue=False,
 			api_name=False,
+		).then(
+			fn=get_payload,
+			inputs=[
+				model_name,
+				lora_selection,
+				face_file,
+				pose_file,
+				prompt,
+				negative_prompt,
+				style,
+				num_steps,
+				identitynet_strength_ratio,
+				adapter_strength_ratio,
+				#[
+					pose_strength,
+					canny_strength,
+					depth_strength,
+					lineart_strength,
+				#],
+				controlnet_selection,
+				guidance_scale,
+				seed,
+				scheduler,
+			],
+			outputs=api_payload,
 		).then(
 			fn=generate_image,
 			inputs=[