mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import random
|
|
import datetime
|
|
from typing import *
|
|
from starlette.requests import Request
|
|
from starlette.responses import *
|
|
from royalnet.constellation import *
|
|
from royalnet.utils import *
|
|
from ..tables import *
|
|
from ..utils import Emotion
|
|
|
|
|
|
class ApiKei(PageStar):
|
|
path = "/api/kei"
|
|
|
|
methods = ["POST"]
|
|
|
|
async def _generate(self, person, form, session) -> Tuple[Emotion, str]:
|
|
return Emotion.NEUTRAL, "..."
|
|
|
|
async def page(self, request: Request) -> JSONResponse:
|
|
async with self.session_acm() as session:
|
|
form = await request.form()
|
|
person = session.query(self.alchemy.get(KeiPerson)).filter_by(kpid=form["kpid"]).one_or_none()
|
|
if person is None:
|
|
person = self.alchemy.get(KeiPerson)(kpid=form["kpid"])
|
|
session.add(person)
|
|
message = self.alchemy.get(KeiMessage)(kei_person=person, message=form["message"])
|
|
session.add(message)
|
|
await asyncify(session.commit)
|
|
try:
|
|
emotion, text = await self._generate(person, form, session)
|
|
except Exception as e:
|
|
print(e)
|
|
emotion, text = Emotion.NEUTRAL, "...?"
|
|
return JSONResponse({
|
|
"emotion": str(emotion),
|
|
"text": text,
|
|
})
|