Crizomb/ChatGPT-python-bash-interface
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update driver.py

Browse source
This commit is contained in:
Crizomb 2023-04-06 10:38:45 +02:00 • committed by GitHub
parent d19ed0f9fc
commit ac4438a632
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

15
gpt_env/driver.py
View file

@ -15,6 +15,7 @@ from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.support import expected_conditions as EC
def better_send_keys(element, text): def better_send_keys(element, text):
# Like send keys but doesn't press enter when viewing '\n'
lines = text.split("\n") lines = text.split("\n")
for i, line in enumerate(lines): for i, line in enumerate(lines):
element.send_keys(line) element.send_keys(line)
@ -33,11 +34,10 @@ class Driver:
chrome_options.add_argument(f"user-data-dir={data_dir}") chrome_options.add_argument(f"user-data-dir={data_dir}")
self.driver = Chrome(options=chrome_options, use_subprocess=True) self.driver = Chrome(options=chrome_options, use_subprocess=True)
#full screen
self.driver.maximize_window() self.driver.maximize_window()
def bad_wait_until(self, by, value): def bad_wait_until(self, by, value):
"""Wait until the element is found""" """Wait for chatgpt answer"""
while True: while True:
try: try:
element = self.driver.find_element(by, value) element = self.driver.find_element(by, value)
@ -88,10 +88,9 @@ class Driver:
print("send_inputs") print("send_inputs")
driver.send_inputs(prompt) driver.send_inputs(prompt)
time.sleep(2) time.sleep(2)
#wait until the chat is updated and button is shown
#wait until the chat is updated and the button "regenerate answer" is shown
xpath = """//*[@id="__next"]/div[2]/div[2]/main/div[2]/form/div/div[1]/button/div""" xpath = """//*[@id="__next"]/div[2]/div[2]/main/div[2]/form/div/div[1]/button/div"""
self.bad_wait_until(By.XPATH, xpath) self.bad_wait_until(By.XPATH, xpath)
print("get_chat") print("get_chat")
@ -149,10 +148,12 @@ def main():
answer = run_shell_command(code.code) answer = run_shell_command(code.code)
time.sleep(1) time.sleep(1)
if (j+14)%24 == 0: if (j+1)%24 == 0:
#If using gpt4 wait for 3 hours after 25 msg
print("Waiting 3 hours")
time.sleep(60*60*3) time.sleep(60*60*3)
print("HEY WAKE UP BIATCH")
time.sleep(60*5) driver.close()
if __name__ == "__main__": if __name__ == "__main__":