chatbot/multiAccountTestVersion.py at main · bittechio/chatbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
from time import sleep
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.remote.webelement import WebElement
from multiprocessing import Process

def Find_Element(driver : webdriver.Chrome, by, value : str) -> WebElement:
    while True:
        try:
            element = driver.find_element(by, value)
            break
        except:
            pass
        sleep(0.1)
    return element

def run_bot(driver_num, email, password):
    driver = webdriver.Chrome()
    chat_url = 'https://huggingface.co/chat/'
    driver.get(chat_url)
    sleep(1)

    # Click "Sign in with Hugging Face" button.
    goToSignBar = Find_Element(driver, By.XPATH, '/html/body/div[2]/div/div/div/div/form/button')
    driver.execute_script('arguments[0].click();', goToSignBar)
    sleep(1)

    # Sign in.
    emailBox = Find_Element(driver, By.NAME, 'username')
    emailBox.send_keys(email)
    sleep(0.5)
    passwordBox = Find_Element(driver, By.NAME, 'password')
    passwordBox.send_keys(password)
    loginButton = Find_Element(driver, By.XPATH, '/html/body/div[1]/main/div/section/form/div[2]/button')
    driver.execute_script('arguments[0].click();', loginButton)
    sleep(1)

    # Input text in textBox and Enter.
    for i in range(1, 501):
        sleep(2)
        enter_prompt = Find_Element(driver, By.TAG_NAME, 'textarea')
        enter_prompt.send_keys('This is a test prompt')
        sleep(1)
        enter_prompt.send_keys(Keys.ENTER)
        print(f'Browser-{driver_num}, Prompt {i}')
        sleep(60)

    # Save answers as a text file.
    answers = driver.find_elements(By.XPATH, '//*[@id="app"]/div[1]/div/div[1]/div/div/div')
    with open(f'answer_{driver_num}.txt', 'w') as file:
        for i in range(1, len(answers), 2):
            file.write(answers[i].text + '\n')

    driver.quit()

if __name__ == '__main__':

    accounts = [
        {'email': 'dalmaciotelfair@gmail.com', 'password': 'lxoviatiuspiM1'},
        {'email': 'dalmakempinski@gmail.com', 'password': 'M11dn45jf7wx'},
        {'email': 'damarispastano@gmail.com', 'password': 'M1ttt92x1sjxvn'},
        {'email': 'damasodavenport@gmail.com', 'password': 'M18xeze0knldds'},
        {'email': 'dambergervicky@gmail.com', 'password': 'M1ei1xa0m59l82'},
        {'email': 'damiogilbert@gmail.com', 'password': 'M1ehbgith6e6wqlv'},
        {'email': 'damnneal813@gmail.com', 'password': 'M1gombqq1zel'},
        {'email': 'damoclesstratman@gmail.com', 'password': 'M1ncgfmxohw0b'},
        {'email': 'daniastoldt@gmail.com', 'password': '7M1j9n4wgtgddsd'},
        {'email': 'danielamiddents@gmail.com', 'password': 'M1626rqpigzgslaj'},
        {'email': 'daniellequinchia89@gmail.com', 'password': 'M1pjs8hh04ajbe'},
        {'email': 'darcivirtue58@gmail.com', 'password': 'M1y92ze3jb72wiq'},
        {'email': 'dariarosalind245@gmail.com', 'password': 'M1mrvv9obwss'},
        {'email': 'darrickscoobz@gmail.com', 'password': 'M1589lnuptgdsd'},
        {'email': 'dativaalexandre@gmail.com', 'password': 'M1lpk32tpm5cpe'},
        {'email': 'dauntstarr@gmail.com', 'password': 'M1hhg2m6gesasdff'},
        {'email': 'davionemoncus@gmail.com', 'password': 'M1pbhde37fju'},
        {'email': 'davorspalitto@gmail.com', 'password': 'M1j1vfvtw24kw1'},
        {'email': 'dawsonsatordi@gmail.com', 'password': 'M11ruoy3c6dsdf'},
        {'email': 'dayaniraphaneuf@gmail.com', 'password': 'M1j3ojre951r8z'}
    ]

    processes = []
    for i, account in enumerate(accounts):
        process = Process(target = run_bot, args = (i + 1, account["email"], account["password"]))
        processes.append(process)
        process.start()

    for process in processes:
        process.join()