Update main.py

main.py

@@ -1,5 +1,5 @@
 from typing import List, Union
-from fastapi import FastAPI, WebSocket
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.responses import HTMLResponse, JSONResponse
 
 from answerer import Answerer
@@ -72,7 +72,8 @@ async def answer(ws: WebSocket):
 
   print("ws accepted!")
       
-  input = await ws.receive_text()
+  try: input = await ws.receive_text()
+  except WebSocketDisconnect: return
 
   print("input received!")