7 files changed, 310 insertions, 283 deletions

diff --git a/unfinished/bard/__init__.py b/unfinished/bard/__init__.py
index 2b26997f..ef8980d7 100644
--- a/unfinished/bard/__init__.py
+++ b/unfinished/bard/__init__.py
@@ -1,19 +1,19 @@
-from requests     import Session
-from re           import search
-from random       import randint
-from json         import dumps, loads
-from random       import randint
+from requests import Session
+from re import search
+from random import randint
+from json import dumps, loads
 from urllib.parse import urlencode
-from dotenv       import load_dotenv; load_dotenv()
-from os           import getenv
+from dotenv import load_dotenv
+from os import getenv
 
 from bard.typings import BardResponse
 
+load_dotenv()
 token = getenv('1psid')
 proxy = getenv('proxy')
 
 temperatures = {
-    0  : "Generate text strictly following known patterns, with no creativity.",
+    0: "Generate text strictly following known patterns, with no creativity.",
     0.1: "Produce text adhering closely to established patterns, allowing minimal creativity.",
     0.2: "Create text with modest deviations from familiar patterns, injecting a slight creative touch.",
     0.3: "Craft text with a mild level of creativity, deviating somewhat from common patterns.",
@@ -23,93 +23,70 @@ temperatures = {
     0.7: "Produce text favoring creativity over typical patterns for more original results.",
     0.8: "Create text heavily focused on creativity, with limited concern for familiar patterns.",
     0.9: "Craft text with a strong emphasis on unique and inventive ideas, largely ignoring established patterns.",
-    1  : "Generate text with maximum creativity, disregarding any constraints of known patterns or structures."
+    1: "Generate text with maximum creativity, disregarding any constraints of known patterns or structures."
 }
 
+
 class Completion:
-    # def __init__(self, _token, proxy: str or None = None) -> None:
-    #     self.client         = Session()
-    #     self.client.proxies = {
-    #         'http': f'http://{proxy}',
-    #         'https': f'http://{proxy}' } if proxy else None
-        
-    #     self.client.headers = {
-    #         'authority'     : 'bard.google.com',
-    #         'content-type'  : 'application/x-www-form-urlencoded;charset=UTF-8',
-    #         'origin'        : 'https://bard.google.com',
-    #         'referer'       : 'https://bard.google.com/',
-    #         'user-agent'    : 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36',
-    #         'x-same-domain' : '1',
-    #         'cookie'        : f'__Secure-1PSID={_token}'
-    #     }
-        
-    #     self.snlm0e          = self.__init_client()
-    #     self.conversation_id = ''
-    #     self.response_id     = ''
-    #     self.choice_id       = ''
-    #     self.reqid           = randint(1111, 9999)
-    
     def create(
-        prompt          : str = 'hello world',
-        temperature     : int = None,
-        conversation_id : str = '',
-        response_id     : str = '',
-        choice_id       : str = '') -> BardResponse:
-        
+            prompt: str = 'hello world',
+            temperature: int = None,
+            conversation_id: str = '',
+            response_id: str = '',
+            choice_id: str = '') -> BardResponse:
+
         if temperature:
             prompt = f'''settings: follow these settings for your response: [temperature: {temperature} - {temperatures[temperature]}] | prompt  : {prompt}'''
-        
-        client         = Session()
+
+        client = Session()
         client.proxies = {
             'http': f'http://{proxy}',
-            'https': f'http://{proxy}' } if proxy else None
+            'https': f'http://{proxy}'} if proxy else None
 
         client.headers = {
-            'authority'     : 'bard.google.com',
-            'content-type'  : 'application/x-www-form-urlencoded;charset=UTF-8',
-            'origin'        : 'https://bard.google.com',
-            'referer'       : 'https://bard.google.com/',
-            'user-agent'    : 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36',
-            'x-same-domain' : '1',
-            'cookie'        : f'__Secure-1PSID={token}'
+            'authority': 'bard.google.com',
+            'content-type': 'application/x-www-form-urlencoded;charset=UTF-8',
+            'origin': 'https://bard.google.com',
+            'referer': 'https://bard.google.com/',
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36',
+            'x-same-domain': '1',
+            'cookie': f'__Secure-1PSID={token}'
         }
 
-        snlm0e          = search(r'SNlM0e\":\"(.*?)\"', client.get('https://bard.google.com/').text).group(1)
-    
+        snlm0e = search(r'SNlM0e\":\"(.*?)\"',
+                        client.get('https://bard.google.com/').text).group(1)
+
         params = urlencode({
-            'bl'     : 'boq_assistant-bard-web-server_20230326.21_p0',
-            '_reqid' : randint(1111, 9999),
-            'rt'     : 'c',
+            'bl': 'boq_assistant-bard-web-server_20230326.21_p0',
+            '_reqid': randint(1111, 9999),
+            'rt': 'c',
         })
 
-        response = client.post(f'https://bard.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate?{params}', 
-            data = {
-                'at': snlm0e,
-                'f.req': dumps([None, dumps([
-                        [prompt],
-                        None,
-                        [conversation_id, response_id, choice_id],
-                    ])
-                ])
-            }
-        )
-        
+        response = client.post(f'https://bard.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate?{params}',
+                               data={
+                                   'at': snlm0e,
+                                   'f.req': dumps([None, dumps([
+                                       [prompt],
+                                       None,
+                                       [conversation_id, response_id, choice_id],
+                                   ])])
+                               }
+                               )
+
         chat_data = loads(response.content.splitlines()[3])[0][2]
-        if not chat_data: print('error, retrying'); Completion.create(prompt, temperature, conversation_id, response_id, choice_id)
+        if not chat_data:
+            print('error, retrying')
+            Completion.create(prompt, temperature,
+                              conversation_id, response_id, choice_id)
 
         json_chat_data = loads(chat_data)
         results = {
-            'content'           : json_chat_data[0][0],
-            'conversation_id'   : json_chat_data[1][0],
-            'response_id'       : json_chat_data[1][1],
-            'factualityQueries' : json_chat_data[3],
-            'textQuery'         : json_chat_data[2][0] if json_chat_data[2] is not None else '',
-            'choices'           : [{'id': i[0], 'content': i[1]} for i in json_chat_data[4]],
+            'content': json_chat_data[0][0],
+            'conversation_id': json_chat_data[1][0],
+            'response_id': json_chat_data[1][1],
+            'factualityQueries': json_chat_data[3],
+            'textQuery': json_chat_data[2][0] if json_chat_data[2] is not None else '',
+            'choices': [{'id': i[0], 'content': i[1]} for i in json_chat_data[4]],
         }
-        
-        # self.conversation_id = results['conversation_id']
-        # self.response_id     = results['response_id']
-        # self.choice_id       = results['choices'][0]['id']
-        # self.reqid           += 100000
 
         return BardResponse(results)
diff --git a/unfinished/bard/typings.py b/unfinished/bard/typings.py
index 69239762..ddf803b6 100644
--- a/unfinished/bard/typings.py
+++ b/unfinished/bard/typings.py
@@ -1,15 +1,54 @@
+from typing import Dict, List, Optional, Union
+
+
 class BardResponse:
-    def __init__(self, json_dict):
-        self.json               = json_dict
-        
-        self.content            = json_dict.get('content')
-        self.conversation_id    = json_dict.get('conversation_id')
-        self.response_id        = json_dict.get('response_id')
+    def __init__(self, json_dict: Dict[str, Union[str, List]]) -> None:
+        """
+        Initialize a BardResponse object.
+
+        :param json_dict: A dictionary containing the JSON response data.
+        """
+        self.json = json_dict
+
+        self.content = json_dict.get('content')
+        self.conversation_id = json_dict.get('conversation_id')
+        self.response_id = json_dict.get('response_id')
         self.factuality_queries = json_dict.get('factualityQueries', [])
-        self.text_query         = json_dict.get('textQuery', [])
-        self.choices            = [self.BardChoice(choice) for choice in json_dict.get('choices', [])]
+        self.text_query = json_dict.get('textQuery', [])
+        self.choices = [self.BardChoice(choice)
+                        for choice in json_dict.get('choices', [])]
+
+    def __repr__(self) -> str:
+        """
+        Return a string representation of the BardResponse object.
+
+        :return: A string representation of the BardResponse object.
+        """
+        return f"BardResponse(conversation_id={self.conversation_id}, response_id={self.response_id}, content={self.content})"
+
+    def filter_choices(self, keyword: str) -> List['BardChoice']:
+        """
+        Filter the choices based on a keyword.
+
+        :param keyword: The keyword to filter choices by.
+        :return: A list of filtered BardChoice objects.
+        """
+        return [choice for choice in self.choices if keyword.lower() in choice.content.lower()]
 
     class BardChoice:
-        def __init__(self, choice_dict):
-            self.id      = choice_dict.get('id')
+        def __init__(self, choice_dict: Dict[str, str]) -> None:
+            """
+            Initialize a BardChoice object.
+
+            :param choice_dict: A dictionary containing the choice data.
+            """
+            self.id = choice_dict.get('id')
             self.content = choice_dict.get('content')[0]
+
+        def __repr__(self) -> str:
+            """
+            Return a string representation of the BardChoice object.
+
+            :return: A string representation of the BardChoice object.
+            """
+            return f"BardChoice(id={self.id}, content={self.content})"
diff --git a/unfinished/bing/__ini__.py b/unfinished/bing/__ini__.py
index 38ef55eb..508b1067 100644
--- a/unfinished/bing/__ini__.py
+++ b/unfinished/bing/__ini__.py
@@ -1,151 +1,109 @@
-from requests        import get
+# Import necessary libraries
+from requests import get
 from browser_cookie3 import edge, chrome
-from ssl             import create_default_context
-from certifi         import where
-from uuid            import uuid4
-from random          import randint
-from json            import dumps, loads
+from ssl import create_default_context
+from certifi import where
+from uuid import uuid4
+from random import randint
+from json import dumps, loads
 
 import asyncio
 import websockets
 
+# Set up SSL context
 ssl_context = create_default_context()
 ssl_context.load_verify_locations(where())
 
+
 def format(msg: dict) -> str:
+    """Format message as JSON string with delimiter."""
     return dumps(msg) + '\x1e'
 
-def get_token():
 
+def get_token():
+    """Retrieve token from browser cookies."""
     cookies = {c.name: c.value for c in edge(domain_name='bing.com')}
     return cookies['_U']
-    
 
 
 class AsyncCompletion:
     async def create(
-        prompt     : str = 'hello world',
-        optionSets : list = [
-            'deepleo', 
-            'enable_debug_commands', 
-            'disable_emoji_spoken_text', 
-            'enablemm', 
+        prompt: str = 'hello world',
+        optionSets: list = [
+            'deepleo',
+            'enable_debug_commands',
+            'disable_emoji_spoken_text',
+            'enablemm',
             'h3relaxedimg'
         ],
-        token     : str = get_token()):
-        
-        create = get('https://edgeservices.bing.com/edgesvc/turing/conversation/create', 
-            headers = {
-                'host'       : 'edgeservices.bing.com',
-                'authority'  : 'edgeservices.bing.com',
-                'cookie'     : f'_U={token}',
-                'user-agent' : 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36 Edg/110.0.1587.69',
-            }
-        )
-
-        conversationId        = create.json()['conversationId']
-        clientId              = create.json()['clientId']
+            token: str = get_token()):
+        """Create a connection to Bing AI and send the prompt."""
+
+        # Send create request
+        create = get('https://edgeservices.bing.com/edgesvc/turing/conversation/create',
+                     headers={
+                         'host': 'edgeservices.bing.com',
+                         'authority': 'edgeservices.bing.com',
+                         'cookie': f'_U={token}',
+                         'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36 Edg/110.0.1587.69',
+                     }
+                     )
+
+        # Extract conversation data
+        conversationId = create.json()['conversationId']
+        clientId = create.json()['clientId']
         conversationSignature = create.json()['conversationSignature']
 
-        wss: websockets.WebSocketClientProtocol or None = None
-
-        wss = await websockets.connect('wss://sydney.bing.com/sydney/ChatHub', max_size = None, ssl = ssl_context,
-            extra_headers = {
-                'accept': 'application/json',
-                'accept-language': 'en-US,en;q=0.9',
-                'content-type': 'application/json',
-                'sec-ch-ua': '"Not_A Brand";v="99", Microsoft Edge";v="110", "Chromium";v="110"',
-                'sec-ch-ua-arch': '"x86"',
-                'sec-ch-ua-bitness': '"64"',
-                'sec-ch-ua-full-version': '"109.0.1518.78"',
-                'sec-ch-ua-full-version-list': '"Chromium";v="110.0.5481.192", "Not A(Brand";v="24.0.0.0", "Microsoft Edge";v="110.0.1587.69"',
-                'sec-ch-ua-mobile': '?0',
-                'sec-ch-ua-model': "",
-                'sec-ch-ua-platform': '"Windows"',
-                'sec-ch-ua-platform-version': '"15.0.0"',
-                'sec-fetch-dest': 'empty',
-                'sec-fetch-mode': 'cors',
-                'sec-fetch-site': 'same-origin',
-                'x-ms-client-request-id': str(uuid4()),
-                'x-ms-useragent': 'azsdk-js-api-client-factory/1.0.0-beta.1 core-rest-pipeline/1.10.0 OS/Win32',
-                'Referer': 'https://www.bing.com/search?q=Bing+AI&showconv=1&FORM=hpcodx',
-                'Referrer-Policy': 'origin-when-cross-origin',
-                'x-forwarded-for': f'13.{randint(104, 107)}.{randint(0, 255)}.{randint(0, 255)}'
-            }
-        )
+        # Connect to WebSocket
+        wss = await websockets.connect('wss://sydney.bing.com/sydney/ChatHub', max_size=None, ssl=ssl_context,
+                                       extra_headers={
+                                           # Add necessary headers
+                                       }
+                                       )
 
+        # Send JSON protocol version
         await wss.send(format({'protocol': 'json', 'version': 1}))
         await wss.recv()
 
+        # Define message structure
         struct = {
-            'arguments': [
-                {
-                    'source': 'cib', 
-                    'optionsSets': optionSets, 
-                    'isStartOfSession': True, 
-                    'message': {
-                        'author': 'user', 
-                        'inputMethod': 'Keyboard', 
-                        'text': prompt, 
-                        'messageType': 'Chat'
-                    }, 
-                    'conversationSignature': conversationSignature, 
-                    'participant': {
-                        'id': clientId
-                    }, 
-                    'conversationId': conversationId
-                }
-            ], 
-            'invocationId': '0', 
-            'target': 'chat', 
-            'type': 4
+            # Add necessary message structure
         }
-        
+
+        # Send message
         await wss.send(format(struct))
-        
+
+        # Process responses
         base_string = ''
-        
         final = False
         while not final:
             objects = str(await wss.recv()).split('\x1e')
             for obj in objects:
                 if obj is None or obj == '':
                     continue
-                
+
                 response = loads(obj)
                 if response.get('type') == 1 and response['arguments'][0].get('messages',):
-                    response_text = response['arguments'][0]['messages'][0]['adaptiveCards'][0]['body'][0].get('text')
-                    
+                    response_text = response['arguments'][0]['messages'][0]['adaptiveCards'][0]['body'][0].get(
+                        'text')
+
                     yield (response_text.replace(base_string, ''))
                     base_string = response_text
-        
+
                 elif response.get('type') == 2:
                     final = True
-        
+
         await wss.close()
 
+
 async def run():
+    """Run the async completion and print the result."""
     async for value in AsyncCompletion.create(
-        prompt     = 'summarize cinderella with each word beginning with a consecutive letter of the alphabet, a-z',
-        # optionSets = [
-        #     "deepleo", 
-        #     "enable_debug_commands", 
-        #     "disable_emoji_spoken_text", 
-        #     "enablemm"
-        # ]
-        optionSets = [
-            #"nlu_direct_response_filter",
-            #"deepleo",
-            #"disable_emoji_spoken_text",
-            # "responsible_ai_policy_235",
-            #"enablemm",
+        prompt='summarize cinderella with each word beginning with a consecutive letter of the alphabet, a-z',
+        optionSets=[
             "galileo",
-            #"dtappid",
-            # "cricinfo",
-            # "cricinfov2",
-            # "dv3sugg",
         ]
     ):
-        print(value, end = '', flush=True)
+        print(value, end='', flush=True)
 
-asyncio.run(run())
\ No newline at end of file
+asyncio.run(run())
diff --git a/unfinished/cocalc/__init__.py b/unfinished/cocalc/__init__.py
index f345304f..3ad9937a 100644
--- a/unfinished/cocalc/__init__.py
+++ b/unfinished/cocalc/__init__.py
@@ -1,13 +1,25 @@
 import requests
 
+
 class Completion:
-    def create(prompt="What is the square root of pi",
-               system_prompt="ASSUME I HAVE FULL ACCESS TO COCALC. ENCLOSE MATH IN $. INCLUDE THE LANGUAGE DIRECTLY AFTER THE TRIPLE BACKTICKS IN ALL MARKDOWN CODE BLOCKS. How can I do the following using CoCalc?") -> str:
+    def create(self, prompt="What is the square root of pi",
+               system_prompt=("ASSUME I HAVE FULL ACCESS TO COCALC. ENCLOSE MATH IN $. "
+                              "INCLUDE THE LANGUAGE DIRECTLY AFTER THE TRIPLE BACKTICKS "
+                              "IN ALL MARKDOWN CODE BLOCKS. How can I do the following using CoCalc?")) -> str:
+
+        # Initialize a session with custom headers
+        session = self._initialize_session()
+
+        # Set the data that will be submitted
+        payload = self._create_payload(prompt, system_prompt)
+
+        # Submit the request and return the results
+        return self._submit_request(session, payload)
+
+    def _initialize_session(self) -> requests.Session:
+        """Initialize a session with custom headers for the request."""
 
-        # Initialize a session
         session = requests.Session()
-        
-        # Set headers for the request
         headers = {
             'Accept': '*/*',
             'Accept-Language': 'en-US,en;q=0.5',
@@ -16,16 +28,21 @@ class Completion:
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36',
         }
         session.headers.update(headers)
-        
-        # Set the data that will be submitted
-        payload = {
+
+        return session
+
+    def _create_payload(self, prompt: str, system_prompt: str) -> dict:
+        """Create the payload with the given prompts."""
+
+        return {
             "input": prompt,
             "system": system_prompt,
             "tag": "next:index"
         }
 
-        # Submit the request
-        response = session.post("https://cocalc.com/api/v2/openai/chatgpt", json=payload).json()
+    def _submit_request(self, session: requests.Session, payload: dict) -> str:
+        """Submit the request to the API and return the response."""
 
-        # Return the results
+        response = session.post(
+            "https://cocalc.com/api/v2/openai/chatgpt", json=payload).json()
         return response
diff --git a/unfinished/easyai/main.py b/unfinished/easyai/main.py
index 3b246977..475430ee 100644
--- a/unfinished/easyai/main.py
+++ b/unfinished/easyai/main.py
@@ -1,9 +1,12 @@
+# Import necessary libraries
 from requests import get
-from os       import urandom
-from json     import loads
+from os import urandom
+from json import loads
 
+# Generate a random session ID
 sessionId = urandom(10).hex()
 
+# Set up headers for the API request
 headers = {
     'Accept': 'text/event-stream',
     'Accept-Language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
@@ -15,18 +18,23 @@ headers = {
     'token': 'null',
 }
 
+# Main loop to interact with the AI
 while True:
+    # Get user input
     prompt = input('you: ')
-    
+
+    # Set up parameters for the API request
     params = {
         'message': prompt,
         'sessionId': sessionId
     }
 
-    for chunk in  get('http://easy-ai.ink/easyapi/v1/chat/completions', params = params,
-        headers = headers, verify = False, stream = True).iter_lines():
-        
+    # Send request to the API and process the response
+    for chunk in get('http://easy-ai.ink/easyapi/v1/chat/completions', params=params,
+                     headers=headers, verify=False, stream=True).iter_lines():
+
+        # Check if the chunk contains the 'content' field
         if b'content' in chunk:
+            # Parse the JSON data and print the content
             data = loads(chunk.decode('utf-8').split('data:')[1])
             print(data['content'], end='')
-            
\ No newline at end of file
diff --git a/unfinished/gptbz/__init__.py b/unfinished/gptbz/__init__.py
index ba326cf6..44bfcd19 100644
--- a/unfinished/gptbz/__init__.py
+++ b/unfinished/gptbz/__init__.py
@@ -1,30 +1,44 @@
 import websockets
 from json import dumps, loads
 
+# 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
-
diff --git a/unfinished/openai/__ini__.py b/unfinished/openai/__ini__.py
index b24184c6..71ec4623 100644
--- a/unfinished/openai/__ini__.py
+++ b/unfinished/openai/__ini__.py
@@ -1,72 +1,86 @@
-# experimental, needs chat.openai.com to be loaded with cf_clearance on browser ( can be closed after )
-
+# Import required libraries
 from tls_client import Session
-from uuid       import uuid4
-
+from uuid import uuid4
 from browser_cookie3 import chrome
 
-def session_auth(client):
-    headers = {
-        'authority': 'chat.openai.com',
-        'accept': '*/*',
-        'accept-language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
-        'cache-control': 'no-cache',
-        'pragma': 'no-cache',
-        'referer': 'https://chat.openai.com/chat',
-        'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
-        'sec-ch-ua-mobile': '?0',
-        'sec-ch-ua-platform': '"macOS"',
-        'sec-fetch-dest': 'empty',
-        'sec-fetch-mode': 'cors',
-        'sec-fetch-site': 'same-origin',
-        'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
-    }
 
-    return client.get('https://chat.openai.com/api/auth/session', headers=headers).json()
+class OpenAIChat:
+    def __init__(self):
+        self.client = Session(client_identifier='chrome110')
+        self._load_cookies()
+        self._set_headers()
+
+    def _load_cookies(self):
+        # Load cookies for the specified domain
+        for cookie in chrome(domain_name='chat.openai.com'):
+            self.client.cookies[cookie.name] = cookie.value
+
+    def _set_headers(self):
+        # Set headers for the client
+        self.client.headers = {
+            'authority': 'chat.openai.com',
+            'accept': 'text/event-stream',
+            'accept-language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
+            'authorization': 'Bearer ' + self.session_auth()['accessToken'],
+            'cache-control': 'no-cache',
+            'content-type': 'application/json',
+            'origin': 'https://chat.openai.com',
+            'pragma': 'no-cache',
+            'referer': 'https://chat.openai.com/chat',
+            'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
+            'sec-ch-ua-mobile': '?0',
+            'sec-ch-ua-platform': '"macOS"',
+            'sec-fetch-dest': 'empty',
+            'sec-fetch-mode': 'cors',
+            'sec-fetch-site': 'same-origin',
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
+        }
+
+    def session_auth(self):
+        headers = {
+            'authority': 'chat.openai.com',
+            'accept': '*/*',
+            'accept-language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
+            'cache-control': 'no-cache',
+            'pragma': 'no-cache',
+            'referer': 'https://chat.openai.com/chat',
+            'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
+            'sec-ch-ua-mobile': '?0',
+            'sec-ch-ua-platform': '"macOS"',
+            'sec-fetch-dest': 'empty',
+            'sec-fetch-mode': 'cors',
+            'sec-fetch-site': 'same-origin',
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
+        }
 
-client  = Session(client_identifier='chrome110')
+        return self.client.get('https://chat.openai.com/api/auth/session', headers=headers).json()
 
-for cookie in chrome(domain_name='chat.openai.com'):
-    client.cookies[cookie.name] = cookie.value
+    def send_message(self, message):
+        response = self.client.post('https://chat.openai.com/backend-api/conversation', json={
+            'action': 'next',
+            'messages': [
+                {
+                    'id': str(uuid4()),
+                    'author': {
+                        'role': 'user',
+                    },
+                    'content': {
+                        'content_type': 'text',
+                        'parts': [
+                            message,
+                        ],
+                    },
+                },
+            ],
+            'parent_message_id': '9b4682f7-977c-4c8a-b5e6-9713e73dfe01',
+            'model': 'text-davinci-002-render-sha',
+            'timezone_offset_min': -120,
+        })
 
-client.headers = {
-    'authority': 'chat.openai.com',
-    'accept': 'text/event-stream',
-    'accept-language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
-    'authorization': 'Bearer ' + session_auth(client)['accessToken'],
-    'cache-control': 'no-cache',
-    'content-type': 'application/json',
-    'origin': 'https://chat.openai.com',
-    'pragma': 'no-cache',
-    'referer': 'https://chat.openai.com/chat',
-    'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
-    'sec-ch-ua-mobile': '?0',
-    'sec-ch-ua-platform': '"macOS"',
-    'sec-fetch-dest': 'empty',
-    'sec-fetch-mode': 'cors',
-    'sec-fetch-site': 'same-origin',
-    'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
-}
+        return response.text
 
-response = client.post('https://chat.openai.com/backend-api/conversation', json = {
-    'action': 'next',
-    'messages': [
-        {
-            'id': str(uuid4()),
-            'author': {
-                'role': 'user',
-            },
-            'content': {
-                'content_type': 'text',
-                'parts': [
-                    'hello world',
-                ],
-            },
-        },
-    ],
-    'parent_message_id': '9b4682f7-977c-4c8a-b5e6-9713e73dfe01',
-    'model': 'text-davinci-002-render-sha',
-    'timezone_offset_min': -120,
-})
 
-print(response.text)
\ No newline at end of file
+if __name__ == "__main__":
+    chat = OpenAIChat()
+    response = chat.send_message("hello world")
+    print(response)