mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
9c89d3452c
feat(nodes): add ResultsServiceABC & SqliteResultsService **Doesn't actually work bc of circular imports. Can't even test it.** - add a base class for ResultsService and SQLite implementation - use `graph_execution_manager` `on_changed` callback to keep `results` table in sync fix(nodes): fix results service bugs chore(ui): regen api fix(ui): fix type guards feat(nodes): add `result_type` to results table, fix types fix(nodes): do not shadow `list` builtin feat(nodes): add results router It doesn't work due to circular imports still fix(nodes): Result class should use outputs classes, not fields feat(ui): crude results router fix(ui): send to canvas in currentimagebuttons not working feat(nodes): add core metadata builder feat(nodes): add design doc feat(nodes): wip latents db stuff feat(nodes): images_db_service and resources router feat(nodes): wip images db & router feat(nodes): update image related names feat(nodes): update urlservice feat(nodes): add high-level images service
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
from fastapi import HTTPException, Path, Query
|
|
from fastapi.routing import APIRouter
|
|
from invokeai.app.models.image import (
|
|
ImageCategory,
|
|
ImageType,
|
|
)
|
|
from invokeai.app.services.image_db import ImageRecordServiceBase
|
|
from invokeai.app.services.image_storage import ImageStorageBase
|
|
from invokeai.app.services.models.image_record import ImageRecord
|
|
from invokeai.app.services.item_storage import PaginatedResults
|
|
|
|
from ..dependencies import ApiDependencies
|
|
|
|
image_records_router = APIRouter(prefix="/v1/records/images", tags=["records"])
|
|
|
|
|
|
@image_records_router.get("/{image_type}/{image_name}", operation_id="get_image_record")
|
|
async def get_image_record(
|
|
image_type: ImageType = Path(description="The type of the image record to get"),
|
|
image_name: str = Path(description="The id of the image record to get"),
|
|
) -> ImageRecord:
|
|
"""Gets an image record by id"""
|
|
|
|
try:
|
|
return ApiDependencies.invoker.services.images_new.get_record(
|
|
image_type=image_type, image_name=image_name
|
|
)
|
|
except ImageRecordServiceBase.ImageRecordNotFoundException:
|
|
raise HTTPException(status_code=404)
|
|
|
|
|
|
@image_records_router.get(
|
|
"/",
|
|
operation_id="list_image_records",
|
|
)
|
|
async def list_image_records(
|
|
image_type: ImageType = Query(description="The type of image records to get"),
|
|
image_category: ImageCategory = Query(
|
|
description="The kind of image records to get"
|
|
),
|
|
page: int = Query(default=0, description="The page of image records to get"),
|
|
per_page: int = Query(
|
|
default=10, description="The number of image records per page"
|
|
),
|
|
) -> PaginatedResults[ImageRecord]:
|
|
"""Gets a list of image records by type and category"""
|
|
|
|
images = ApiDependencies.invoker.services.images_new.get_many(
|
|
image_type=image_type,
|
|
image_category=image_category,
|
|
page=page,
|
|
per_page=per_page,
|
|
)
|
|
|
|
return images
|
|
|
|
|
|
@image_records_router.delete("/{image_type}/{image_name}", operation_id="delete_image")
|
|
async def delete_image_record(
|
|
image_type: ImageType = Query(description="The type of image records to get"),
|
|
image_name: str = Path(description="The name of the image to delete"),
|
|
) -> None:
|
|
"""Deletes an image record"""
|
|
|
|
try:
|
|
ApiDependencies.invoker.services.images_new.delete(
|
|
image_type=image_type, image_name=image_name
|
|
)
|
|
except ImageStorageBase.ImageFileDeleteException:
|
|
# TODO: log this
|
|
pass
|
|
except ImageRecordServiceBase.ImageRecordDeleteException:
|
|
# TODO: log this
|
|
pass
|