Update clientChaindesk.py

clientChaindesk.py

@@ -10,16 +10,16 @@ client_ports = []
 
 # Define the websocket client class
 class WebSocketClient6:
-    def __init__(self):
-        # Initialize the uri attribute        
-        self.name = f"Chaindesk client port: {self.clientPort}"
+    def __init__(self, clientPort):
+        # Initialize the uri attribute
+        self.uri = f'ws://localhost:{clientPort}'   
+        self.name = f"Chaindesk client port: {clientPort}"
         self.status = st.sidebar.status(label=self.name, state="complete", expanded=False)
+        st.session_state.clientPort = clientPort
 
         if "client_ports" not in st.session_state:
             st.session_state['client_ports'] = ""
-        if "clientPort" not in st.session_state:
-            st.session_state.clientPort = ""
-
+        
     async def askChaindesk(self, question):
         
         if "agentID" not in st.session_state:
@@ -75,14 +75,11 @@ class WebSocketClient6:
         print("Stopping WebSocket client...")    
 
     # Define a coroutine that will connect to the server and exchange messages
-    async def startClient(self, clientPort):
-        uri = f'ws://localhost:{clientPort}'
-        client_ports.append(clientPort)
-        st.session_state['client_ports'] = client_ports
-        st.session_state.clientPort = clientPort
+    async def startClient(self):
+        
         status = self.status       
         # Connect to the server
-        async with websockets.connect(uri) as websocket:
+        async with websockets.connect(self.uri) as websocket:
             # Loop forever
             while True:
                 status.update(label=self.name, state="running", expanded=True)