Update api.py

api.py

@@ -1,5 +1,4 @@
 import os
-from typing import Union
 
 from fastapi import FastAPI, WebSocket
 from epub2txt import epub2txt
@@ -12,11 +11,10 @@ prompt = os.getenv("prompt")
 app = FastAPI()
 
 
-@app.websocket("/{f_name}/ws")
-async def read_root(*,
-                    websocket: WebSocket,
-                   f_name: str):
+@app.websocket("/ws")
+async def read_root(websocket: WebSocket):
     await websocket.accept()
+    f_name = await websocket.receive_text()
     ch_list = epub2txt(f_name, outputlist=True)
     chapter_titles = epub2txt.content_titles
     title = epub2txt.title