Upload NeuralAgents.py

pages/NeuralAgents.py

@@ -0,0 +1,492 @@
+import os
+import sqlite3
+import websockets
+import json
+import asyncio
+import time
+import conteneiro
+import streamlit as st
+from AgentGPT import AgentsGPT
+from agents import Llama2, Copilot, ChatGPT, Claude3, ForefrontAI, Flowise, Chaindesk, CharacterAI
+
+st.set_page_config(layout="wide")
+
+servers = {}
+clients = {}
+inputs = []
+outputs = []
+used_ports = []
+server_ports = []
+client_ports = []
+
+async def main():
+
+    if "googleAPI" not in st.session_state:
+        st.session_state.googleAPI = ""        
+    if "cseID" not in st.session_state:
+        st.session_state.cseID = ""                        
+    if "api_key" not in st.session_state:
+        st.session_state.api_key = "" 
+    if "server_ports" not in st.session_state:
+        st.session_state['server_ports'] = ""
+    if "client_ports" not in st.session_state:
+        st.session_state['client_ports'] = ""
+    if "servers" not in st.session_state:
+        st.session_state['servers'] = ""
+    if "clients" not in st.session_state:
+        st.session_state['clients'] = ""
+    if "gradio_Port" not in st.session_state:
+        st.session_state.gradio_Port = "" 
+    if "server" not in st.session_state:
+        st.session_state.server = False    
+    if "client" not in st.session_state:
+        st.session_state.client = False
+    if "api_key" not in st.session_state:
+        st.session_state.api_key = ""
+    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 "forefront_api" not in st.session_state:
+        st.session_state.forefront_api = ""    
+    if "tokenChar" not in st.session_state:
+        st.session_state.tokenChar = ""           
+    if "charName" not in st.session_state:
+        st.session_state.charName = ""
+    if "character_ID" not in st.session_state:
+        st.session_state.character_ID = "" 
+    if "flow" not in st.session_state:
+        st.session_state.flow = ""        
+    if "agentID" not in st.session_state:
+        st.session_state.agentID = "" 
+    if "server_state" not in st.session_state:
+        st.session_state.server_state = "complete"
+    if "client_state" not in st.session_state:
+        st.session_state.client_state = "complete"    
+
+    userInput = st.chat_input("Ask Agent")
+
+    with st.expander("Personal API tokens"):
+        fireworks_api = st.text_input("Fireworks API")
+        forefront_api = st.text_input("Forefront AI API")
+        anthropic_api = st.text_input("Anthropic API")
+        characterToken = st.text_input("Character.ai personal token")
+        character_ID = st.text_input("Character ID")
+        chaindeskAgent = st.text_input("Chaindesk agent ID")
+        flowID = st.text_input("Flowise flow ID")
+
+    c1, c2 = st.columns(2)
+
+    selectAgent = st.selectbox("Select agent", ("Llama2", "Copilot", "ChatGPT", "Forefront AI", "Claude-3", "Character.ai", "Chaindesk", "Flowise"))
+
+
+
+    with c1:
+        websocketPort = st.number_input("Websocket server port", min_value=1000, max_value=9999, value=1000)   
+        startServer = st.button("Start server")
+        stopServer = st.button("Stop server")
+        st.text("Servers")
+        serverPorts1 = st.empty()
+        serverPorts = serverPorts1.status(label="websocket servers", state="complete", expanded=False)
+        serverPorts.write(conteneiro.servers)
+    
+    with c2:
+        clientPort = st.number_input("Websocket client port", min_value=1000, max_value=9999, value=1000)
+        runClient = st.button("Start client")
+        stopClient = st.button("Stop client")        
+        st.text("Websocket clients")
+        clientPorts1 = st.empty()
+        clientPorts = clientPorts1.status(label="websocket clients", state="complete", expanded=False)
+        clientPorts.write(conteneiro.clients)
+
+    with st.sidebar:
+        # Wyświetlanie danych, które mogą być modyfikowane na różnych stronach       
+        st.text("Websocket servers")
+        srv_status = st.empty()
+        server_status1 = srv_status.status(label="websocket servers", state="complete", expanded=True)
+        server_status1.write(conteneiro.servers)
+        if st.session_state.server == True:
+            st.session_state.server_state = "running"
+            srv_status.empty()
+            server_status1 = srv_status.status(label="websocket servers", state=st.session_state.server_state, expanded=True)
+            server_status1.write(conteneiro.servers)
+
+        st.text("Websocket clients")
+        cli_status = st.empty()
+        client_status1 = cli_status.status(label="websocket clients", state="complete", expanded=True)
+        client_status1.write(conteneiro.clients)
+        if st.session_state.client == True:
+            st.session_state.client_state = "running"
+            cli_status.empty()
+            client_status1 = cli_status.status(label="websocket clients", state=st.session_state.client_state, expanded=True)
+            client_status1.write(conteneiro.clients)
+
+    if  selectAgent == "Llama2":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = Llama2(fireworks_api)
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name = f"Llama2 client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = Llama2(fireworks_api)             
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name = f"Llama2 server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label=srv_name, state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = Llama2(fireworks_api)              
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+    if  selectAgent == "Copilot":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = Copilot()
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name1 = f"Copilot client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name1, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name1, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = Copilot()
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name1 = f"Copilot server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label="websocket servers", state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label="websocket servers", state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = Copilot()
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+    if  selectAgent == "ChatGPT":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = ChatGPT()
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name2 = f"GPT-3,5 client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name2, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name2, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = ChatGPT()         
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name2 = f"GPT-3,5 server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label=srv_name2, state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label=srv_name2, state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = ChatGPT()
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+    if  selectAgent == "Forefront AI":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = ForefrontAI(forefront_api)
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name = f"Forefront AI client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = ForefrontAI(forefront_api)        
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name = f"Forefroont AI server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label=srv_name, state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = ForefrontAI(forefront_api)
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+    if  selectAgent == "Claude-3":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = Claude3(anthropic_api)
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name = f"Claude-3 client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = Claude3(anthropic_api)
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name = f"Claude-3 server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label=srv_name, state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = Claude3(anthropic_api)        
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+    if  selectAgent == "Character.ai":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = CharacterAI(characterToken, character_ID)
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name = f"Claude-3 client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = CharacterAI(characterToken, character_ID)
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name = f"Claude-3 server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label=srv_name, state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = CharacterAI(characterToken, character_ID)    
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+    if  selectAgent == "Chaindesk":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = Chaindesk(chaindeskAgent)   
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name = f"Claude-3 client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = Chaindesk(chaindeskAgent)
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name = f"Claude-3 server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label=srv_name, state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = Chaindesk(chaindeskAgent)   
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+    if  selectAgent == "Flowise":
+
+        if userInput:
+            print(f"User B: {userInput}")
+            user_input = st.chat_message("human")
+            user_input.markdown(userInput)
+            agent = Flowise(flowID)
+            results = await agent.handleInput(userInput)
+            print(results)
+
+        if runClient:
+            st.session_state.client = True
+            cli_name = f"Claude-3 client port: {clientPort}"
+            cli_status.empty()
+            client_status1 = cli_status.status(label=cli_name, state="running", expanded=True)
+            client_status1.write(conteneiro.clients)
+            clientPorts1.empty()            
+            clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
+            clientPorts.write(conteneiro.clients)            
+            try:
+                client = Flowise(flowID)
+                await client.startClient(clientPort)
+                print(f"Connecting client on port {clientPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")              
+
+        if startServer:
+            st.session_state.server = True
+            srv_name = f"Claude-3 server port: {websocketPort}"
+            srv_status.empty()
+            server_status1 = srv_status.status(label=srv_name, state="running", expanded=True)
+            server_status1.write(conteneiro.servers)
+            serverPorts1.empty()
+            serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
+            serverPorts.write(conteneiro.servers)            
+            try:
+                server = Flowise(flowID)
+                await server.start_server(websocketPort)
+                print(f"Starting WebSocket server on port {websocketPort}...")
+                await asyncio.Future()
+
+            except Exception as e:
+                print(f"Error: {e}")
+
+asyncio.run(main())