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 get_uri method to image storage

Browse Source 
- gets the external URI of an image
This commit is contained in:
psychedelicious
2023-04-25 00:24:03 +10:00
parent 74013b6611
commit 711d09a107
2 changed files with 35 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -104,17 +104,19 @@ async def upload_image(
invokeai_metadata = ApiDependencies.invoker.services.metadata.get_metadata(img)
image_url = ApiDependencies.invoker.services.images.get_uri(
ImageType.UPLOAD, saved_image.image_name
)
thumbnail_url = ApiDependencies.invoker.services.images.get_uri(
ImageType.UPLOAD, saved_image.image_name, True
)
res = ImageResponse(
image_type=ImageType.UPLOAD,
image_name=saved_image.image_name,
image_url=request.url_for(
"get_image", image_type=ImageType.UPLOAD.value, image_name=saved_image.image_name
),
thumbnail_url=request.url_for(
"get_thumbnail",
thumbnail_type=ImageType.UPLOAD.value,
thumbnail_name=saved_image.thumbnail_name,
),
image_url=image_url,
thumbnail_url=thumbnail_url,
metadata=ImageResponseMetadata(
created=saved_image.created,
width=img.width,