tiktok-nav.py

import os
import time
import json
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains
from tkinter import Tk  # To get screen dimensions

# Set up Chrome options
chrome_options = Options()
chrome_options.add_argument("--disable-notifications")  # Disable native pop-ups

# Path to your chromedriver executable
chromedriver_path = "C:\\Users\\avery\\OneDrive\\Documents\\GitHub\\Neuro-Stress-Monitor\\chromedriver.exe"  # Update this with your chromedriver path

# Initialize WebDriver
service = Service(chromedriver_path)
driver = webdriver.Chrome(service=service, options=chrome_options)

# Initialize ActionChains for keyboard interactions
actions = ActionChains(driver)

# Set the browser window to 50% of the screen width and position it on the right
def set_browser_position(driver):
    # Get the screen width and height
    root = Tk()
    root.withdraw()  # Hide the Tkinter window
    screen_width = root.winfo_screenwidth()
    screen_height = root.winfo_screenheight()

    # Calculate window dimensions and position
    window_width = screen_width // 2  # 50% of the screen width
    window_height = screen_height  # Full screen height
    window_x = screen_width // 2  # Start at the middle of the screen (right side)
    window_y = 0  # Start at the top of the screen

    # Set the window size and position
    driver.set_window_rect(window_x, window_y, window_width, window_height)
    print(f"Browser positioned at x={window_x}, y={window_y}, width={window_width}, height={window_height}")

# Initialize counts
focused_count = 0
unfocused_count = 0
# Display a notification directly in the browser using the Notification API
def show_browser_notification(driver, title, message):
    """
    Display a notification directly in the browser using the Notification API.
    Falls back to a custom modal if notifications are not supported or denied.
    Automatically closes the notification after 2 seconds.
    """
    script = '''
    (function(title, message) {
        function createCustomModal(modalTitleText, modalMessageText) {
            // Check if the modal already exists
            var existingModal = document.getElementById('custom-modal');
            if (existingModal) {
                existingModal.remove();
            }

            // Create modal elements
            var modal = document.createElement('div');
            modal.id = 'custom-modal';
            modal.style.position = 'fixed';
            modal.style.top = '0';
            modal.style.left = '0';
            modal.style.width = '100%';
            modal.style.height = '100%';
            modal.style.backgroundColor = 'rgba(0, 0, 0, 0.5)';
            modal.style.display = 'flex';
            modal.style.alignItems = 'center';
            modal.style.justifyContent = 'center';
            modal.style.zIndex = '9999';

            var modalContent = document.createElement('div');
            modalContent.style.backgroundColor = '#fff';
            modalContent.style.padding = '20px';
            modalContent.style.borderRadius = '5px';
            modalContent.style.textAlign = 'center';
            modalContent.style.maxWidth = '80%';
            modalContent.style.color = '#000';  // Ensure text color is black

            var modalTitle = document.createElement('h2');
            modalTitle.innerText = modalTitleText;
            modalTitle.style.color = '#000';  // Set text color to black

            var modalMessage = document.createElement('p');
            modalMessage.innerText = modalMessageText;
            modalMessage.style.color = '#000';  // Set text color to black

            modalContent.appendChild(modalTitle);
            modalContent.appendChild(modalMessage);
            modal.appendChild(modalContent);
            document.body.appendChild(modal);

            // Auto-close the modal after 2 seconds
            setTimeout(function() {
                var modal = document.getElementById('custom-modal');
                if (modal) {
                    modal.remove();
                }
            }, 2000);
        }

        if (typeof Notification !== "undefined") {
            if (Notification.permission === "granted") {
                var notification = new Notification(title, {
                    body: message,
                    icon: "https://cdn-icons-png.flaticon.com/512/633/633600.png"
                });
                setTimeout(function() { notification.close(); }, 2000);
            } else if (Notification.permission !== "denied") {
                Notification.requestPermission().then(function(permission) {
                    if (permission === "granted") {
                        var notification = new Notification(title, {
                            body: message,
                            icon: "https://cdn-icons-png.flaticon.com/512/633/633600.png"
                        });
                        setTimeout(function() { notification.close(); }, 2000);
                    } else {
                        createCustomModal(title, message);
                    }
                });
            } else {
                createCustomModal(title, message);
            }
        } else {
            createCustomModal(title, message);
        }
    })(arguments[0], arguments[1]);
    '''
    driver.execute_script(script, title, message)

# Initialize counts
focused_count = 0
unfocused_count = 0

# Path to the `file.txt`
file_path = os.path.join(os.path.dirname(__file__), "renderer", "file.txt")

try:
    # Set browser position
    set_browser_position(driver)

    # Step 1: Go to TikTok
    driver.get("https://www.tiktok.com/foryou")
    print("Navigated to TikTok.")

    # Wait for the page to load
    time.sleep(5)

    # Accept cookies if prompted (TikTok may display a cookie consent dialog)
    try:
        consent_button = driver.find_element(By.XPATH, '//button[contains(text(), "Accept all cookies")]')
        consent_button.click()
        print("Accepted cookies.")
    except Exception as e:
        print("No cookie consent dialog found or error accepting cookies.")

    # Wait a bit after accepting cookies
    time.sleep(2)

    # Optional: Close any pop-ups or modals (e.g., login prompts)
    try:
        close_button = driver.find_element(By.XPATH, '//button[@aria-label="Close"]')
        close_button.click()
        print("Closed pop-up modal.")
    except Exception as e:
        print("No pop-up modal found or error closing it.")

    # Ensure the page is ready
    time.sleep(2)

    # Step 2: Start monitoring "file.txt" every 10 seconds
    while True:
        time.sleep(10)  # Wait for 10 seconds

        # Read the content of "file.txt"
        try:
            with open(file_path, "r") as f:
                status = f.read().strip()
                print(f"Status read from file.txt: {status}")
        except Exception as e:
            print(f"Error reading file.txt: {e}")
            continue  # Skip this iteration if file cannot be read

        if status == "Focused":
            focused_count += 1
            unfocused_count = 0  # Reset unfocused count
            if focused_count % 2 == 0:  # Trigger every 2nd "Focused"
                # Show congratulatory notification
                title = "Great Job!"
                message = "Congratulations! You are being focused. Keep up the pace!"
                show_browser_notification(driver, title, message)

        elif status == "Unfocused":
            unfocused_count += 1
            focused_count = 0  # Reset focused count

            if unfocused_count == 1:
                # First "Unfocused": Go to the next video
                try:
                    # Simulate pressing the down arrow key
                    actions.send_keys(Keys.ARROW_DOWN).perform()
                    print("Pressed down arrow key to go to next video.")
                except Exception as e:
                    print(f"Error pressing down arrow key: {e}")
            elif unfocused_count == 2:
                # Second "Unfocused": Show a warning notification
                title = "Attention!"
                message = "You are not focused. Please refocus or the session will end soon."
                show_browser_notification(driver, title, message)
            elif unfocused_count == 3:
                # Third "Unfocused": Show a final notification and close the browser
                title = "Session Ended"
                message = "You are not focused. Take a 5-10 minute break to re-energize!"
                show_browser_notification(driver, title, message)
                print("Closing the browser in 3 seconds.")
                time.sleep(3)  # Wait before closing
                break  # Exit the loop to close the browser

        else:
            print(f"Unrecognized status in file.txt: {status}")

finally:
    # Close the browser
    driver.quit()
    print("Browser closed.")