Spaces:

jnkr36
/

demo

Runtime error

App Files Files Community

demo / script_examples /basic_api_example.py

jnkr36

Upload basic_api_example.py

8e0d1ae about 1 year ago

raw history blame contribute delete

No virus

2.56 kB

	import json
	from urllib import request, parse
	import random

	#this is the ComfyUI api prompt format. If you want it for a specific workflow you can copy it from the prompt section
	#of the image metadata of images generated with ComfyUI
	#keep in mind ComfyUI is pre alpha software so this format will change a bit.

	#this is the one for the default workflow
	prompt_text = """
	{
	"3": {
	"class_type": "KSampler",
	"inputs": {
	"cfg": 8,
	"denoise": 1,
	"latent_image": [
	"5",
	0
	],
	"model": [
	"4",
	0
	],
	"negative": [
	"7",
	0
	],
	"positive": [
	"6",
	0
	],
	"sampler_name": "euler",
	"scheduler": "normal",
	"seed": 8566257,
	"steps": 20
	}
	},
	"4": {
	"class_type": "CheckpointLoaderSimple",
	"inputs": {
	"ckpt_name": "v1-5-pruned-emaonly.ckpt"
	}
	},
	"5": {
	"class_type": "EmptyLatentImage",
	"inputs": {
	"batch_size": 1,
	"height": 512,
	"width": 512
	}
	},
	"6": {
	"class_type": "CLIPTextEncode",
	"inputs": {
	"clip": [
	"4",
	1
	],
	"text": "masterpiece best quality girl"
	}
	},
	"7": {
	"class_type": "CLIPTextEncode",
	"inputs": {
	"clip": [
	"4",
	1
	],
	"text": "bad hands"
	}
	},
	"8": {
	"class_type": "VAEDecode",
	"inputs": {
	"samples": [
	"3",
	0
	],
	"vae": [
	"4",
	2
	]
	}
	},
	"9": {
	"class_type": "SaveImage",
	"inputs": {
	"filename_prefix": "ComfyUI",
	"images": [
	"8",
	0
	]
	}
	}
	}
	"""

	def queue_prompt(prompt):
	p = {"prompt": prompt}
	data = json.dumps(p).encode('utf-8')
	req = request.Request("http://127.0.0.1:8188/prompt", data=data)
	request.urlopen(req)


	prompt = json.loads(prompt_text)
	#set the text prompt for our positive CLIPTextEncode
	prompt["6"]["inputs"]["text"] = "masterpiece best quality man"

	#set the seed for our KSampler node
	prompt["3"]["inputs"]["seed"] = 5


	queue_prompt(prompt)