From c3ce88fd6f42a319f35d608dc0d688113c29e280 Mon Sep 17 00:00:00 2001 From: Radionic Date: Mon, 27 Nov 2023 13:14:18 +0800 Subject: [PATCH] feat: custom workflow --- routes.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/routes.py b/routes.py index f9db3f7..e7ad1e4 100644 --- a/routes.py +++ b/routes.py @@ -217,7 +217,7 @@ with open( os.path.dirname(__file__), "workflow_templates/api/avatar_generation_api.json" ) ) as f: - api_prompt_string = "\n".join(f.readlines()) + default_workflow = "\n".join(f.readlines()) @server.PromptServer.instance.routes.post("/avatar_generation") @@ -227,9 +227,10 @@ async def post_prompt_block(request): image = post.get("image") image_name = save_image(image) - api_prompt = json.loads(api_prompt_string.replace("IMAGE_REFERENCE", image_name)) - json_data = {"prompt": api_prompt} - res = post_prompt(json_data) + workflow = post.get("workflow") + workflow = default_workflow if workflow is None else workflow + api_prompt = json.loads(workflow.replace("IMAGE_REFERENCE", image_name)) + res = post_prompt({"prompt": api_prompt}) prompt_id = json.loads(res.text)["prompt_id"] while True: history = prompt_server.prompt_queue.get_history(prompt_id=prompt_id)