main.py

import os

from fastapi import FastAPI, BackgroundTasks

from utils.cache_layer import get_summarize_from_cache, summarize_un_cache_page
from utils.data_proto import SummaryReq, SummariesReq
from utils.logging import read_last_n_logs
from utils.summary_utils import summarize

app = FastAPI()


@app.get("/")
async def root():
    return {"message": "Hello World"}


@app.post("/summary/")
async def summary(request: SummaryReq):
    if request.key != os.environ.get("AccessKey"):
        return 'Unauthorized'
    return summarize(request.id, request.text)


@app.post("/summaries/")
async def summaries(background_tasks: BackgroundTasks, request: SummariesReq):
    if request.key != os.environ.get("AccessKey"):
        return 'Unauthorized'
    pages_summaries, uncached_pages = get_summarize_from_cache(request.pages)
    background_tasks.add_task(summarize_un_cache_page, uncached_pages)
    return pages_summaries


@app.get("/logs/")
async def logs(n: int = 100):
    return read_last_n_logs(n)