|
|
|
|
|
from contextlib import asynccontextmanager |
|
|
from fastapi import FastAPI, Request, Form |
|
|
from fastapi.responses import HTMLResponse |
|
|
from fastapi.templating import Jinja2Templates |
|
|
from api.endpoints import router |
|
|
from rag.db.initializer import initialize |
|
|
from rag.service.search import search |
|
|
|
|
|
templates = Jinja2Templates(directory="templates") |
|
|
|
|
|
@asynccontextmanager |
|
|
async def lifespan(_app: FastAPI): |
|
|
initialize() |
|
|
yield |
|
|
|
|
|
app = FastAPI(lifespan=lifespan) |
|
|
app.include_router(router) |
|
|
|
|
|
@app.get("/", response_class=HTMLResponse) |
|
|
def index(request: Request): |
|
|
return templates.TemplateResponse("index.html", {"request": request, "results": None}) |
|
|
|
|
|
@app.post("/demo", response_class=HTMLResponse) |
|
|
def demo(request: Request, query: str = Form(...)): |
|
|
results = search(query) |
|
|
return templates.TemplateResponse("index.html", {"request": request, "results": results, "query": query}) |
|
|
|
|
|
|
|
|
|
