Update pages/Connect.py

pages/Connect.py

@@ -1,3 +1,4 @@
+import os
 import g4f
 import json
 import websockets
@@ -18,11 +19,12 @@ from clientG4F import WebSocketClient1
 from forefront import ForefrontClient
 from clientG4F2 import WebSocketClient3
 from ServFlowise import WebSocketServer5
+from ServAgentGPT import WebSocketServer7
 from clientFlowise import WebSocketClient5
 from ServForefront import WebSocketServer4
 from ServChaindesk import WebSocketServer6
-from clientChaindesk import WebSocketClient6
 from PyCharacterAI import Client
+from clientChaindesk import WebSocketClient6
 from clientForefront import WebSocketClient4
 from clientFireworks import WebSocketClient
 from clientCharacter import WebSocketClient2
@@ -38,6 +40,9 @@ used_ports = []
 server_ports = []
 client_ports = []
 
+FIREWORKS_API_KEY = os.getenv("FIREWORKS_API_KEY")
+fireworks_api_key = os.getenv("FIREWORKS_API_KEY")
+
 # Stop the WebSocket server
 async def stop_websockets():    
     global server
@@ -60,8 +65,7 @@ async def stop_client():
 async def main():
        
     st.set_page_config(layout="wide")
-    st.title("serverovnia")
-        
+    st.title("serverovnia")        
 
     if "server_ports" not in st.session_state:
         st.session_state['server_ports'] = ""
@@ -79,8 +83,6 @@ async def main():
         st.session_state['server_ports'] = ""
     if "client_ports" not in st.session_state:
         st.session_state['client_ports'] = ""
-    if "user_ID" not in st.session_state:
-        st.session_state.user_ID = ""
     if "gradio_Port" not in st.session_state:
         st.session_state.gradio_Port = "" 
     if "servers" not in st.session_state:
@@ -102,7 +104,11 @@ async def main():
     if "flow" not in st.session_state:
         st.session_state.flow = ""        
     if "agentID" not in st.session_state:
-        st.session_state.agentID = ""   
+        st.session_state.agentID = "" 
+    if "googleAPI" not in st.session_state:
+        st.session_state.googleAPI = ""        
+    if "cseID" not in st.session_state:
+        st.session_state.cseID = ""                        
         
     if "http_server" not in st.session_state:
         
@@ -117,7 +123,7 @@ async def main():
 
     userInput = st.chat_input("Ask agent")    
 
-    selectServ = st.selectbox("Select source", ("Fireworks", "Bing", "GPT-3,5", "character.ai", "Forefront", "ChainDesk", "Flowise", "DocsBot"))
+    selectServ = st.selectbox("Select source", ("Fireworks", "Bing", "GPT-3,5", "character.ai", "Forefront", "AgentsGPT", "ChainDesk", "Flowise", "DocsBot"))
 
     c1, c2 = st.columns(2)
         
@@ -165,15 +171,17 @@ async def main():
         fireworksAPI = st.text_input("Fireworks API") 
 
         if startServer:
-            st.session_state.api_key = fireworksAPI
+            fireworks.client.api_key = fireworksAPI
+            st.session_state.api_key = fireworks.client.api_key
             server_ports.append(websocketPort)
             st.session_state.server = True
             st.session_state['server_ports'] = server_ports
             serverPorts1.markdown(st.session_state['server_ports'])
             try:
-                server = WebSocketServer("localhost", websocketPort)    
+                server = WebSocketServer("localhost")
+                servers.append(server)
                 print(f"Starting WebSocket server on port {websocketPort}...")
-                await server.start_server()
+                await server.start_server(websocketPort)
                 status.update(label="runs", state="running", expanded=True)
                 await asyncio.Future()
 
@@ -204,8 +212,8 @@ async def main():
             st.session_state.api_key = fireworksAPI
             user_input = st.chat_message("human")
             user_input.markdown(userInput)
-            fireworks = WebSocketServer("localhost", websocketPort)
-            response1 = await fireworks.chatCompletion(userInput)
+            fireworks1 = WebSocketServer()
+            response1 = await fireworks1.chatCompletion(userInput)
             print(response1)
             outputMsg = st.chat_message("ai") 
             outputMsg.markdown(response1)
@@ -234,8 +242,7 @@ async def main():
             st.session_state['client_ports'] = client_ports
             clientPorts1.markdown(st.session_state['client_ports'])
             try:
-                uri = f'ws://localhost:{clientPort}'
-                client1 = WebSocketClient1(uri)    
+                client1 = WebSocketClient1(clientPort)    
                 print(f"Connecting client on port {clientPort}...")
                 await client1.startClient()
                 st.session_state.client = client1
@@ -276,8 +283,7 @@ async def main():
             st.session_state['client_ports'] = client_ports
             clientPorts1.markdown(st.session_state['client_ports'])
             try:
-                uri = f'ws://localhost:{clientPort}'
-                client2 = WebSocketClient3(uri)    
+                client2 = WebSocketClient3(clientPort)    
                 print(f"Connecting client on port {clientPort}...")
                 await client2.startClient()
                 st.session_state.client = client2
@@ -294,6 +300,53 @@ async def main():
             outputMsg2 = st.chat_message("ai") 
             outputMsg2.markdown(response)
 
+    if selectServ == "character.ai":
+
+        characterToken = st.text_input("Character AI user token") 
+        characterID = st.text_input("Your characters ID") 
+
+        if startServer:
+            st.session_state.tokenChar = characterToken
+            st.session_state.character_ID = characterID
+            server_ports.append(websocketPort)
+            st.session_state.server = True
+            st.session_state['server_ports'] = server_ports
+            serverPorts1.markdown(st.session_state['server_ports'])
+            try:
+                server = WebSocketServer2(characterToken, characterID)    
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await server.start_server(websocketPort)
+                status.update(label="runs", state="running", expanded=True)
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+        if runClient:
+            st.session_state.tokenChar = characterToken
+            st.session_state.character_ID = characterID
+            client_ports.append(clientPort)
+            st.session_state['client_ports'] = client_ports
+            clientPorts1.markdown(st.session_state['client_ports'])
+            try:
+                uri = f'ws://localhost:{clientPort}'
+                client = WebSocketClient2(clientPort)    
+                print(f"Connecting client on port {clientPort}...")
+                await client.startClient()
+                st.session_state.client = client
+                status.update(label="runs", state="running", expanded=True)
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")                
+
+        if userInput:
+            print(f"User B: {userInput}")
+            character = WebSocketServer2(characterToken, characterID)
+            response1 = await character.askCharacter(userInput)
+            print(response1)
+            return response1
+
     if selectServ == "ChainDesk":
 
         agentID = st.text_input("Agent ID")
@@ -332,7 +385,7 @@ async def main():
             try:
                 client6 = WebSocketClient6(clientPort)    
                 print(f"Connecting client on port {clientPort}...")
-                await client6.startClient(clientPort)
+                await client6.startClient()
                 st.session_state.client = client6
                 await asyncio.Future()