#pip install selenium

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager
from os import getcwd

chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument("--use-fake-ui-for-media-stream")
chrome_options.add_argument("--use-fake-device-for-media-stream")
chrome_options.add_argument("--headless=new")
chrome_options.add_argument("--auto-select-desktop-capture-source=Screen 1")
chrome_options.add_argument("--allow-file-access-from-files")
chrome_options.add_argument("--allow-insecure-localhost")
chrome_options.add_argument("--use-file-for-fake-video-capture=fake.y4m")

driver = webdriver.Chrome(
    service=Service(ChromeDriverManager().install()),
    options=chrome_options
)

website = "https://allorizenproject1.netlify.app/"
driver.get(website)

rec_file = getcwd() + r"\input.txt"

def listen():
    try:
        start_button = WebDriverWait(driver, 20).until(
            EC.element_to_be_clickable((By.ID, 'startButton'))
        )
        start_button.click()
        print("Listening...")

        output_text = ""
        is_second_click = False

        while True:
            # Re-locate button so .text updates
            start_button = driver.find_element(By.ID, 'startButton')

            output_element = WebDriverWait(driver, 20).until(
                EC.presence_of_element_located((By.ID, 'output'))
            )
            current_text = output_element.text.strip()

            # BUTTON RESET CHECK
            if "Start Listening" in start_button.text and is_second_click:
                if output_text:
                    is_second_click = False

            # LISTENING ACTIVE
            elif "Listening" in start_button.text:
                is_second_click = True

                if current_text != output_text:
                    output_text = current_text

                    with open(rec_file, "w") as file:
                        file.write(output_text.lower())

                    print("USER : " + output_text)

    except KeyboardInterrupt:
        pass
    except Exception as e:
        print(e)

listen()
