chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: ensure logs are disabled by default

Browse Source
This commit is contained in:
bennykok 2024-01-30 15:17:25 +08:00
parent f73baa091a
commit b6b34c9062
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
custom_routes.py
View File

@ -26,6 +26,7 @@ import threading
api = None api = None
api_task = None api_task = None
prompt_metadata = {} prompt_metadata = {}
cd_enable_log = os.environ.get('CD_ENABLE_LOG', 'false').lower() == 'true'
def post_prompt(json_data): def post_prompt(json_data):
prompt_server = server.PromptServer.instance prompt_server = server.PromptServer.instance
@ -157,6 +158,8 @@ async def websocket_handler(request):
try: try:
# Send initial state to the new client # Send initial state to the new client
await send("status", { 'sid': sid }, sid) await send("status", { 'sid': sid }, sid)
if cd_enable_log:
await send_first_time_log(sid) await send_first_time_log(sid)
async for msg in ws: async for msg in ws:
@ -480,4 +483,5 @@ def run_in_new_thread(coroutine):
t.start() t.start()
asyncio.run_coroutine_threadsafe(coroutine, new_loop) asyncio.run_coroutine_threadsafe(coroutine, new_loop)
run_in_new_thread(watch_file_changes(log_file_path, send_logs_to_websocket)) if cd_enable_log:
run_in_new_thread(watch_file_changes(log_file_path, send_logs_to_websocket))