upd flagging mode

app.py

@@ -181,30 +181,28 @@ def activate(hf_users: str, ms_users: str):
     if not ms_users:
         ms_users = hf_users
 
+    spaces, monitors, studios, output = [], [], [], []
     hf_usernames = hf_users.split(";")
     ms_usernames = ms_users.split(";")
-    spaces = []
     for user in tqdm(hf_usernames, desc="Collecting spaces..."):
         username = user.strip()
         if username:
             spaces += get_spaces(username)
             time.sleep(DELAY)
 
-    monitors, studios = [], []
     for space in spaces:
         if "keep-spaces-active" in space:
             monitors.append(space)
         else:
             studios.append(space)
-    spaces = monitors + studios
 
+    spaces = monitors + studios
     for user in tqdm(ms_usernames, desc="Collecting studios..."):
         username = user.strip()
         if username:
             spaces += get_studios(username)
             time.sleep(DELAY)
 
-    output = []
     for space in tqdm(spaces, desc="Activating spaces..."):
         output.append(
             {
@@ -249,45 +247,48 @@ def listasks():
     return "None"
 
 
-with gr.Blocks() as demo:
-    gr.Interface(
-        title="Start keeping all spaces active periodically",
-        fn=monitor,
-        inputs=[
-            gr.Textbox(
-                label="HuggingFace",
-                placeholder="Usernames joint by ;",
-            ),
-            gr.Textbox(
-                label="ModelScope",
-                placeholder="Usernames joint by ;",
-            ),
-        ],
-        outputs=None,
-        allow_flagging="never",
-    )
-    gr.Interface(
-        title="See current task status",
-        fn=listasks,
-        inputs=None,
-        outputs=gr.Textbox(label="Current task details"),
-        allow_flagging="never",
-    )
-    gr.Interface(
-        title="Test activation for all spaces once",
-        fn=activate,
-        inputs=[
-            gr.Textbox(
-                label="HuggingFace",
-                placeholder="Usernames joint by ;",
-            ),
-            gr.Textbox(
-                label="ModelScope",
-                placeholder="Usernames joint by ;",
-            ),
-        ],
-        outputs=gr.Dataframe(label="Activated spaces"),
-        allow_flagging="never",
-    )
-
-demo.launch()
+if __name__ == "__main__":
+    with gr.Blocks() as demo:
+        gr.Interface(
+            title="Start keeping all spaces active periodically",
+            fn=monitor,
+            inputs=[
+                gr.Textbox(
+                    label="HuggingFace",
+                    placeholder="Usernames joint by ;",
+                ),
+                gr.Textbox(
+                    label="ModelScope",
+                    placeholder="Usernames joint by ;",
+                ),
+            ],
+            outputs=None,
+            flagging_mode="never",
+        )
+
+        gr.Interface(
+            title="See current task status",
+            fn=listasks,
+            inputs=None,
+            outputs=gr.Textbox(label="Current task details"),
+            flagging_mode="never",
+        )
+
+        gr.Interface(
+            title="Test activation for all spaces once",
+            fn=activate,
+            inputs=[
+                gr.Textbox(
+                    label="HuggingFace",
+                    placeholder="Usernames joint by ;",
+                ),
+                gr.Textbox(
+                    label="ModelScope",
+                    placeholder="Usernames joint by ;",
+                ),
+            ],
+            outputs=gr.Dataframe(label="Activated spaces"),
+            flagging_mode="never",
+        )
+
+    demo.launch()