mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
from fastapi import APIRouter, Body, HTTPException, Path
|
|
|
|
from invokeai.app.api.dependencies import ApiDependencies
|
|
from invokeai.app.services.style_preset_records.style_preset_records_common import (
|
|
StylePresetChanges,
|
|
StylePresetNotFoundError,
|
|
StylePresetRecordDTO,
|
|
StylePresetWithoutId,
|
|
)
|
|
|
|
style_presets_router = APIRouter(prefix="/v1/style_presets", tags=["style_presets"])
|
|
|
|
|
|
@style_presets_router.get(
|
|
"/i/{style_preset_id}",
|
|
operation_id="get_style_preset",
|
|
responses={
|
|
200: {"model": StylePresetRecordDTO},
|
|
},
|
|
)
|
|
async def get_style_preset(
|
|
style_preset_id: str = Path(description="The style preset to get"),
|
|
) -> StylePresetRecordDTO:
|
|
"""Gets a style preset"""
|
|
try:
|
|
return ApiDependencies.invoker.services.style_preset_records.get(style_preset_id)
|
|
except StylePresetNotFoundError:
|
|
raise HTTPException(status_code=404, detail="Style preset not found")
|
|
|
|
|
|
@style_presets_router.patch(
|
|
"/i/{style_preset_id}",
|
|
operation_id="update_style_preset",
|
|
responses={
|
|
200: {"model": StylePresetRecordDTO},
|
|
},
|
|
)
|
|
async def update_style_preset(
|
|
style_preset_id: str = Path(description="The id of the style preset to update"),
|
|
changes: StylePresetChanges = Body(description="The updated style preset", embed=True),
|
|
) -> StylePresetRecordDTO:
|
|
"""Updates a style preset"""
|
|
return ApiDependencies.invoker.services.style_preset_records.update(id=style_preset_id, changes=changes)
|
|
|
|
|
|
@style_presets_router.delete(
|
|
"/i/{style_preset_id}",
|
|
operation_id="delete_style_preset",
|
|
)
|
|
async def delete_style_preset(
|
|
style_preset_id: str = Path(description="The style preset to delete"),
|
|
) -> None:
|
|
"""Deletes a style preset"""
|
|
ApiDependencies.invoker.services.style_preset_records.delete(style_preset_id)
|
|
|
|
|
|
@style_presets_router.post(
|
|
"/",
|
|
operation_id="create_style_preset",
|
|
responses={
|
|
200: {"model": StylePresetRecordDTO},
|
|
},
|
|
)
|
|
async def create_style_preset(
|
|
style_preset: StylePresetWithoutId = Body(description="The style preset to create", embed=True),
|
|
) -> StylePresetRecordDTO:
|
|
"""Creates a style preset"""
|
|
return ApiDependencies.invoker.services.style_preset_records.create(style_preset=style_preset)
|
|
|
|
|
|
@style_presets_router.get(
|
|
"/",
|
|
operation_id="list_style_presets",
|
|
responses={
|
|
200: {"model": list[StylePresetRecordDTO]},
|
|
},
|
|
)
|
|
async def list_style_presets() -> list[StylePresetRecordDTO]:
|
|
"""Gets a page of style presets"""
|
|
return ApiDependencies.invoker.services.style_preset_records.get_many()
|