mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
50bef87da7
Metadata for the Linear UI is now sneakily provided via a `MetadataAccumulator` node, which the client populates / hooks up while building the graph.
Additionally, we provide the unexpanded graph with the metadata API response.
Both of these are embedded into the PNGs.
- Remove `metadata` from `ImageDTO`
- Split up the `images/` routes to accomodate this; metadata is only retrieved per-image
- `images/{image_name}` now gets the DTO
- `images/{image_name}/metadata` gets the new metadata
- `images/{image_name}/full` gets the full-sized image file
- Remove old metadata service
- Add `MetadataAccumulator` node, `CoreMetadataField`, hook up to `LatentsToImage` node
- Add `get_raw()` method to `ItemStorage`, retrieves the row from DB as a string, no pydantic parsing
- Update `images`related services to handle storing and retrieving the new metadata
- Add `get_metadata_graph_from_raw_session` which extracts the `graph` from `session` without needing to hydrate the session in pydantic, in preparation for providing it as metadata; also removes all references to the `MetadataAccumulator` node
26 lines
804 B
Python
26 lines
804 B
Python
import os
|
|
from abc import ABC, abstractmethod
|
|
|
|
|
|
class UrlServiceBase(ABC):
|
|
"""Responsible for building URLs for resources."""
|
|
|
|
@abstractmethod
|
|
def get_image_url(self, 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_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_basename}/thumbnail"
|
|
|
|
return f"{self._base_url}/images/{image_basename}/full"
|