Upload folder using huggingface_hub

README.md

@@ -8,5 +8,3 @@ sdk_version: 6.9.0
 app_file: app.py
 pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,158 @@
+import gradio as gr
+import os
+from time import sleep
+from huggingface_hub import HfApi
+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
+#
+#
+#
+#
+
+
+# --- 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")
+REPOS = [t.strip() for t in repos_env.split(",")]
+RESTART_TIMES = [t.strip() for t in times_env.split(",")]
+
+
+# Global variable to track the specific target timestamp
+next_reboot_datetime = None
+
+
+def get_next_scheduled_time():
+    """Finds the next scheduled time from the list relative to now."""
+    now = datetime.now()
+    candidates = []
+
+    # Convert all string times to datetime objects for today
+    for t_str in RESTART_TIMES:
+        try:
+            h, m = map(int, t_str.split(':'))
+            # Create a datetime for today at this time
+            dt = now.replace(hour=h, minute=m, second=0, microsecond=0)
+            candidates.append(dt)
+        except ValueError:
+            print(f"Invalid time format: {t_str}")
+
+    candidates.sort()
+
+    # Find the first time that is in the future
+    for dt in candidates:
+        if dt > now:
+            return dt
+
+    # If no time is left today, pick the earliest time and move it to tomorrow
+    return candidates[0] + timedelta(days=1)
+
+
+# Initialize the first target time on startup
+next_reboot_datetime = get_next_scheduled_time()
+
+
+def time_until_next_reboot():
+    """Calculates and formats the time until the next planned reboot."""
+    global next_reboot_datetime
+
+    # Safety check in case it wasn't set
+    if next_reboot_datetime is None:
+        next_reboot_datetime = get_next_scheduled_time()
+
+    remaining_time = next_reboot_datetime - datetime.now()
+
+    # If we are past the time (but haven't triggered yet), show 0
+    if remaining_time.total_seconds() <= 0:
+        return "Rebooting..."
+
+    mins, secs = divmod(remaining_time.seconds, 60)
+    hours, mins = divmod(mins, 60)
+    return f"{hours:02}:{mins:02}:{secs:02}"
+
+
+def get_last_reboot_status():
+    """Returns the current target schedule info."""
+    global next_reboot_datetime
+    return f"Next target: {next_reboot_datetime.strftime('%Y-%m-%d %H:%M:%S')}"
+
+
+def check_and_reboot():
+    """Checks if current time has passed the scheduled time."""
+    global next_reboot_datetime
+
+    now = datetime.now()
+
+    # Check if we have reached or passed the target time
+    if now >= next_reboot_datetime:
+        status_msg = f"Reboot triggered at {now.strftime('%H:%M:%S')}"
+        print(status_msg)
+
+        # 1. Execute Reboots
+        if HF_TOKEN:
+            api = HfApi()
+            for repo in REPOS:
+                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)
+                except Exception as e:
+                    print(f"Failed to restart {repo}: {e}")
+
+        # 2. IMPORTANT: Advance the schedule immediately to prevent infinite loops
+        # We calculate the NEXT time starting from 1 minute in the future
+        # to ensure we don't pick the same time slot we just triggered.
+        next_reboot_datetime = get_next_scheduled_time()
+
+        return f"Just rebooted. Next scheduled: {next_reboot_datetime.strftime('%H:%M:%S')}"
+
+    return "Space running normally."
+
+
+def pipeline():
+    # Run the check
+    status = check_and_reboot()
+
+    # Return updated UI elements
+    countdown = time_until_next_reboot()
+    target_info = get_last_reboot_status()
+
+    return countdown, status, target_info
+
+
+# --- Gradio Interface ---
+with gr.Blocks() as demo:
+    gr.Markdown("## Space Rebooter")
+
+    with gr.Row():
+        next_reboot_display = gr.Textbox(label="Scheduled Target", value=get_last_reboot_status())
+        next_reboot_countdown = gr.Textbox(label="Time Until Next Reboot", value=time_until_next_reboot())
+
+    status_output = gr.Textbox(label="Status Log", value="Space running normally.")
+
+    # Timer ticks every 1 second
+    timer = gr.Timer(interval=1).tick(
+        fn=pipeline,
+        inputs=None,
+        outputs=[next_reboot_countdown, status_output, next_reboot_display],
+        every=1
+    )
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)