Update scheduler.py

scheduler.py

@@ -1,25 +1,24 @@
 import os
 import time
+import sys
 import schedule
 import subprocess
 import datetime
 import psutil
 
-# ✅ Persistent path in Hugging Face Spaces
 MODEL_DIR = "/app/models"
 TIMESTAMP_FILE = os.path.join(MODEL_DIR, "last_train.txt")
-APP_PATH = "./app.py"
+APP_PATH = "app.py"
+TRAIN_SCRIPT = "train.py"
 TRAIN_INTERVAL = datetime.timedelta(days=2)
 
-# Make sure model dir exists
+# Ensure model dir exists
 os.makedirs(MODEL_DIR, exist_ok=True)
 
 def get_latest_model():
     """Return the latest model file path, or None if no models exist."""
     models = [f for f in os.listdir(MODEL_DIR) if f.endswith(".pth")]
-    if not models:
-        return None
-    return os.path.join(MODEL_DIR, sorted(models)[-1])
+    return os.path.join(MODEL_DIR, sorted(models)[-1]) if models else None
 
 def should_train():
     """Return True if it's time to retrain the model."""
@@ -47,36 +46,43 @@ def is_app_running():
 def launch_app():
     """Start app.py if not already running."""
     if not is_app_running():
-        print(f"[{datetime.datetime.now()}] 🚀 Launching app.py...")
-        subprocess.Popen(["python", APP_PATH])
+        print(f"[{datetime.datetime.now()}]  Launching app.py...", flush=True)
+        try:
+            subprocess.Popen([sys.executable, APP_PATH])
+        except FileNotFoundError:
+            print(f"[{datetime.datetime.now()}]  app.py not found!", flush=True)
     else:
-        print(f"[{datetime.datetime.now()}] ✅ app.py is already running.")
+        print(f"[{datetime.datetime.now()}] app.py is already running.", flush=True)
 
 def train_model():
     """Train model if needed and launch app."""
     if should_train():
-        print(f"[{datetime.datetime.now()}] 🚀 Training model...")
+        print(f"[{datetime.datetime.now()}]  Training model...", flush=True)
         latest = get_latest_model()
-        if latest:
+        if latest and os.path.exists(latest):
             os.remove(latest)
-            print(f"🗑️ Deleted old model: {latest}")
+            print(f" Deleted old model: {latest}", flush=True)
+
         try:
-            subprocess.run(["python", "./train.py"], check=True)
+            subprocess.run([sys.executable, TRAIN_SCRIPT], check=True)
             update_timestamp()
-            print(f"[{datetime.datetime.now()}] ✅ Training completed.")
+            print(f"[{datetime.datetime.now()}]  Training completed.", flush=True)
         except subprocess.CalledProcessError as e:
-            print(f"[{datetime.datetime.now()}] ❌ Training failed: {e}")
+            print(f"[{datetime.datetime.now()}]  Training failed: {e}", flush=True)
+        except FileNotFoundError:
+            print(f"[{datetime.datetime.now()}]  train.py not found!", flush=True)
     else:
-        print(f"[{datetime.datetime.now()}] ⏳ Skipping training (recent model exists).")
+        print(f"[{datetime.datetime.now()}]  Skipping training (recent model exists).", flush=True)
 
     launch_app()
 
-# Run once at startup
-train_model()
+if __name__ == "__main__":
+    # Run once at startup
+    train_model()
 
-# Then every 60 minutes
-schedule.every(60).minutes.do(train_model)
+    # Then schedule every 60 minutes
+    # schedule.every(60).minutes.do(train_model)
 
-while True:
-    schedule.run_pending()
-    time.sleep(3)
+    while True:
+        schedule.run_pending()
+        time.sleep(3)