bot.py

# see 'main.py'

from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium import webdriver
from pynput import keyboard
import pyautogui as gui
import http.client
import json
import gc

# Configuration
PATH = "/opt/chromedriver"  # path to your downloaded webdriver
driver = webdriver.Chrome(PATH)
driver.get('https://web.whatsapp.com')
print(driver.title)  # prints title of the webpage
conn = http.client.HTTPSConnection("acobot-brainshop-ai-v1.p.rapidapi.com")
headers = {
    'x-rapidapi-key': "85632300dbmsha01f5765f1a7303p18df83jsn83e04aa1780a",
    'x-rapidapi-host': "acobot-brainshop-ai-v1.p.rapidapi.com"
}


class Bot:
    def __init__(self, contact):
        self.contact = contact
        elem = WebDriverWait(driver, 30).until(EC.element_to_be_clickable((By.XPATH, f'//*[text() = "{self.contact}" ]')))
        elem.click()
        self.incoming = WebDriverWait(driver, 30).until(
            EC.visibility_of_all_elements_located((By.XPATH, f'//div[contains(@data-pre-plain-text, "{self.contact}")]')))
        self.received_msgs = len(self.incoming)
        self.running = True
        self.pressed_ctrl = False

    def make_call(self, request):
        request = request.replace(' ', '%20').replace('\n', '')
        conn.request("GET", f'/get?bid=178&key=sX5A2PcYZbsN5EY6&uid=mashape&msg={request}', headers=headers)

        res = conn.getresponse()
        data = res.read()

        response = json.loads(data.decode('UTF-8'))["cnt"]
        return response.replace('Aco', 'RAZ').replace('acobot.ai', 'a remote repo')  # replace name and location in response 

    def new_msg_received(self):
        incoming = driver.find_elements_by_xpath(f'//div[contains(@data-pre-plain-text, "{self.contact}")]')
        if len(incoming) != self.received_msgs:
            return True
        else:
            return False

    def send_message(self, text):
        input_box = driver.find_elements_by_class_name('copyable-text')
        input_box = input_box[len(input_box) - 1]
        input_box.click()
        input_box.send_keys(text, Keys.RETURN)
        incoming = driver.find_elements_by_xpath(f'//div[contains(@data-pre-plain-text, "{self.contact}")]')
        self.received_msgs = len(incoming)

    def on_press(self, key):
        print(key, 'Key pressed')
        if key == keyboard.Key.ctrl_l:  # If 'Left Ctrl' is the key pressed
            try:
                self.pressed_ctrl = True
            except Exception as e:
                print(f'{self.contact}: No such contact')
                print(e)
                pass

    def init_bot(self):
        while self.running:
            if self.new_msg_received():
                try:
                    self.incoming = driver.find_elements_by_xpath(
                        f'//div[contains(@data-pre-plain-text, "{self.contact}")]')
                    self.received_msgs = len(self.incoming)
                    last_msg = self.incoming[self.received_msgs - 1].text
                    print(last_msg)
                    self.send_message('_Thinking..._')
                    self.send_message(self.make_call(last_msg))
                except Exception as e:
                    summary = str(e)
                    self.send_message(summary)
                    print(e)
                    pass
            if self.pressed_ctrl:
                break
        self.contact = gui.prompt('Enter contact\'s name', 'WhatsApp Chat bot')
        print(self.contact, type(self.contact))
        driver.find_element_by_xpath(f'//*[text() = "{self.contact}" ]').click()
        self.pressed_ctrl = False
        self.init_bot()

    # In case you have to stop the program for some reason
    def stop_bot(self):
        self.running = False
        print('Bot stopped')
        gc.collect()