Update config.py
Browse files
config.py
CHANGED
|
@@ -1,7 +1,38 @@
|
|
| 1 |
import os
|
| 2 |
|
|
|
|
| 3 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 4 |
-
MODEL_ID = "Qwen/Qwen1.5-1.8B-Chat"
|
| 5 |
-
|
| 6 |
if not HF_TOKEN:
|
| 7 |
raise ValueError("❌ HF_TOKEN not set in environment variables.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
+
# Hugging Face Token
|
| 4 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
|
|
|
|
| 5 |
if not HF_TOKEN:
|
| 6 |
raise ValueError("❌ HF_TOKEN not set in environment variables.")
|
| 7 |
+
|
| 8 |
+
# Model to Load
|
| 9 |
+
MODEL_ID = os.getenv("MODEL_ID", "Qwen/Qwen1.5-1.8B-Chat") # Allows override via env
|
| 10 |
+
|
| 11 |
+
# Paths and Settings
|
| 12 |
+
TEMP_DIR = os.getenv("TEMP_DIR", "/tmp") # For screenshots, temporary files, etc.
|
| 13 |
+
ENABLE_EXECUTION = os.getenv("ENABLE_EXECUTION", "true").lower() == "true" # toggle command execution
|
| 14 |
+
|
| 15 |
+
# OS Commands (override or customize per OS)
|
| 16 |
+
OS_COMMANDS = {
|
| 17 |
+
"open": {
|
| 18 |
+
"chrome": "google-chrome",
|
| 19 |
+
"firefox": "firefox",
|
| 20 |
+
"notepad": "notepad",
|
| 21 |
+
"calculator": "gnome-calculator",
|
| 22 |
+
"terminal": "gnome-terminal",
|
| 23 |
+
"vscode": "code"
|
| 24 |
+
},
|
| 25 |
+
"close": {
|
| 26 |
+
"chrome": "pkill chrome",
|
| 27 |
+
"firefox": "pkill firefox",
|
| 28 |
+
"notepad": "pkill notepad",
|
| 29 |
+
"calculator": "pkill gnome-calculator",
|
| 30 |
+
"vscode": "pkill code"
|
| 31 |
+
},
|
| 32 |
+
"screenshot": {
|
| 33 |
+
"default": "import -window root screenshot.png"
|
| 34 |
+
},
|
| 35 |
+
"shutdown": "shutdown now",
|
| 36 |
+
"restart": "reboot",
|
| 37 |
+
"lock": "gnome-screensaver-command -l"
|
| 38 |
+
}
|