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): allow uploaded images to be any ImageType (eg intermediates)

Browse Source
This commit is contained in:
psychedelicious 2023-05-04 22:07:14 +10:00
parent 5e09dd380d
commit 0b49997bb6
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -83,7 +83,7 @@ async def get_thumbnail(
status_code=201, status_code=201,
) )
async def upload_image( async def upload_image(
file: UploadFile, request: Request, response: Response file: UploadFile, image_type: ImageType, request: Request, response: Response
) -> ImageResponse: ) -> ImageResponse:
if not file.content_type.startswith("image"): if not file.content_type.startswith("image"):
raise HTTPException(status_code=415, detail="Not an image") raise HTTPException(status_code=415, detail="Not an image")
@ -99,21 +99,21 @@ async def upload_image(
filename = f"{uuid.uuid4()}_{str(int(datetime.now(timezone.utc).timestamp()))}.png" filename = f"{uuid.uuid4()}_{str(int(datetime.now(timezone.utc).timestamp()))}.png"
saved_image = ApiDependencies.invoker.services.images.save( saved_image = ApiDependencies.invoker.services.images.save(
ImageType.UPLOAD, filename, img image_type, filename, img
) )
invokeai_metadata = ApiDependencies.invoker.services.metadata.get_metadata(img) invokeai_metadata = ApiDependencies.invoker.services.metadata.get_metadata(img)
image_url = ApiDependencies.invoker.services.images.get_uri( image_url = ApiDependencies.invoker.services.images.get_uri(
ImageType.UPLOAD, saved_image.image_name image_type, saved_image.image_name
) )
thumbnail_url = ApiDependencies.invoker.services.images.get_uri( thumbnail_url = ApiDependencies.invoker.services.images.get_uri(
ImageType.UPLOAD, saved_image.image_name, True image_type, saved_image.image_name, True
) )
res = ImageResponse( res = ImageResponse(
image_type=ImageType.UPLOAD, image_type=image_type,
image_name=saved_image.image_name, image_name=saved_image.image_name,
image_url=image_url, image_url=image_url,
thumbnail_url=thumbnail_url, thumbnail_url=thumbnail_url,