Upload scheduler.py

scheduler.py

@@ -1,15 +1,32 @@
 
-import schedule
 import time
-from agent import run_agent
-from logger import logger
-
-def scheduled_task():
-    logger.info("Scheduled task running...")
-    run_agent("Daily self-check")
-
-def start_scheduler():
-    schedule.every().day.at("10:00").do(scheduled_task)
-    while True:
-        schedule.run_pending()
-        time.sleep(1)
+import threading
+
+class TaskScheduler:
+    def __init__(self):
+        self.tasks = []
+        self.running = False
+
+    def add_task(self, name, func, interval=10):
+        task = {"name": name, "func": func, "interval": interval, "last_run": 0}
+        self.tasks.append(task)
+
+    def start(self):
+        self.running = True
+        threading.Thread(target=self.run_loop, daemon=True).start()
+
+    def stop(self):
+        self.running = False
+
+    def run_loop(self):
+        while self.running:
+            now = time.time()
+            for task in self.tasks:
+                if now - task["last_run"] >= task["interval"]:
+                    try:
+                        result = task["func"]()
+                        print(f"[Scheduler] Ran task '{task['name']}': {result}")
+                    except Exception as e:
+                        print(f"[Scheduler Error] {task['name']}: {e}")
+                    task["last_run"] = now
+            time.sleep(1)