From 343990e204579ab03f912cf1bf1aef718d056e13 Mon Sep 17 00:00:00 2001
From: Hesiod <oneword1word@gmail.com>
Date: Sat, 29 Apr 2023 15:35:32 +1000
Subject: [PATCH 1/2] Add model selection to class, if model is 4, use GPT-4
 for chat

---
 src/pyChatGPT/pyChatGPT.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/pyChatGPT/pyChatGPT.py b/src/pyChatGPT/pyChatGPT.py
index aa0b605..8f3771f 100644
--- a/src/pyChatGPT/pyChatGPT.py
+++ b/src/pyChatGPT/pyChatGPT.py
@@ -61,6 +61,7 @@ def __init__(
         chrome_args: list = [],
         moderation: bool = True,
         verbose: bool = False,
+        model: str = "3.5"
     ):
         '''
         Initialize the ChatGPT object\n
@@ -90,6 +91,7 @@ def __init__(
         self.__proxy = proxy
         self.__chrome_args = chrome_args
         self.__moderation = moderation
+        self.__model = model
 
         if not self.__session_token and (
             not self.__email or not self.__password or not self.__auth_type
@@ -227,8 +229,23 @@ def __init_browser(self) -> None:
         self.__check_blocking_elements()
 
         self.__is_active = True
+        if self.__model == "4":
+            self.__select_gpt4()
         Thread(target=self.__keep_alive, daemon=True).start()
 
+    def __select_gpt4(self) -> None:
+        partial_id = "headlessui-listbox-button-"
+        css_selector = f'[id*="{partial_id}"]'
+        select_button = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.CSS_SELECTOR, css_selector)))
+        select_button.click()
+        partial_id = "headlessui-listbox-option-"
+        css_selector = f'[id*="{partial_id}"]'
+        WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.CSS_SELECTOR, css_selector)))
+        model_items = self.driver.find_elements(By.CSS_SELECTOR, css_selector)
+        for model in model_items:
+            if model.text == "GPT-4":
+                model.click()
+
     def __ensure_cf(self, retry: int = 3) -> None:
         '''
         Ensure Cloudflare cookies are set\n
@@ -473,6 +490,9 @@ def reset_conversation(self) -> None:
         except SeleniumExceptions.NoSuchElementException:
             self.logger.debug('New chat button not found')
             self.driver.save_screenshot('reset_conversation_failed.png')
+        if model == "gpt-4":
+            self.__select_gpt4()
+        self.__conversation_id = None
 
     def clear_conversations(self) -> None:
         '''
@@ -509,3 +529,5 @@ def refresh_chat_page(self) -> None:
         self.driver.get(chatgpt_chat_url)
         self.__check_capacity(chatgpt_chat_url)
         self.__check_blocking_elements()
+        if self.__model == "gpt-4":
+            self.__select_gpt4()

From 6395b4ce0efeed4a17b382b2ce8a746dc2bb704a Mon Sep 17 00:00:00 2001
From: Hesiod <oneword1word@gmail.com>
Date: Sat, 29 Apr 2023 15:49:08 +1000
Subject: [PATCH 2/2] Fix inconsitency in model names

---
 src/pyChatGPT/pyChatGPT.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/pyChatGPT/pyChatGPT.py b/src/pyChatGPT/pyChatGPT.py
index 8f3771f..5ff27fa 100644
--- a/src/pyChatGPT/pyChatGPT.py
+++ b/src/pyChatGPT/pyChatGPT.py
@@ -490,7 +490,7 @@ def reset_conversation(self) -> None:
         except SeleniumExceptions.NoSuchElementException:
             self.logger.debug('New chat button not found')
             self.driver.save_screenshot('reset_conversation_failed.png')
-        if model == "gpt-4":
+        if self.__model == "4":
             self.__select_gpt4()
         self.__conversation_id = None
 
@@ -529,5 +529,5 @@ def refresh_chat_page(self) -> None:
         self.driver.get(chatgpt_chat_url)
         self.__check_capacity(chatgpt_chat_url)
         self.__check_blocking_elements()
-        if self.__model == "gpt-4":
+        if self.__model == "4":
             self.__select_gpt4()