Create server.py

server.py

@@ -0,0 +1,54 @@
+from typing import Optional
+from fastapi import BackgroundTasks, FastAPI
+from pydantic import BaseModel
+import json
+import time
+import random
+
+
+app = FastAPI()
+answer_dict = dict()
+
+
+class DialogInfo(BaseModel):
+    q_id: str = None
+    query: str = None
+    history: list = []
+    
+
+@app.get("/")
+def read_root():
+    return {"Hello": "World!"}
+
+
+@app.get("/chatbot/dialoginfo")
+def read_dialog(dialogid: str):
+    # return dialogid
+    return answer_dict[dialogid]
+
+
+@app.get("/chatbot/finishquery")
+def read_dialog(dialogid: str):
+    # return dialogid
+    return random.choices([True, False], (0.3, 0.7))[0]
+
+
+def background_process(diainfo: DialogInfo):
+    answer_dict[diainfo.q_id].q_id = diainfo.q_id
+    answer_dict[diainfo.q_id].history = diainfo.history
+    answer_dict[diainfo.q_id].query = diainfo.query
+    for _ in range(5):
+        answer_dict[diainfo.q_id].history = diainfo.history
+        if diainfo.q_id.startswith('0'):
+            answer_dict[diainfo.q_id].query = answer_dict[diainfo.q_id].query + 'i'
+        elif diainfo.q_id.startswith('9'):
+            answer_dict[diainfo.q_id].query = answer_dict[diainfo.q_id].query + 'o'
+        time.sleep(0.5)
+    print(answer_dict[diainfo.q_id])
+
+
+@app.post("/chatbot/")
+def read_query(chat: DialogInfo, background_tasks: BackgroundTasks):
+    answer_dict[chat.q_id] = chat
+    background_tasks.add_task(background_process, chat)
+    return {"code": 202, "msg": f"Query id {chat.q_id} accepted."}