Update main.py

main.py

@@ -1,5 +1,6 @@
+from typing import List, Union
 from fastapi import FastAPI, WebSocket
-from fastapi.responses import HTMLResponse
+from fastapi.responses import HTMLResponse, JSONResponse
 
 from answerer import Answerer
 from mapper import Mapper
@@ -61,8 +62,9 @@ def index():
   return HTMLResponse(HTML)
 
 @app.get("/map")
-def map():
-  return HTMLResponse(HTML)
+def map(query: str, items: List[str]):
+  indices = mapper(query, items)
+  return JSONResponse(indices)
 
 @app.websocket("/answer")
 async def answer(ws: WebSocket):