Update pages/NeuralAgents.py

pages/NeuralAgents.py

@@ -165,7 +165,7 @@ async def main():
             fireworks_api = fire.text_input("Fireworks API key")
         else:
             fireworks_api = fire.container(border=True)
-            fireworks_api.write(st.session_state.fireworks_api)
+            fireworks_api.write(str(st.session_state.fireworks_api))
 
         st.session_state.forefront_api = data["APIforefront"]
         
@@ -173,7 +173,7 @@ async def main():
             forefront_api = fore.text_input("Forefront API key")
         else:
             forefront_api = fore.container(border=True)
-            forefront_api.write(st.session_state.forefront_api)
+            forefront_api.write(str(st.session_state.forefront_api))
 
         st.session_state.anthropicAPI = data["APIanthropic"]
 
@@ -181,7 +181,7 @@ async def main():
             anthropic_api = anthro.text_input("Anthropic API key")
         else:
             anthropic_api = anthro.container(border=True)
-            anthropic_api.write(st.session_state.anthropicAPI)
+            anthropic_api.write(str(st.session_state.anthropicAPI))
 
         st.session_state.tokenChar = data["TokenCharacter"]
 
@@ -189,7 +189,7 @@ async def main():
             characterToken = char.text_input("Character.ai user token")
         else:
             characterToken = char.container(border=True)
-            characterToken.write(st.session_state.tokenChar)
+            characterToken.write(str(st.session_state.tokenChar))
 
         st.session_state.character_ID = data["char_ID"]
 
@@ -197,7 +197,7 @@ async def main():
             character_ID = charID.text_input("Your Character ID")
         else:
             character_ID = charID.container(border=True)
-            character_ID.write(st.session_state.character_ID)
+            character_ID.write(str(st.session_state.character_ID))
 
         st.session_state.agentID = data["chaindeskID"]
 
@@ -205,7 +205,7 @@ async def main():
             chaindeskAgent = chain.text_input("Chaindesk agent ID:")
         else:
             chaindeskAgent = chain.container(border=True)
-            chaindeskAgent.write(st.session_state.agentID)
+            chaindeskAgent.write(str(st.session_state.agentID))
 
         st.session_state.flow = data["FlowiseID"]
 
@@ -213,7 +213,7 @@ async def main():
             flowID = flo.text_input("Flowise flow ID:")
         else:
             flowID = flo.container(border=True)
-            flowID.write(st.session_state.flow)
+            flowID.write(str(st.session_state.flow))
 
     if saveAPI:
         credentials.clear()
@@ -232,14 +232,13 @@ async def main():
         st.session_state.flow = flowID
         credentials.append(st.session_state.flow)
 
-
     if  selectAgent == "Llama2":
 
         if userInput:
             print(f"User B: {userInput}")
             user_input = st.chat_message("human")
             user_input.markdown(userInput)
-            agent = Llama2(fireworks_api)
+            agent = Llama2(str(st.session_state.fireworks_api))
             results = await agent.handleInput(userInput)
             print(results)
 
@@ -253,7 +252,7 @@ async def main():
             clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
             clientPorts.write(conteneiro.clients)            
             try:
-                client = Llama2(fireworks_api)             
+                client = Llama2(str(st.session_state.fireworks_api))             
                 await client.startClient(clientPort)
                 print(f"Connecting client on port {clientPort}...")
                 await asyncio.Future()
@@ -271,7 +270,7 @@ async def main():
             serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
             serverPorts.write(conteneiro.servers)            
             try:
-                server = Llama2(fireworks_api)              
+                server = Llama2(str(st.session_state.fireworks_api))              
                 await server.start_server(websocketPort)
                 print(f"Starting WebSocket server on port {websocketPort}...")
                 await asyncio.Future()
@@ -377,7 +376,7 @@ async def main():
             print(f"User B: {userInput}")
             user_input = st.chat_message("human")
             user_input.markdown(userInput)
-            agent = ForefrontAI(forefront_api)
+            agent = ForefrontAI(str(st.session_state.forefront_api))
             results = await agent.handleInput(userInput)
             print(results)
 
@@ -391,7 +390,7 @@ async def main():
             clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
             clientPorts.write(conteneiro.clients)            
             try:
-                client = ForefrontAI(forefront_api)        
+                client = ForefrontAI(str(st.session_state.forefront_api))        
                 await client.startClient(clientPort)
                 print(f"Connecting client on port {clientPort}...")
                 await asyncio.Future()
@@ -409,7 +408,7 @@ async def main():
             serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
             serverPorts.write(conteneiro.servers)            
             try:
-                server = ForefrontAI(forefront_api)
+                server = ForefrontAI(str(st.session_state.forefront_api))
                 await server.start_server(websocketPort)
                 print(f"Starting WebSocket server on port {websocketPort}...")
                 await asyncio.Future()
@@ -423,7 +422,7 @@ async def main():
             print(f"User B: {userInput}")
             user_input = st.chat_message("human")
             user_input.markdown(userInput)
-            agent = Claude3(anthropic_api)
+            agent = Claude3(str(st.session_state.anthropicAPI))
             results = await agent.handleInput(userInput)
             print(results)
 
@@ -437,7 +436,7 @@ async def main():
             clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
             clientPorts.write(conteneiro.clients)            
             try:
-                client = Claude3(anthropic_api)
+                client = Claude3(str(st.session_state.anthropicAPI))
                 await client.startClient(clientPort)
                 print(f"Connecting client on port {clientPort}...")
                 await asyncio.Future()
@@ -455,7 +454,7 @@ async def main():
             serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
             serverPorts.write(conteneiro.servers)            
             try:
-                server = Claude3(anthropic_api)        
+                server = Claude3(str(st.session_state.anthropicAPI))        
                 await server.start_server(websocketPort)
                 print(f"Starting WebSocket server on port {websocketPort}...")
                 await asyncio.Future()
@@ -465,11 +464,14 @@ async def main():
 
     if  selectAgent == "Character.ai":
 
+        usrToken = str(st.session_state.tokenChar)
+        charID = str(st.session_state.character_ID)
+
         if userInput:
             print(f"User B: {userInput}")
             user_input = st.chat_message("human")
             user_input.markdown(userInput)
-            agent = CharacterAI(characterToken, character_ID)
+            agent = CharacterAI(usrToken, charID)
             results = await agent.handleInput(userInput)
             print(results)
 
@@ -483,7 +485,7 @@ async def main():
             clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
             clientPorts.write(conteneiro.clients)            
             try:
-                client = CharacterAI(characterToken, character_ID)
+                client = CharacterAI(usrToken, charID)
                 await client.startClient(clientPort)
                 print(f"Connecting client on port {clientPort}...")
                 await asyncio.Future()
@@ -501,7 +503,7 @@ async def main():
             serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
             serverPorts.write(conteneiro.servers)            
             try:
-                server = CharacterAI(characterToken, character_ID)    
+                server = CharacterAI(usrToken, charID)
                 await server.start_server(websocketPort)
                 print(f"Starting WebSocket server on port {websocketPort}...")
                 await asyncio.Future()
@@ -515,7 +517,7 @@ async def main():
             print(f"User B: {userInput}")
             user_input = st.chat_message("human")
             user_input.markdown(userInput)
-            agent = Chaindesk(chaindeskAgent)   
+            agent = Chaindesk(str(st.session_state.agentID))   
             results = await agent.handleInput(userInput)
             print(results)
 
@@ -529,7 +531,7 @@ async def main():
             clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
             clientPorts.write(conteneiro.clients)            
             try:
-                client = Chaindesk(chaindeskAgent)
+                client = Chaindesk(str(st.session_state.agentID))
                 await client.startClient(clientPort)
                 print(f"Connecting client on port {clientPort}...")
                 await asyncio.Future()
@@ -547,7 +549,7 @@ async def main():
             serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
             serverPorts.write(conteneiro.servers)            
             try:
-                server = Chaindesk(chaindeskAgent)   
+                server = Chaindesk(str(st.session_state.agentID))
                 await server.start_server(websocketPort)
                 print(f"Starting WebSocket server on port {websocketPort}...")
                 await asyncio.Future()
@@ -561,7 +563,7 @@ async def main():
             print(f"User B: {userInput}")
             user_input = st.chat_message("human")
             user_input.markdown(userInput)
-            agent = Flowise(flowID)
+            agent = Flowise(str(st.session_state.flow))
             results = await agent.handleInput(userInput)
             print(results)
 
@@ -575,7 +577,7 @@ async def main():
             clientPorts = clientPorts1.status(label=cli_name, state="running", expanded=True)
             clientPorts.write(conteneiro.clients)            
             try:
-                client = Flowise(flowID)
+                client = Flowise(str(st.session_state.flow))
                 await client.startClient(clientPort)
                 print(f"Connecting client on port {clientPort}...")
                 await asyncio.Future()
@@ -593,7 +595,7 @@ async def main():
             serverPorts = serverPorts1.status(label=srv_name, state="running", expanded=True)
             serverPorts.write(conteneiro.servers)            
             try:
-                server = Flowise(flowID)
+                server = Flowise(str(st.session_state.flow))
                 await server.start_server(websocketPort)
                 print(f"Starting WebSocket server on port {websocketPort}...")
                 await asyncio.Future()