Add files via upload

chatgpt_code_interface/driver.py

@@ -0,0 +1,135 @@
+from selenium.webdriver.common.keys import Keys
+
+from undetected_chromedriver import Chrome
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+import bs4
+import re
+import time
+
+from exec_python_code import create_namespace, run_code
+from exec_term_code import run_shell_command
+from formating_tools import isolate_code_bloc
+
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+
+def better_send_keys(element, text):
+    lines = text.split("\n")
+    for i, line in enumerate(lines):
+        element.send_keys(line)
+        if i != len(lines)-1:
+            element.send_keys(Keys.SHIFT + Keys.ENTER)
+
+class GPTDriver:
+    def __init__(self, headless=False, data_dir="selenium1"):
+        time.sleep(1)
+        chrome_options = Options()
+
+        if headless :
+            chrome_options.add_argument("--headless")
+
+
+        chrome_options.add_argument(f"user-data-dir={data_dir}")
+
+        self.driver = Chrome(options=chrome_options, use_subprocess=True)
+        #full screen
+        self.driver.maximize_window()
+
+    def wait_answer(self):
+        """Wait until the answer is ready"""
+        xpath = """//*[@id="__next"]/div[2]/div[2]/main/div[2]/form/div/div[1]/button/div"""
+        while True:
+            try:
+                element = self.driver.find_element(By.XPATH, xpath)
+                #if element contain "stop generating" then continue
+                if "stop generating" in element.text.lower():
+                    time.sleep(0.1)
+                    continue
+                return element
+            except:
+                time.sleep(0.1)
+                continue
+
+    def create_user_data_dir(self, data_dir="selenium1"):
+        """Create a user data dir for selenium"""
+        chrome_options = Options()
+        chrome_options.add_argument(f"user-data-dir={data_dir}")
+        chrome_options.add_argument("--disable-images")
+        self.driver = Chrome(options=chrome_options)
+        self.driver.set_window_size(600, 1000)
+        self.driver.get("https://chat.openai.com/chat")
+        time.sleep(60)
+        self.driver.quit()
+
+    def connect(self):
+        self.driver.get("https://chat.openai.com/chat")
+
+    def get_chat(self):
+        """Get the chat history"""
+        soup = bs4.BeautifulSoup(self.driver.page_source, "html.parser")
+        chat = soup.find("div", class_=re.compile("flex flex-col items-center text-sm"))
+        return chat
+
+    def get_last_chat(self):
+        """Get the last chat message"""
+        soup = bs4.BeautifulSoup(self.driver.page_source, "html.parser")
+        #last chat is the last div with class = " group w-full text-gray-800 dark:text-gray-100 border-b border-black/10 dark:border-gray-900/50 bg-gray-50 dark:bg-[#444654]"
+        chat = soup.find_all("div", class_="""group w-full text-gray-800 dark:text-gray-100 border-b border-black/10 dark:border-gray-900/50 bg-gray-50 dark:bg-[#444654]""")[-1]
+        return chat
+
+    def send_message(self, txt):
+        textarea = self.driver.find_element(By.XPATH, '//*[@id="__next"]/div[2]/div[2]/main/div[2]/form/div/div[2]/textarea')
+        better_send_keys(textarea, txt)
+        time.sleep(1)
+        button_send = self.driver.find_element(By.XPATH, '//*[@id="__next"]/div[2]/div[2]/main/div[2]/form/div/div[2]/button')
+        button_send.click()
+
+    def add_run_button(self):
+        #html of a big red button with border-radius of 10%, green background and white text pady by 10px up and padx by 10px left
+        button_html = '<button id="continueButton" style="position: fixed; bottom: 10px; right: 10px; font-size: 2em; background-color: green; color: white; border-radius: 10%;">Continue</button>'
+        xpath = """//*[@id="__next"]/div[2]/div[2]/main/div[2]/div/span"""
+
+        # Find the element using the given XPath
+        element = None
+        while not element:
+            try:
+                element = self.driver.find_element(By.XPATH, xpath)
+            except:
+                time.sleep(0.1)
+
+        # Inject the button after the element found using the XPath
+        self.driver.execute_script(f"arguments[0].insertAdjacentHTML('afterend', '{button_html}');", element)
+
+        # Add the click event listener for the button
+        self.driver.execute_script("""
+            document.getElementById('continueButton').addEventListener('click', function() {
+                window.loopControl1 = 'continue';
+            });
+        """)
+
+    def add_send_result_button(self):
+        #html of a big red button with border-radius of 10%, green background and white text pady by 10px up and padx by 10px left
+        button_html = '<button id="sendResultButton" style="position: fixed; bottom: 10px; right: 10px; font-size: 2em; background-color: green; color: white; border-radius: 10%;">Send result</button>'
+        xpath = """//*[@id="__next"]/div[2]/div[2]/main/div[2]/div/span"""
+
+        # Find the element using the given XPath
+        element = None
+        while not element:
+            try:
+                element = self.driver.find_element(By.XPATH, xpath)
+            except:
+                time.sleep(0.1)
+
+        # Inject the button after the element found using the XPath
+        self.driver.execute_script(f"arguments[0].insertAdjacentHTML('afterend', '{button_html}');", element)
+
+        # Add the click event listener for the button
+        self.driver.execute_script("""
+            document.getElementById('sendResultButton').addEventListener('click', function() {
+                window.loopControl2 = 'send_result';
+            });
+        """)
+
+    def close(self):
+        self.driver.close()