Update clientCharacter.py

clientCharacter.py

@@ -1,20 +1,54 @@
+import json
 import asyncio
 import websockets
 import threading
 import sqlite3
+import requests
 import streamlit as st
-from PyCharacterAI import Client
+
+client_ports = []
 
 # Define the websocket client class
-class WebSocketClient2:
+class WebSocketClient6:
     def __init__(self, clientPort):
+        # Initialize the uri attribute
+        self.uri = f'ws://localhost:{clientPort}'   
+        self.name = f"Chaindesk client port: {clientPort}"
+        self.status = st.sidebar.status(label=self.name, state="complete", expanded=False)
+        st.session_state.clientPort = clientPort
+
+        if "client_ports" not in st.session_state:
+            st.session_state['client_ports'] = ""
+        
+    async def askChaindesk(self, question):
+        
+        if "agentID" not in st.session_state:
+            st.session_state.agentID = ""     
+
+        id = st.session_state.agentID
 
-        self.uri = f'ws://localhost:{clientPort}'
+        url = f"https://api.chaindesk.ai/agents/{id}/query"
+        
+        payload = {
+            "query": question
+        }
+        
+        headers = {
+            "Authorization": "Bearer fe77e704-bc5a-4171-90f2-9d4b0d4ac942",
+            "Content-Type": "application/json"
+        }
+        try:            
+            response = requests.request("POST", url, json=payload, headers=headers)
+            print(response.text)
+            response_text = response.text
+            responseJson = json.loads(response_text)
+            answer = responseJson["answer"]
 
-        if "tokenChar" not in st.session_state:
-            st.session_state.tokenChar = ""
-        if "character_ID" not in st.session_state:
-            st.session_state.character_ID = ""         
+            print(response.text)
+            return answer
+
+        except Exception as e:
+            print(e)
 
     # Define a function that will run the client in a separate thread
     def run(self):
@@ -32,31 +66,35 @@ class WebSocketClient2:
         # Run the client until it is stopped
         loop.run_until_complete(self.client())
 
+    # Stop the WebSocket client
+    async def stop_client():
+        global ws
+        # Close the connection with the server
+        await ws.close()
+        client_ports.pop()
+        print("Stopping WebSocket client...")    
+
     # Define a coroutine that will connect to the server and exchange messages
     async def startClient(self):
-        status = st.sidebar.status(label="runs", state="complete", expanded=False)
-        client = Client()
-        await client.authenticate_with_token(st.session_state.tokenChar)
-        chat = await client.create_or_continue_chat(st.session_state.character_ID)
+        
+        status = self.status       
         # Connect to the server
         async with websockets.connect(self.uri) as websocket:
             # Loop forever
             while True:
-                status.update(label="runs", state="running", expanded=True)
+                status.update(label=self.name, state="running", expanded=True)               
                 # Listen for messages from the server
                 input_message = await websocket.recv()
                 print(f"Server: {input_message}")
                 input_Msg = st.chat_message("assistant")
                 input_Msg.markdown(input_message)
                 try:
-                    answer = await chat.send_message(input_message)
-                    response = f"{answer.src_character_name}: {answer.text}"
-                    print(response)
-                    outputMsg1 = st.chat_message("ai")
-                    outputMsg1.markdown(response)
-                    await websocket.send(response)
-                    status.update(label="runs", state="complete", expanded=True)
-                    continue  
+                    response = await self.askChaindesk(input_message)
+                    res1 = f"Client: {response}"
+                    output_Msg = st.chat_message("ai")
+                    output_Msg.markdown(res1)
+                    await websocket.send(res1)
+                    status.update(label=self.name, state="complete", expanded=True)
 
                 except websockets.ConnectionClosed:
                     print("client disconnected")