Update executor.py

executor.py

@@ -1,17 +1,48 @@
+import os
 import subprocess
+from datetime import datetime
+from config import OS_COMMANDS, TEMP_DIR
 
 def execute_command(command_type: str, command_text: str) -> str:
     try:
         if command_type == "open_app":
-            if "chrome" in command_text.lower():
-                subprocess.Popen(["chrome"])
-                return "✅ Opening Chrome..."
+            for app, cmd in OS_COMMANDS.get("open", {}).items():
+                if app in command_text.lower():
+                    subprocess.Popen(cmd.split())
+                    return f"✅ Opening {app.title()}..."
+
+        elif command_type == "close_app":
+            for app, cmd in OS_COMMANDS.get("close", {}).items():
+                if app in command_text.lower():
+                    os.system(cmd)
+                    return f"🛑 Closing {app.title()}..."
+
         elif command_type == "screenshot":
-            return "📸 Simulated screenshot taken."
+            screenshot_path = os.path.join(TEMP_DIR, f"screenshot_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png")
+            cmd = OS_COMMANDS["screenshot"]["default"]
+            cmd = cmd.replace("screenshot.png", screenshot_path)
+            os.system(cmd)
+            return f"📸 Screenshot saved to: {screenshot_path}"
+
         elif command_type == "create_file":
-            with open("new_file.txt", "w") as f:
+            filename = "new_file.txt"
+            with open(filename, "w") as f:
                 f.write("Hello from your AI assistant.")
-            return "✅ File 'new_file.txt' created."
+            return f"✅ File '{filename}' created."
+
+        elif command_type == "shutdown":
+            os.system(OS_COMMANDS["shutdown"])
+            return "🔌 Shutting down..."
+
+        elif command_type == "restart":
+            os.system(OS_COMMANDS["restart"])
+            return "🔄 Restarting system..."
+
+        elif command_type == "lock":
+            os.system(OS_COMMANDS["lock"])
+            return "🔒 System locked."
+
     except Exception as e:
         return f"❌ Error while executing command: {e}"
-    return None
+
+    return "⚠️ Command not recognized or not supported yet."