Upload folder using huggingface_hub

app.py

@@ -1,7 +1,9 @@
 import os
+import time
 from time import sleep
 from huggingface_hub import HfApi
 from datetime import datetime, timedelta
+import threading
 import gradio as gr
 
 #
@@ -18,16 +20,11 @@ import gradio as gr
 #
 
 
-# --- Configuration ---
-# Get Space ID and Token from environment variables (Secrets)
 HF_TOKEN = os.getenv("HF_TOKEN")
-# REPOS = ["lainlives/bldr", "lainlives/starter"]
 repos_env = os.getenv("REPOS")
-# List of times to restart (24-hour format, UTC usually)
-# RESTART_TIMES = ["06:00", "12:00", "18:00", "00:00"]
+times_env = os.getenv("RESTART_TIMES")
 
 
-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"
@@ -92,28 +89,16 @@ 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 trigger_helper():
+    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}")
 
 
 def check_and_reboot():
@@ -152,7 +137,6 @@ 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()
@@ -160,6 +144,22 @@ def pipeline():
     return countdown, status, target_info
 
 
+def sanitize():
+    now = datetime.now()
+    target_time = datetime.strptime(SELF_TIME, "%H:%M").time()
+    next_run = datetime.combine(now.date(), target_time)
+
+    # If time has passed today, schedule for tomorrow
+    if next_run <= now:
+        next_run += timedelta(days=1)
+
+    delay = (next_run - now).total_seconds()
+
+    timer = threading.Timer(delay, trigger_helper)
+    timer.start()
+    print(f"Scheduled for {next_run}")
+
+
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
     gr.Markdown("## Space Rebooter")
@@ -178,4 +178,5 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
+    sanitize()
     demo.launch(server_name="0.0.0.0", server_port=7860)