Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -142,7 +142,7 @@ def extract_text(html_content: str) -> str:
     return text
 
 
-async def websocket_main(code: str, websocket: WebSocket):
+async def websocket_main(code: str, brand_name: Optional[str] = None, websocket: WebSocket):
     access_token = code
     messages = get_messages(access_token)
     await websocket.send_json({"total_messages": len(messages)})