mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
160267c71a
- Remove `ImageType` entirely, it is confusing - Create `ResourceOrigin`, may be `internal` or `external` - Revamp `ImageCategory`, may be `general`, `mask`, `control`, `user`, `other`. Expect to add more as time goes on - Update images `list` route to accept `include_categories` OR `exclude_categories` query parameters to afford finer-grained querying. All services are updated to accomodate this change. The new setup should account for our types of images, including the combinations we couldn't really handle until now: - Canvas init and masks - Canvas when saved-to-gallery or merged
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
import os
|
|
from abc import ABC, abstractmethod
|
|
|
|
from invokeai.app.models.image import ResourceOrigin
|
|
from invokeai.app.util.thumbnails import get_thumbnail_name
|
|
|
|
|
|
class UrlServiceBase(ABC):
|
|
"""Responsible for building URLs for resources."""
|
|
|
|
@abstractmethod
|
|
def get_image_url(
|
|
self, image_origin: ResourceOrigin, image_name: str, thumbnail: bool = False
|
|
) -> str:
|
|
"""Gets the URL for an image or thumbnail."""
|
|
pass
|
|
|
|
|
|
class LocalUrlService(UrlServiceBase):
|
|
def __init__(self, base_url: str = "api/v1"):
|
|
self._base_url = base_url
|
|
|
|
def get_image_url(
|
|
self, image_origin: ResourceOrigin, image_name: str, thumbnail: bool = False
|
|
) -> str:
|
|
image_basename = os.path.basename(image_name)
|
|
|
|
# These paths are determined by the routes in invokeai/app/api/routers/images.py
|
|
if thumbnail:
|
|
return (
|
|
f"{self._base_url}/images/{image_origin.value}/{image_basename}/thumbnail"
|
|
)
|
|
|
|
return f"{self._base_url}/images/{image_origin.value}/{image_basename}"
|