jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 15:04:50 +08:00
Code Issues Activity

Merge branch 'master' into save-images

Browse Source
This commit is contained in:
m957ymj75urz
2023-03-15 10:48:15 +00:00
committed by GitHub
parent 6daf9bb22c 54593db6dc
commit 3d2f60b315
28 changed files with 619 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server.py
View File

@@ -113,7 +113,7 @@ class PromptServer():
async def view_image(request):
if "file" in request.rel_url.query:
type = request.rel_url.query.get("type", "output")
if type != "output" and type != "input":
if type not in ["output", "input", "temp"]:
return web.Response(status=400)
output_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), type)
@@ -267,7 +267,7 @@ class PromptServer():
msg = await self.messages.get()
await self.send(*msg)
async def start(self, address, port, verbose=True):
async def start(self, address, port, verbose=True, call_on_start=None):
runner = web.AppRunner(self.app)
await runner.setup()
site = web.TCPSite(runner, address, port)
@@ -278,3 +278,6 @@ class PromptServer():
if verbose:
print("Starting server\n")
print("To see the GUI go to: http://{}:{}".format(address, port))
if call_on_start is not None:
call_on_start(address, port)