from fastapi import FastAPI
from pydantic import BaseModel
from qabot import llm_chain

app = FastAPI()

class Query(BaseModel):
    query: str

@app.post("/ask")
def ask_question(query: Query):
    result = llm_chain.invoke({"query": query.query})
    return {"answer": result["result"]}