File size: 544 Bytes
f5eeece |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# api.py
from fastapi import FastAPI, UploadFile, File, Form
from model import load_vectorstore, ask_question
import shutil
app = FastAPI()
@app.post("/upload")
async def upload(file: UploadFile = File(...)):
with open("document.pdf", "wb") as buffer:
shutil.copyfileobj(file.file, buffer)
load_vectorstore("document.pdf")
return {"message": "File indexed successfully."}
@app.post("/ask")
async def ask(question: str = Form(...)):
answer = ask_question(question)
return {"question": question, "answer": answer}
|