Update pages/NeuralAgents.py

pages/NeuralAgents.py

@@ -71,17 +71,35 @@ async def main():
     if "client_state" not in st.session_state:
         st.session_state.client_state = "complete"    
      
+    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)
+
     userInput = st.chat_input("Ask Agent")
 
     with st.expander("Personal API tokens"):
-        d1, d2 = st.columns(2)
+        uploadAPI = st.file_uploader(label="Upload credentials")
         fire = st.empty()        
         fore = st.empty()        
         anthro = st.empty()
         char = st.empty()        
         charID = st.empty()        
         chain = st.empty()        
-        flo = st.empty()        
+        flo = st.empty()
+        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")
 
         if st.session_state.tokens == None:
             fireworks_api = fire.text_input("Fireworks API")
@@ -95,121 +113,7 @@ async def main():
     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)   
         startServer = st.button("Start server")
@@ -250,6 +154,85 @@ async def main():
             client_status1 = cli_status.status(label="websocket clients", state=st.session_state.client_state, expanded=True)
             client_status1.write(conteneiro.clients)
 
+    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)
+
+    if saveAPI:
+        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)
+
+
     if  selectAgent == "Llama2":
 
         if userInput: