Update pages/NeuralAgents.py

pages/NeuralAgents.py

@@ -6,16 +6,17 @@ import asyncio
 import time
 import conteneiro
 import streamlit as st
-from AgentGPT import AgentsGPT
+from io import StringIO
 from agents import Llama2, Copilot, ChatGPT, Claude3, ForefrontAI, Flowise, Chaindesk, CharacterAI
 
 st.set_page_config(layout="wide")
 
-servers = {}
-clients = {}
+servers = []
+clients = []
 inputs = []
 outputs = []
 used_ports = []
+credentials = []
 server_ports = []
 client_ports = []
 
@@ -27,6 +28,10 @@ async def main():
         st.session_state.cseID = ""                        
     if "api_key" not in st.session_state:
         st.session_state.api_key = "" 
+    if "fireworks_api" not in st.session_state:
+        st.session_state.fireworks_api = ""
+    if "anthropicAPI" not in st.session_state:
+        st.session_state.anthropicAPI = ""
     if "server_ports" not in st.session_state:
         st.session_state['server_ports'] = ""
     if "client_ports" not in st.session_state:
@@ -41,45 +46,169 @@ async def main():
         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 = ""    
+        st.session_state.forefront_api = ""
     if "tokenChar" not in st.session_state:
-        st.session_state.tokenChar = ""           
+        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 = "" 
+        st.session_state.character_ID = ""
     if "flow" not in st.session_state:
-        st.session_state.flow = ""        
+        st.session_state.flow = ""
     if "agentID" not in st.session_state:
-        st.session_state.agentID = "" 
+        st.session_state.agentID = ""
+    if "tokens" not in st.session_state:
+        st.session_state.tokens = None
+    if 'credentials' not in st.session_state:
+        st.session_state.credentials = []
     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")
+        d1, d2 = st.columns(2)
+        fire = st.empty()        
+        fore = st.empty()        
+        anthro = st.empty()
+        char = st.empty()        
+        charID = st.empty()        
+        chain = st.empty()        
+        flo = st.empty()        
+
+        if st.session_state.tokens == None:
+            fireworks_api = fire.text_input("Fireworks API")
+            forefront_api = fore.text_input("Forefront AI API")
+            anthropic_api = anthro.text_input("Anthropic API")
+            characterToken = char.text_input("Character.ai personal token")
+            character_ID = charID.text_input("Character ID")
+            chaindeskAgent = chain.text_input("Chaindesk agent ID")
+            flowID = flo.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 d1:
+
+        uploadAPI = st.file_uploader(label="Upload credentials")
+   
+        if uploadAPI is not None:
+            # Read the content of the file into a byte string
+            byte_content = uploadAPI.getvalue()
+            # Decode the byte string into a string
+            string_content = byte_content.decode('utf-8')
+            # Convert the string into a JSON object (list in this case)
+            tokens = json.loads(string_content)
+            st.session_state.tokens = tokens
+            print(tokens)
+            # Now you have a list of tokens, you can append them to credentials
+            credentials.clear()
+            credentials.extend(tokens)
+
+        if uploadAPI is not None:
+
+            data = json.load(uploadAPI)
+            print(data)
+
+            st.session_state.fireworks_api = data["APIfireworks"]
+            
+            if st.session_state.fireworks_api == "":
+                fireworks_api = fire.text_input("Fireworks API key")
+            else:
+                fireworks_api = fire.container(border=True)
+                fireworks_api.write(st.session_state.fireworks_api)
+
+            st.session_state.forefront_api = data["APIforefront"]
+            
+            if  st.session_state.forefront_api == "":
+                forefront_api = fore.text_input("Forefront API key")
+            else:
+                forefront_api = fore.container(border=True)
+                forefront_api.write(st.session_state.forefront_api)
+
+            st.session_state.anthropicAPI = data["APIanthropic"]
+
+            if st.session_state.anthropicAPI == "":
+                anthropic_api = anthro.text_input("Anthropic API key")
+            else:
+                anthropic_api = anthro.container(border=True)
+                anthropic_api.write(st.session_state.anthropicAPI)
+
+            st.session_state.tokenChar = data["TokenCharacter"]
+
+            if st.session_state.tokenChar == "":
+                characterToken = char.text_input("Character.ai user token")
+            else:
+                characterToken = char.container(border=True)
+                characterToken.write(st.session_state.tokenChar)
+
+            st.session_state.character_ID = data["char_ID"]
+
+            if st.session_state.character_ID == "":
+                character_ID = charID.text_input("Your Character ID")
+            else:
+                character_ID = charID.container(border=True)
+                character_ID.write(st.session_state.character_ID)
+
+            st.session_state.agentID = data["chaindeskID"]
+
+            if st.session_state.agentID == "":
+                chaindeskAgent = chain.text_input("Chaindesk agent ID:")
+            else:
+                chaindeskAgent = chain.container(border=True)
+                chaindeskAgent.write(st.session_state.agentID)
+
+            st.session_state.flow = data["FlowiseID"]
+
+            if st.session_state.flow == "":
+                flowID = flo.text_input("Flowise flow ID:")
+            else:
+                flowID = flo.container(border=True)
+                flowID.write(st.session_state.flow)
+   
+    with d2:
+        credentials.clear()
+        st.session_state.fireworks_api = fireworks_api
+        credentials.append(st.session_state.fireworks_api)
+        st.session_state.forefront_api = forefront_api
+        credentials.append(st.session_state.forefront_api)
+        st.session_state.anthropicAPI = anthropic_api
+        credentials.append(st.session_state.anthropicAPI)
+        st.session_state.tokenChar = characterToken
+        credentials.append(st.session_state.tokenChar)
+        st.session_state.character_ID = character_ID
+        credentials.append(st.session_state.character_ID)
+        st.session_state.agentID = chaindeskAgent
+        credentials.append(st.session_state.agentID)
+        st.session_state.flow = flowID
+        credentials.append(st.session_state.flow)
+
+        APItokens = {
+            "APIfireworks": str(st.session_state.fireworks_api),
+            "APIforefront": str(st.session_state.forefront_api),
+            "APIanthropic": str(st.session_state.anthropicAPI),
+            "TokenCharacter": str(st.session_state.tokenChar),
+            "char_ID": str(st.session_state.character_ID),
+            "chaindeskID": str(st.session_state.agentID),
+            "FlowiseID": str(st.session_state.flow)
+        }
+
+        json_credentials = json.dumps(APItokens)
+        print(json_credentials)
+        
+        saveAPI = st.download_button(label="Download API tokens",
+                                    data=json_credentials,
+                                    file_name=f"APItokens.json",
+                                    mime="application/json",
+                                    help="Click to save your API keys")
 
     with c1:
         websocketPort = st.number_input("Websocket server port", min_value=1000, max_value=9999, value=1000)