from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI()


app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


@app.get("/ping")
async def ping():
    return "Hello, I am alive"


@app.post("/qa")
async def section(request: Request):

    data = await request.json()

    from qa import query
    answer = query(data["Question"])

    return {"Answer": answer}

@app.post("/csv")
async def section(request: Request):

    data = await request.json()

    from chat_csv import quiz
    answer = quiz(data["Question"])

    return {"Answer": answer}