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

fix: duplicated file upload

Browse Source
This commit is contained in:
bennykok 2024-04-25 16:14:04 +08:00
parent 95c15f095d
commit cd5e4a5d01
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
custom_routes.py
View File

@ -833,7 +833,7 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p
"Content-Type": content_type, "Content-Type": content_type,
"Content-Length": str(len(data)), "Content-Length": str(len(data)),
} }
response = requests.put(ok.get("url"), headers=headers, data=data) # response = requests.put(ok.get("url"), headers=headers, data=data)
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.put(ok.get("url"), headers=headers, data=data) as response: async with session.put(ok.get("url"), headers=headers, data=data) as response:
print("Upload file response", response.status) print("Upload file response", response.status)
@ -977,7 +977,8 @@ async def update_run_with_output(prompt_id, data, node_id=None):
if have_upload: if have_upload:
await update_file_status(prompt_id, data, True, node_id=node_id) await update_file_status(prompt_id, data, True, node_id=node_id)
asyncio.create_task(upload_in_background(prompt_id, data, node_id=node_id, have_upload=have_upload)) # asyncio.create_task(upload_in_background(prompt_id, data, node_id=node_id, have_upload=have_upload))
await upload_in_background(prompt_id, data, node_id=node_id, have_upload=have_upload)
except Exception as e: except Exception as e:
await handle_error(prompt_id, data, e) await handle_error(prompt_id, data, e)