Update app.py

app.py

@@ -1,9 +1,17 @@
 from fastapi import FastAPI
-
+from pydantic import BaseModel
+from qabot import llm_chain
 
 app = FastAPI()
 
-
 @app.get("/")
-def root():
-    return {"message": "FastAPI backend is running!"}
+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"]}