config.py

import os

# Hugging Face Token – not required for public models like Mistral
HF_TOKEN = os.getenv("HF_TOKEN")
if HF_TOKEN == "":
    HF_TOKEN = None  # Normalize empty string to None

# ✅ Public model — No token required
MODEL_ID = os.getenv("MODEL_ID", "Qwen/Qwen3-1.7B")



# Paths and Settings
TEMP_DIR = os.getenv("TEMP_DIR", "/tmp")  # Temporary storage (screenshots, temp files, etc.)
ENABLE_EXECUTION = os.getenv("ENABLE_EXECUTION", "true").lower() == "true"  # Allow shell execution

# OS Commands (can be adjusted per OS)
OS_COMMANDS = {
    "open": {
        "chrome": "google-chrome",
        "firefox": "firefox",
        "notepad": "notepad",
        "calculator": "gnome-calculator",
        "terminal": "gnome-terminal",
        "vscode": "code"
    },
    "close": {
        "chrome": "pkill chrome",
        "firefox": "pkill firefox",
        "notepad": "pkill notepad",
        "calculator": "pkill gnome-calculator",
        "vscode": "pkill code"
    },
    "screenshot": {
        "default": "import -window root screenshot.png"
    },
    "shutdown": "shutdown now",
    "restart": "reboot",
    "lock": "gnome-screensaver-command -l"
}