add try

activate.py

@@ -121,44 +121,50 @@ def activate_studio(repo: str, holding_delay=5):
 
 
 def trigger(users=USERS):
-    spaces, studios, failures = [], [], []
-    usernames = users.split(";")
-    for user in tqdm(usernames, desc="Collecting spaces"):
-        username = user.strip()
-        if username:
-            sleeps, errors = get_spaces(username)
-            spaces += sleeps
-            failures += errors
+    try:
+        spaces, studios, failures = [], [], []
+        usernames = users.split(";")
+        for user in tqdm(usernames, desc="Collecting spaces"):
+            username = user.strip()
+            if username:
+                sleeps, errors = get_spaces(username)
+                spaces += sleeps
+                failures += errors
+                time.sleep(DELAY)
+
+        for space in tqdm(spaces, desc="Activating spaces"):
+            activate_space(space)
             time.sleep(DELAY)
 
-    for space in tqdm(spaces, desc="Activating spaces"):
-        activate_space(space)
-        time.sleep(DELAY)
-
-    for user in tqdm(usernames, desc="Collecting studios"):
-        username = user.strip()
-        if username:
-            sleeps, errors = get_studios(username)
-            studios += sleeps
-            failures += errors
+        for user in tqdm(usernames, desc="Collecting studios"):
+            username = user.strip()
+            if username:
+                sleeps, errors = get_studios(username)
+                studios += sleeps
+                failures += errors
+                time.sleep(DELAY)
+
+        for studio in tqdm(studios, desc="Activating studios"):
+            threading.Thread(
+                target=activate_studio, args=(studio,), daemon=True
+            ).start()
             time.sleep(DELAY)
 
-    for studio in tqdm(studios, desc="Activating studios"):
-        threading.Thread(target=activate_studio, args=(studio,), daemon=True).start()
-        time.sleep(DELAY)
+        log = (
+            "\n".join(spaces + studios)
+            + f"\n[{fix_datetime(datetime.now())}] Activation complete!"
+        )
+        print(log)
+        content = ""
+        for failure in failures:
+            errepo: str = failure
+            errepo = errepo.replace(HF_DOMAIN, "").replace(MS_DOMAIN, "")
+            content += f"<br><a href='{failure}'>{errepo[1:]}</a><br>"
+
+        if content:
+            send_email(content)
 
-    log = (
-        "\n".join(spaces + studios)
-        + f"\n[{fix_datetime(datetime.now())}] Activation complete!"
-    )
-    print(log)
-    content = ""
-    for failure in failures:
-        errepo: str = failure
-        errepo = errepo.replace(HF_DOMAIN, "").replace(MS_DOMAIN, "")
-        content += f"<br><a href='{failure}'>{errepo[1:]}</a><br>"
-
-    if content:
-        send_email(content)
-
-    return log
+        return log
+
+    except Exception as e:
+        return f"{e}"

app.py

@@ -25,36 +25,43 @@ def monitor(period=PERIOD):
 
 
 def tasklist():
-    outputs = f"Has been running for {time_diff(START_TIME, datetime.now())}"
-    jobs = schedule.get_jobs()
-    for job in jobs:
-        last_run = fix_datetime(job.last_run)
-        if not last_run:
-            last_run = "never"
+    try:
+        outputs = f"Has been running for {time_diff(START_TIME, datetime.now())}"
+        jobs = schedule.get_jobs()
+        for job in jobs:
+            last_run = fix_datetime(job.last_run)
+            if not last_run:
+                last_run = "never"
 
-        next_run = fix_datetime(job.next_run)
-        outputs += f"\n\nEvery {job.interval}h do {job.job_func} (last run: {last_run}, next run: {next_run})"
+            next_run = fix_datetime(job.next_run)
+            outputs += f"\n\nEvery {job.interval}h do {job.job_func} (last run: {last_run}, next run: {next_run})"
 
-    return outputs
+        return outputs
+
+    except Exception as e:
+        return f"{e}"
 
 
 if __name__ == "__main__":
     monitor()
     with gr.Blocks() as demo:
-        gr.Interface(
-            title="See current task status",
-            fn=tasklist,
-            inputs=None,
-            outputs=gr.TextArea(label="Current task details"),
-            flagging_mode="never",
-        )
-
-        gr.Interface(
-            title="Trigger once manually",
-            fn=trigger,
-            inputs=None,
-            outputs=gr.TextArea(label="Activation logs"),
-            flagging_mode="never",
-        )
+        with gr.Row():
+            with gr.Column():
+                gr.Interface(
+                    title="See current task status",
+                    fn=tasklist,
+                    inputs=None,
+                    outputs=gr.TextArea(label="Current task details"),
+                    flagging_mode="never",
+                )
+
+            with gr.Column():
+                gr.Interface(
+                    title="Trigger once manually",
+                    fn=trigger,
+                    inputs=None,
+                    outputs=gr.TextArea(label="Activation logs"),
+                    flagging_mode="never",
+                )
 
     demo.launch()