38 lines
937 B
Python
38 lines
937 B
Python
from pydantic import BaseModel, Field
|
|
from typing import List, Optional, Union, Dict, Any
|
|
|
|
|
|
class TaggerInterrogateRequest(BaseModel):
|
|
path: str
|
|
interrogator_model: str = Field(
|
|
default="wd14-convnextv2-v2"
|
|
)
|
|
threshold: float = Field(
|
|
default=0.35,
|
|
ge=0,
|
|
le=1
|
|
)
|
|
additional_tags: str = ""
|
|
exclude_tags: str = ""
|
|
escape_tag: bool = True
|
|
batch_input_recursive: bool = False
|
|
batch_output_action_on_conflict: str = "ignore"
|
|
replace_underscore: bool = True
|
|
replace_underscore_excludes: str = Field(
|
|
default="0_0, (o)_(o), +_+, +_-, ._., <o>_<o>, <|>_<|>, =_=, >_<, 3_3, 6_9, >_o, @_@, ^_^, o_o, u_u, x_x, |_|, ||_||"
|
|
)
|
|
|
|
|
|
class APIResponse(BaseModel):
|
|
status: str
|
|
message: Optional[str]
|
|
data: Optional[Dict]
|
|
|
|
|
|
class APIResponseSuccess(APIResponse):
|
|
status: str = "success"
|
|
|
|
|
|
class APIResponseFail(APIResponse):
|
|
status: str = "fail"
|