Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(nodes): add uuid to uploaded images filename

Browse Source
This commit is contained in:
psychedelicious 2023-04-04 14:24:32 +10:00
parent a065f7db56
commit ed7fe23436
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/api/routers/images.py
View File

@ -1,6 +1,7 @@
# Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654) # Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654)
import io import io
from datetime import datetime, timezone from datetime import datetime, timezone
import uuid
from fastapi import Path, Query, Request, UploadFile from fastapi import Path, Query, Request, UploadFile
from fastapi.responses import FileResponse, Response from fastapi.responses import FileResponse, Response
@ -55,14 +56,14 @@ async def upload_image(file: UploadFile, request: Request):
# Error opening the image # Error opening the image
return Response(status_code=415) return Response(status_code=415)
filename = f"{str(int(datetime.now(timezone.utc).timestamp()))}.png" filename = f"{uuid.uuid4()}_{str(int(datetime.now(timezone.utc).timestamp()))}.png"
ApiDependencies.invoker.services.images.save("uploads", filename, im) ApiDependencies.invoker.services.images.save(ImageType.UPLOAD, filename, im)
return Response( return Response(
status_code=201, status_code=201,
headers={ headers={
"Location": request.url_for( "Location": request.url_for(
"get_image", image_type="uploads", image_name=filename "get_image", image_type=ImageType.UPLOAD.value, image_name=filename
) )
}, },
) )