fix websocket error

Force IPv4 usage to prevent address conflicts
Add more debug logging to help diagnose issues
Handle errors more gracefully
Give more time for the servers to initialize properly

Author: mahesh0199

main.py

@@ -23,8 +23,10 @@
 class BrokerSelector:
     def __init__(self):
         self.selected_broker = None
+        print("BrokerSelector initialized")
 
     async def handle_client(self, reader, writer):
+        print("New client connection received")
         data = await reader.read(100)
         message = data.decode()
         addr = writer.get_extra_info("peername")
@@ -56,38 +58,56 @@ async def handle_client(self, reader, writer):
         await writer.wait_closed()
 
     async def start_ipc_server(self):
-        server = await asyncio.start_server(self.handle_client, "127.0.0.1", IPC_PORT)
-
-        addr = server.sockets[0].getsockname()
-        print(f"Serving on {addr}")
+        try:
+            server = await asyncio.start_server(
+                self.handle_client, 
+                '127.0.0.1',
+                IPC_PORT
+            )
+            addr = server.sockets[0].getsockname()
+            print(f"IPC Server running on {addr}")
 
-        async with server:
-            await server.serve_forever()
+            async with server:
+                await server.serve_forever()
+        except Exception as e:
+            print(f"Error starting IPC server: {e}")
 
 
 async def main():
-    broker_selector = BrokerSelector()
-
-    # Start the IPC server
-    asyncio.create_task(broker_selector.start_ipc_server())
-
-    print("WebSocket server starting...")
-    while not broker_selector.selected_broker:
-        await asyncio.sleep(1)
-
-    selected_broker = broker_selector.selected_broker
-    print(f"Selected broker: {selected_broker}")
-
-    ws_port = BROKER_PORTS[selected_broker]
-    print(f"Starting {selected_broker.capitalize()} WebSocket on port {ws_port}...")
-
-    if selected_broker == "flattrade":
-        flattrade_initialize_api()
-        await flattrade_main(ws_port)
-    elif selected_broker == "shoonya":
-        shoonya_initialize_api()
-        await shoonya_main(ws_port)
+    try:
+        broker_selector = BrokerSelector()
+        
+        # Start the IPC server and store the task
+        ipc_server_task = asyncio.create_task(broker_selector.start_ipc_server())
+        
+        print("WebSocket server starting...")
+        print("Waiting for broker selection...")
+        
+        while not broker_selector.selected_broker:
+            await asyncio.sleep(1)
+            print("Waiting for broker selection...")
+
+        selected_broker = broker_selector.selected_broker
+        print(f"Selected broker: {selected_broker}")
+
+        ws_port = BROKER_PORTS[selected_broker]
+        print(f"Starting {selected_broker.capitalize()} WebSocket on port {ws_port}...")
+
+        if selected_broker == "flattrade":
+            flattrade_initialize_api()
+            await flattrade_main(ws_port)
+        elif selected_broker == "shoonya":
+            shoonya_initialize_api()
+            await shoonya_main(ws_port)
+
+    except Exception as e:
+        print(f"Error in main: {e}")
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("Server shutting down...")
+    except Exception as e:
+        print(f"Fatal error: {e}")