restart assigns on trigger

modules/activate.py

@@ -7,6 +7,7 @@ from datetime import datetime
 from huggingface_hub import HfApi
 from modules.smtp import send_email
 from modules.times import fix_datetime
+from modules.restart import restart_repos
 from utils import HF_DOMAIN, HEADER, MS_HEADER, TIMEOUT, MS_DOMAIN, DELAY, USERS
 
 
@@ -184,6 +185,8 @@ def trigger(users=USERS):
         if content:
             send_email(content)
 
+        restart_repos()
+
     except Exception as e:
         status = f"{e}"
 

modules/restart.py

@@ -1,4 +1,5 @@
 import requests
+from tqdm import tqdm
 from modules.smtp import send_email
 from utils import HF_DOMAIN, MS_DOMAIN, HF_HEADER, MS_HEADER, TIMEOUT, REPOS
 
@@ -23,18 +24,19 @@ def restart_private_space(repo: str):
     return f"\n{response.text}\n"
 
 
+def restart_repos():
+    if REPOS:
+        repos = REPOS.split(";")
+        for repo in tqdm(repos, desc="Restarting repos"):
+            private_repo = repo.strip()
+            print(f"Restarting {private_repo}...")
+            restart_private_studio(private_repo)
+            restart_private_space(private_repo)
+
+
 def self_restart():
     try:
-        if REPOS:
-            repos = REPOS.split(";")
-            for repo in repos:
-                private_repo = repo.strip()
-                print(f"Restarting {private_repo}...")
-                restart_private_studio(private_repo)
-                restart_private_space(private_repo)
-
         restart_private_space("kakamond/keeps_alive")
-
     except Exception as e:
         send_email(f"Failed to self-restart: {e}")