Upload 3 files

README.md

@@ -1,12 +1,13 @@
----
-title: Space Waker
-emoji: 🐨
-colorFrom: yellow
-colorTo: green
-sdk: gradio
-sdk_version: 5.4.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Space Waker
+emoji: ⏰
+colorFrom: indigo
+colorTo: purple
+sdk: gradio
+sdk_version: 4.40.0
+app_file: app.py
+pinned: false
+license: mit
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,42 @@
+import gradio as gr
+from huggingface_hub import HfApi
+import os
+
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+HF_USER = os.environ.get("HF_USER")
+
+
+def wakeup_spaces(username: str, hf_token: str="", is_private=True, progress=gr.Progress(track_tqdm=True)):
+    try:
+        api = HfApi(token=hf_token if hf_token else HF_TOKEN)
+        space_infos = api.list_spaces(author=username)
+        spaces = []
+        for s in space_infos:
+            if s.private and not is_private: continue
+            spaces.append(s.id)
+        for s in spaces:
+            if api.get_space_runtime(repo_id=s).stage in ("RUNNING", "APP_STARTING", "BUILDING"):
+                print(f"'{s}' is running.")
+                progress(0, desc=f"'{s}' is running.")
+            else:
+                print(f"'{s}' isn't running. Restarting...")
+                progress(0, desc=f"'{s}' isn't running. Restarting...")
+                api.restart_space(repo_id=s)
+    except Exception as e:
+        print(f"Error: {e}")
+        raise gr.Error(f"Error: {e}")
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        username = gr.Textbox(label="HF Username", value=HF_USER, lines=1)
+        token = gr.Textbox(label="HF Write Token", value="", lines=1)
+        is_private = gr.Checkbox(label="Apply to private space", value=True)
+    run_button = gr.Button("Wake up spaces", variant="primary")
+    info_md = gr.Markdown("<br><br><br>")
+
+    run_button.click(wakeup_spaces, [username, is_private], [info_md])
+
+demo.launch()
+

requirements.txt

@@ -0,0 +1 @@
+huggingface_hub