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

app = FastAPI()

@app.get("/")
def health_check():
    return {"status": "FastAPI is running!"}

class Query(BaseModel):
    query: str

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