from fastapi import FastAPI
from pydantic import BaseModel

from summary import summarize

app = FastAPI()


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


class SummaryReq(BaseModel):
    text: str
    key: str

@app.post("/summary/")
async def summary(request: SummaryReq):
    print('=============')
    print(request)
    return summarize(request.text)


@app.get("/summary2/{text}")
async def summary2(text: str):
    print(text)
    return summarize(text)