108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
# Agenten Plattform
|
||
#
|
||
# (c) 2024 Magnus Bender
|
||
# Institute of Humanities-Centered Artificial Intelligence (CHAI)
|
||
# Universitaet Hamburg
|
||
# https://www.chai.uni-hamburg.de/~bender
|
||
#
|
||
# source code released under the terms of GNU Public License Version 3
|
||
# https://www.gnu.org/licenses/gpl-3.0.txt
|
||
|
||
import os
|
||
|
||
from datetime import datetime
|
||
|
||
from fastapi import FastAPI, Request
|
||
from fastapi.responses import HTMLResponse
|
||
from fastapi.templating import Jinja2Templates
|
||
|
||
from jinja2.runtime import Undefined as JinjaUndefined
|
||
|
||
from ums.management.interface import Interface
|
||
from ums.management.db import DB
|
||
|
||
from ums.utils import AgentMessage, RiddleData, RiddleDataType, RiddleSolution, TEMPLATE_PATH
|
||
|
||
class WebMain():
|
||
|
||
_TIME_FORMAT = "%H:%M:%S %d.%m.%Y"
|
||
|
||
def __init__(self):
|
||
self._init_app()
|
||
self._init_templates()
|
||
|
||
self.db = DB()
|
||
|
||
self._add_routes()
|
||
self._add_routers()
|
||
|
||
|
||
def _init_app(self):
|
||
self.app = FastAPI(
|
||
title="Agenten Plattform",
|
||
description="Agenten Plattform – Management",
|
||
openapi_url="/api/schema.json",
|
||
docs_url='/api',
|
||
redoc_url=None
|
||
)
|
||
|
||
def _init_templates(self):
|
||
self.template = Jinja2Templates(
|
||
directory=TEMPLATE_PATH,
|
||
auto_reload=True
|
||
)
|
||
|
||
def timestamp2date(t:int|JinjaUndefined) -> str:
|
||
return "" if isinstance(t, JinjaUndefined) \
|
||
else datetime.fromtimestamp(t).strftime(self._TIME_FORMAT)
|
||
self.template.env.globals["timestamp2date"] = timestamp2date
|
||
|
||
def date2timestamp(d:str|JinjaUndefined) -> int|str:
|
||
return "" if isinstance(d, JinjaUndefined) \
|
||
else int(datetime.strptime(d, self._TIME_FORMAT).timestamp())
|
||
self.template.env.globals["date2timestamp"] = date2timestamp
|
||
|
||
|
||
def _add_routers(self):
|
||
interface_router = Interface(self.template, self.db)
|
||
self.app.include_router(interface_router.router)
|
||
|
||
def _add_routes(self):
|
||
|
||
@self.app.get("/index", response_class=HTMLResponse, summary="Link list")
|
||
def index(request: Request):
|
||
return self.template.TemplateResponse(
|
||
'index.html',
|
||
{"request" : request}
|
||
)
|
||
|
||
@self.app.get("/test", summary="Test")
|
||
def huhu(request: Request) -> AgentMessage:
|
||
ex = AgentMessage(
|
||
id="ex1",
|
||
riddle={
|
||
"context":"Example 1",
|
||
"question":"Get the name of the person."
|
||
},
|
||
data=[
|
||
RiddleData(
|
||
type=RiddleDataType.TEXT,
|
||
file_plain="./cv.txt"
|
||
)
|
||
]
|
||
)
|
||
ex.status.extract.required = False
|
||
|
||
ex.solution = RiddleSolution(
|
||
solution="Otto",
|
||
explanation="Written in line 6 after 'Name:'"
|
||
)
|
||
|
||
ins_count = self.db.add_message('from', 'to', ex)
|
||
self.db.set_processed(ins_count)
|
||
|
||
return ex
|
||
|
||
if __name__ == "ums.management.main" and os.environ.get('SERVE', 'false') == 'true':
|
||
main = WebMain()
|
||
app = main.app |