108 lines
3.9 KiB
Python
108 lines
3.9 KiB
Python
from fastapi import APIRouter, File, Form, HTTPException, UploadFile
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
|
|
from services.character_card import list_characters, get_character, import_card_file, update_character, update_appearance_tags
|
|
|
|
router = APIRouter(prefix="/characters", tags=["characters"])
|
|
|
|
|
|
class CardPatch(BaseModel):
|
|
name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
personality: Optional[str] = None
|
|
scenario: Optional[str] = None
|
|
first_mes: Optional[str] = None
|
|
mes_example: Optional[str] = None
|
|
appearance_tags: Optional[str] = None
|
|
lora_name: Optional[str] = None
|
|
lora_weight: Optional[float] = None
|
|
|
|
|
|
@router.get("/")
|
|
async def list_all():
|
|
return await list_characters()
|
|
|
|
|
|
@router.get("/{card_id}")
|
|
async def get_one(card_id: str):
|
|
card = await get_character(card_id)
|
|
if not card:
|
|
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
|
return card
|
|
|
|
|
|
@router.patch("/{card_id}")
|
|
async def patch_card(card_id: str, body: CardPatch):
|
|
card = await get_character(card_id)
|
|
if not card:
|
|
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
|
fields = {k: v for k, v in body.model_dump().items() if v is not None}
|
|
await update_character(card_id, fields)
|
|
# sync appearance_tags and lora to persona
|
|
from services.personas import update_persona_appearance
|
|
if "appearance_tags" in fields:
|
|
await update_persona_appearance(f"card_{card_id}", fields["appearance_tags"])
|
|
if {"lora_name", "lora_weight"} & fields.keys():
|
|
from services.personas import update_persona_lora
|
|
await update_persona_lora(f"card_{card_id}", fields.get("lora_name"), fields.get("lora_weight"))
|
|
# rebuild system prompt if character fields changed
|
|
char_fields = {"name", "description", "personality", "scenario", "first_mes", "mes_example"}
|
|
if char_fields & fields.keys():
|
|
updated = await get_character(card_id)
|
|
from services.character_card import build_system_prompt
|
|
from services.personas import update_persona_prompt
|
|
await update_persona_prompt(f"card_{card_id}", build_system_prompt(updated))
|
|
return await get_character(card_id)
|
|
|
|
|
|
@router.post("/{card_id}/avatar")
|
|
async def upload_avatar(card_id: str, file: UploadFile = File(...)):
|
|
card = await get_character(card_id)
|
|
if not card:
|
|
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
|
content = await file.read()
|
|
if not content.startswith(b"\x89PNG"):
|
|
raise HTTPException(status_code=400, detail="Нужен PNG")
|
|
from services.character_card import _save_avatar_bytes
|
|
rel = _save_avatar_bytes(content, f"card_{card_id}")
|
|
await update_character(card_id, {"avatar_path": rel})
|
|
# sync persona
|
|
from services.personas import patch_persona
|
|
await patch_persona(f"card_{card_id}", {"avatar_path": rel})
|
|
return {"avatar_path": f"/static/{rel}"}
|
|
|
|
|
|
@router.post("/import")
|
|
async def import_card(
|
|
file: UploadFile = File(...),
|
|
lora_name: str = Form(""),
|
|
lora_weight: float = Form(0.8),
|
|
):
|
|
content = await file.read()
|
|
try:
|
|
card = await import_card_file(
|
|
content,
|
|
file.filename or "card.json",
|
|
lora_name=lora_name,
|
|
lora_weight=lora_weight,
|
|
)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
return {
|
|
"status": "imported",
|
|
"card_id": card["card_id"],
|
|
"persona_id": f"card_{card['card_id']}",
|
|
"name": card["name"],
|
|
}
|
|
|
|
|
|
@router.delete("/{card_id}")
|
|
async def remove_card(card_id: str):
|
|
from services.personas import delete_persona
|
|
|
|
if not await delete_persona(f"card_{card_id}"):
|
|
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
|
return {"status": "deleted", "card_id": card_id}
|
|
|