|
@@ -1,6 +1,5 @@
|
|
import asyncio
|
|
import asyncio
|
|
import websocket # NOTE: websocket-client (https://github.com/websocket-client/websocket-client)
|
|
import websocket # NOTE: websocket-client (https://github.com/websocket-client/websocket-client)
|
|
-import uuid
|
|
|
|
import json
|
|
import json
|
|
import urllib.request
|
|
import urllib.request
|
|
import urllib.parse
|
|
import urllib.parse
|
|
@@ -398,7 +397,9 @@ async def comfyui_generate_image(
|
|
return None
|
|
return None
|
|
|
|
|
|
try:
|
|
try:
|
|
- images = await asyncio.to_thread(get_images, ws, comfyui_prompt, client_id, base_url)
|
|
|
|
|
|
+ images = await asyncio.to_thread(
|
|
|
|
+ get_images, ws, comfyui_prompt, client_id, base_url
|
|
|
|
+ )
|
|
except Exception as e:
|
|
except Exception as e:
|
|
log.exception(f"Error while receiving images: {e}")
|
|
log.exception(f"Error while receiving images: {e}")
|
|
images = None
|
|
images = None
|