Pudding48 commited on
Commit
fc399f3
·
verified ·
1 Parent(s): f383878

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -23
app.py CHANGED
@@ -1,23 +1,13 @@
1
- # backend.py
2
- from fastapi import FastAPI, Request
3
- from pydantic import BaseModel
4
- from qabot import llm_chain
5
- from fastapi.middleware.cors import CORSMiddleware
6
-
7
- app = FastAPI()
8
-
9
- # Allow CORS for local frontend testing
10
- app.add_middleware(
11
- CORSMiddleware,
12
- allow_origins=["*"],
13
- allow_methods=["*"],
14
- allow_headers=["*"],
15
- )
16
-
17
- class Query(BaseModel):
18
- query: str
19
-
20
- @app.post("/ask")
21
- async def ask_question(query: Query):
22
- answer = llm_chain.invoke({"query": query.query})
23
- return {"answer": answer["result"]}
 
1
+ from fastapi import FastAPI
2
+ from pydantic import BaseModel
3
+ from qabot import llm_chain
4
+
5
+ app = FastAPI()
6
+
7
+ class Query(BaseModel):
8
+ query: str
9
+
10
+ @app.post("/ask")
11
+ def ask_question(query: Query):
12
+ result = llm_chain.invoke({"query": query.query})
13
+ return {"answer": result["result"]}