1 files changed, 32 insertions, 16 deletions

diff --git a/unfinished/gptbz/__init__.py b/unfinished/gptbz/__init__.py
index ba326cf6..e95d5716 100644
--- a/unfinished/gptbz/__init__.py
+++ b/unfinished/gptbz/__init__.py
@@ -1,30 +1,46 @@
-import websockets
 from json import dumps, loads
 
+import websockets
+
+
+# Define the asynchronous function to test the WebSocket connection
+
+
 async def test():
+    # Establish a WebSocket connection with the specified URL
     async with websockets.connect('wss://chatgpt.func.icu/conversation+ws') as wss:
-        
-        await wss.send(dumps(separators=(',', ':'), obj = {
-            'content_type':'text',
-            'engine':'chat-gpt',
-            'parts':['hello world'],
-            'options':{}
-            }
-        ))
-        
+
+        # Prepare the message payload as a JSON object
+        payload = {
+            'content_type': 'text',
+            'engine': 'chat-gpt',
+            'parts': ['hello world'],
+            'options': {}
+        }
+
+        # Send the payload to the WebSocket server
+        await wss.send(dumps(obj=payload, separators=(',', ':')))
+
+        # Initialize a variable to track the end of the conversation
         ended = None
 
+        # Continuously receive and process messages until the conversation ends
         while not ended:
             try:
-                response      = await wss.recv()
+                # Receive and parse the JSON response from the server
+                response = await wss.recv()
                 json_response = loads(response)
+
+                # Print the entire JSON response
                 print(json_response)
-                
-                ended         = json_response.get('eof')
-                
+
+                # Check for the end of the conversation
+                ended = json_response.get('eof')
+
+                # If the conversation has not ended, print the received message
                 if not ended:
                     print(json_response['content']['parts'][0])
-                
+
+            # Handle cases when the connection is closed by the server
             except websockets.ConnectionClosed:
                 break
-