xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-05 06:47:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BFL] Print download URL of successful task result directly on nodes (#175)

Browse Source
This commit is contained in:
Christian Byrne 2025-05-12 15:23:25 -07:00 committed by Robin Huang
parent b76fcc12b9
commit d932dac5d1
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
comfy_api_nodes/nodes_bfl.py
View File

@ -137,14 +137,19 @@ def convert_mask_to_image(mask: torch.Tensor):
def handle_bfl_synchronous_operation( def handle_bfl_synchronous_operation(
operation: SynchronousOperation, timeout_bfl_calls=360 operation: SynchronousOperation,
timeout_bfl_calls=360,
node_id: Union[str, None] = None,
): ):
response_api: BFLFluxProGenerateResponse = operation.execute() response_api: BFLFluxProGenerateResponse = operation.execute()
return _poll_until_generated( return _poll_until_generated(
response_api.polling_url, timeout=timeout_bfl_calls response_api.polling_url, timeout=timeout_bfl_calls, node_id=node_id
) )
def _poll_until_generated(polling_url: str, timeout=360):
def _poll_until_generated(
polling_url: str, timeout=360, node_id: Union[str, None] = None
):
# used bfl-comfy-nodes to verify code implementation: # used bfl-comfy-nodes to verify code implementation:
# https://github.com/black-forest-labs/bfl-comfy-nodes/tree/main # https://github.com/black-forest-labs/bfl-comfy-nodes/tree/main
start_time = time.time() start_time = time.time()
@ -156,11 +161,21 @@ def _poll_until_generated(polling_url: str, timeout=360):
request = requests.Request(method=HttpMethod.GET, url=polling_url) request = requests.Request(method=HttpMethod.GET, url=polling_url)
# NOTE: should True loop be replaced with checking if workflow has been interrupted? # NOTE: should True loop be replaced with checking if workflow has been interrupted?
while True: while True:
if node_id:
time_elapsed = time.time() - start_time
PromptServer.instance.send_progress_text(
f"Generating ({time_elapsed:.0f}s)", node_id
)
response = requests.Session().send(request.prepare()) response = requests.Session().send(request.prepare())
if response.status_code == 200: if response.status_code == 200:
result = response.json() result = response.json()
if result["status"] == BFLStatus.ready: if result["status"] == BFLStatus.ready:
img_url = result["result"]["sample"] img_url = result["result"]["sample"]
if node_id:
PromptServer.instance.send_progress_text(
f"Result URL: {img_url}", node_id
)
img_response = requests.get(img_url) img_response = requests.get(img_url)
return process_image_response(img_response) return process_image_response(img_response)
elif result["status"] in [ elif result["status"] in [