Update pages/NeuralGPT.py

pages/NeuralGPT.py

@@ -15,7 +15,10 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
+st.session_state.update(st.session_state)
+
 servers = {}
+clients = {}
 inputs = []
 outputs = []
 used_ports = []
@@ -332,39 +335,47 @@ async def stop_client():
     print("Stopping WebSocket client...")
 
 async def main():
-    if "server" not in st.session_state:
-        st.session_state.server = None
+
+    st.session_state.update(st.session_state)
+
+    if 'active_page' not in st.session_state:
+        st.session_state.active_page = 'NeuralGPT'
+    if "sidebar" not in st.session_state:
+        st.session_state.sidebar = True
+    if "server_ports" not in st.session_state:
+        st.session_state['server_ports'] = False
+    if "client_ports" not in st.session_state:
+        st.session_state['client_ports'] = False
+    if "user_id" not in st.session_state:
+        st.session_state.user_id = False
     if "servers" not in st.session_state:
         st.session_state.servers = None
-    if "client" not in st.session_state:
-        st.session_state.client = None    
-    if "server_Ports" not in st.session_state:
-        st.session_state.server_Ports = None
-    if "client_Ports" not in st.session_state:
-        st.session_state.client_Ports = None  
+    if "clients" not in st.session_state:
+        st.session_state.clients = None    
     if "api_key" not in st.session_state:
         st.session_state.api_key = None          
     if "userID" not in st.session_state:
-        st.session_state.user_ID = None
+        st.session_state.userID = None
 
-    st.sidebar.text("Server ports:")
+    st.text("Server ports:")
     serverPorts = st.sidebar.container(border=True)
-    serverPorts.markdown(st.session_state.server_Ports)
+    serverPorts.markdown(st.session_state['server_ports'])
     st.sidebar.text("Client ports")
     clientPorts = st.sidebar.container(border=True)
-    clientPorts.markdown(st.session_state.client_Ports)
+    clientPorts.markdown(st.session_state['client_ports'])
+    st.sidebar.text("Charavter.ai ID")
     user_id = st.sidebar.container(border=True)
-    user_id.markdown(st.session_state.user_ID)
+    user_id.markdown(st.session_state.userID)
 
     c1, c2 = st.columns(2)
     
     with c1:        
         serverPorts1 = st.container(border=True)
-        serverPorts1.markdown(st.session_state.server_Ports)        
+        serverPorts1.markdown(st.session_state['server_ports'])        
     
     with c2:
         clientPorts1 = st.container(border=True)
-        clientPorts1.markdown(st.session_state.client_Ports)        
+        clientPorts1.markdown(st.session_state['client_ports'])        
     
     tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs(["Fireworks", "GPT4Free", "Character.ai", "Chaindesk agent", "Docsbot Wordpress-agent", "Flowise client"])
 
@@ -378,13 +389,13 @@ async def main():
             websocket_Port = st.number_input("Server port", 1000)
             startServer = st.button('Start websocket server 1')
             srvr_ports = st.container(border=True)
-            srvr_ports.markdown(st.session_state.server_Ports)
+            srvr_ports.markdown(st.session_state['server_ports'])
         
         with row1_col2:            
             client_Port = st.number_input("Client port", 1000)
             startClient = st.button('Connect client to server 1')
             cli_ports = st.container(border=True)
-            cli_ports.markdown(st.session_state.client_Ports)
+            cli_ports.markdown(st.session_state['client_ports'])
 
         if userInputt:        
             print(f"User B: {userInputt}")
@@ -399,14 +410,14 @@ async def main():
 
         if startServer:
             fireworks.client.api_key = fireworksAPI
-            st.session_state.api_key = fireworks.client.api_key
-            server_ports.append(websocket_Port)
-            st.session_state.server_Ports = server_ports
-            serverPorts.markdown(st.session_state.server_Ports)
-            serverPorts1.markdown(st.session_state.server_Ports)
-            srvr_ports.markdown(st.session_state.server_Ports)
+            st.session_state.api_key = fireworks.client.api_key            
+            st.session_state['server_ports'] = server_ports.append(websocket_Port)
+            serverPorts.markdown(st.session_state['server_ports'])
+            serverPorts1.markdown(st.session_state['server_ports'])
+            srvr_ports.markdown(st.session_state['server_ports'])
             try:
                 server = await websockets.serve(handleServer, 'localhost', websocket_Port)
+                st.session_state.servers = servers[server]
                 print(f"Launching server at port: {websocket_Port}")
                 while True:
                     await server.wait_closed()
@@ -415,15 +426,16 @@ async def main():
 
 
         if startClient:
-            client_ports.append(client_Port)
-            clientPorts.markdown(client_ports)
-            clientPorts1.markdown(client_ports)
-            cli_ports.markdown(client_ports)
-            await client.authenticate_with_token(fireworksAPI)           
+            fireworks.client.api_key = fireworksAPI
+            st.session_state.api_key = fireworks.client.api_key
+            st.session_state['client_ports'] = client_ports.append(client_Port)
+            clientPorts.markdown(st.session_state['client_ports'])
+            clientPorts1.markdown(st.session_state['client_ports'])
+            cli_ports.markdown(st.session_state['client_ports'])
+            print(f"Connecting to server at port: {client_Port}...")          
             uri = f'ws://localhost:{client_Port}'
             async with websockets.connect(uri) as websocket:
-                while True:
-                    print(f"Connecting to server at port: {client_Port}...")
+                while True:                    
                     # Listen for messages from the server
                     input_message = await websocket.recv()
                     print(f"Server: {input_message}")