feature: chatbot

app/main.py

@@ -1,17 +1,19 @@
 from fastapi import FastAPI
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
-from typing import Union
+from transformers import pipeline
 
 app = FastAPI()
+app.mount('/static', StaticFiles(directory='app/static'), name='static')
+question_answerer = pipeline(task='question-answering', 
+                             model='distilbert-base-cased-distilled-squad')
 
-app.mount("/static", StaticFiles(directory="app/static"), name="static")
-
-@app.get("/", response_class=HTMLResponse)
+@app.get('/', response_class=HTMLResponse)
 def read_root():
-    with open("app/index.html", "r") as html_file:
+    with open('app/index.html', 'r') as html_file:
         return html_file.read()
 
-@app.get("/items/{item_id}")
-def read_item(item_id: int, q: Union[str, None] = None):
-    return {"item_id": item_id, "q": q}
+@app.post('/chatbot')
+async def ask_question(question: str, context: str):
+    result = question_answerer(question=question, context=context)
+    return {'answer': result['answer']}