Update main.py

main.py

@@ -1,4 +1,4 @@
-from typing import List, Union
+from typing import Union
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.responses import HTMLResponse, JSONResponse
 
@@ -35,7 +35,7 @@ HTML = """
     const prompt = document.getElementById("prompt");
     const output = document.getElementById("output");
 
-    const ws = new WebSocket("wss://daniilalpha-answerer-api.hf.space/answer"); 
+    const ws = new WebSocket("ws://127.0.0.1:2403/answer"); 
     ws.onmessage = (e) => answer(e.data);
 
     function ask(event) {
@@ -62,14 +62,14 @@ def index():
   return HTMLResponse(HTML)
 
 @app.get("/map")
-def map(query: str, items: List[str]):
+def map(query: Union[str, None], items: Union[list[str], None]):
   indices = mapper(query, items)
   return JSONResponse(indices)
 
 @app.websocket("/answer")
 async def answer(ws: WebSocket):
-  print("WEBSOCKET: /answer")
-    
+  print("ws started!")
+
   await ws.accept()
 
   print("ws accepted!")