Upload folder using huggingface_hub

app.py

@@ -12,7 +12,7 @@ from datetime import datetime, timedelta
 # This expects a HF_TOKEN and REPOS and RESTART_TIMES env vars
 # Expected format is:
 # RESTART_TIMES: 06:00,12:00,18:00,00:00
-# REPOS: lainlives/bldr,lainlives/starter
+# REPOS: lainlives/bldr,lainlives/ztestzz
 #
 #
 #
@@ -31,8 +31,8 @@ repos_env = os.getenv("REPOS")
 times_env = os.getenv("RESTART_TIMES")
 REPOS = [t.strip() for t in repos_env.split(",")]
 RESTART_TIMES = [t.strip() for t in times_env.split(",")]
-
-
+HELPER = "lainlives/starter"
+HELPER_TIME = "00:30"
 # Global variable to track the specific target timestamp
 next_reboot_datetime = None
 
@@ -92,6 +92,30 @@ def get_last_reboot_status():
     return f"Next target: {next_reboot_datetime.strftime('%Y-%m-%d %H:%M:%S')}"
 
 
+def check_helper():
+    """Checks if current time has passed the scheduled time."""
+
+    now = datetime.now()
+
+    # Check if we have reached or passed the target time
+    if now >= HELPER_TIME:
+        status_msg = f"Reboot triggered at {now.strftime('%H:%M:%S')}"
+        print(status_msg)
+
+        # 1. Execute Reboots
+        if HF_TOKEN:
+            api = HfApi()
+            try:
+                print(f"Restarting space {HELPER}...")
+                api.pause_space(repo_id=HELPER)
+                sleep(3)  # Brief pause to ensure state registers
+                api.restart_space(repo_id=HELPER, token=HF_TOKEN)
+            except Exception as e:
+                print(f"Failed to restart {HELPER}: {e}")
+
+    return "Space running normally."
+
+
 def check_and_reboot():
     """Checks if current time has passed the scheduled time."""
     global next_reboot_datetime
@@ -110,8 +134,8 @@ def check_and_reboot():
                 try:
                     print(f"Restarting space {repo}...")
                     api.pause_space(repo_id=repo)
-                    sleep(1)  # Brief pause to ensure state registers
-                    api.restart_space(repo_id=repo, token=HF_TOKEN)
+                    sleep(3)  # Brief pause to ensure state registers
+                    api.restart_space(repo_id=repo, token=HF_TOKEN, factory_reboot=True)
                 except Exception as e:
                     print(f"Failed to restart {repo}: {e}")
 
@@ -128,7 +152,7 @@ def check_and_reboot():
 def pipeline():
     # Run the check
     status = check_and_reboot()
-
+    check_helper()
     # Return updated UI elements
     countdown = time_until_next_reboot()
     target_info = get_last_reboot_status()