Upload app.py

app.py

@@ -12,17 +12,22 @@ def wakeup_spaces(username: str, hf_token: str="", is_private=True, progress=gr.
         api = HfApi(token=hf_token if hf_token else HF_TOKEN)
         space_infos = api.list_spaces(author=username)
         spaces = []
+        logs = []
         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"):
+            stage = api.get_space_runtime(repo_id=s).stage
+            if 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)
+                logs.append(f"'{s}' is in '{stage}'.")
+                logs.append(f"'{s}' is restarted.")
+        return "<br>\n".join(logs)
     except Exception as e:
         print(f"Error: {e}")
         raise gr.Error(f"Error: {e}")