resetting the swarm regularly and completely

gradio_app/interacton_with_swarm.py

@@ -218,5 +218,6 @@ def read_swarm_logs():
         final_out = "Swarm is starting up..."
     return final_out
 
-def execute_swarm():
-    run_swarm()
+def execute_swarm(smarm_ui_status):
+    _ = run_swarm() # when it finishes, it will set SWARM_IS_RUNNING to False
+    smarm_ui_status = False #let's hope it's mutable

gradio_app/interface.py

@@ -32,16 +32,21 @@ def swarm_interface(swarm_role, swarm_global_goal, swarm_goals, n_managers, n_an
     ]
     set_swarm_agents_config(agents_config)
 
-    t = threading.Thread(target=execute_swarm)
+    t = threading.Thread(target=execute_swarm, args=(SWARM_IS_RUNNING,))
     t.start()
-    print("Swarm is running")
     SWARM_IS_RUNNING = True
+    print(f"Swarm is running. SWARM_IS_RUNNING = {SWARM_IS_RUNNING}")
 
 def create_gradio_interface():
     title = """
     <h1 align="center">🐝🐝 Swarm Intelligence 🐝🐝</h1>
     <div align="center">
     <a style="display:inline-block" href='https://github.com/nicelir1996/GPT-Swarm'><img src='https://img.shields.io/github/stars/nicelir1996/GPT-Swarm?style=social' /></a>
+    </div>
+    <br>
+    <div align="center">
+    ⚠️⚠️We Highly recommend to use a duplicate space!⚠️⚠️
+    <br> Due to huggingface's API limitations, there can only be one instance of the swarm running at a time. You can see it's output, but your input will be ignored. The swarm resets every 10 minutes, so maybe you get lucky.
     <a href="https://huggingface.co/spaces/swarm-agents/swarm-agents?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
     </div>
     """
@@ -95,6 +100,7 @@ def create_gradio_interface():
             return display_logs(), display_output()
         
         def submit_callback(swarm_role, swarm_global_goal, swarm_goals, n_managers, n_analysts, n_googlers):
+            print(f"Swarm is running. SWARM_IS_RUNNING = {SWARM_IS_RUNNING}")
             if not SWARM_IS_RUNNING:
                 swarm_interface(swarm_role, swarm_global_goal, swarm_goals, n_managers, n_analysts, n_googlers)
             return display_logs(), display_output()

swarm_config.yaml

@@ -7,7 +7,7 @@ swarm:
   - n: 2
     type: googler
   run_dir: ./tmp/swarm
-  timeout_min: 10
+  timeout_min: 1
 task:
   global_goal: A new startup just send us their pitch. Find if the startup is worth
     investing in. The startup is called Brainamics and it is in the space of brain

swarmai/Swarm.py

@@ -5,6 +5,7 @@ import yaml
 import threading
 import os
 import json
+import shutil
 
 from pathlib import Path
 
@@ -80,6 +81,16 @@ class Swarm:
         self.output_file = str((self.data_dir / 'output.txt').resolve())
         with open(self.output_file, 'w') as f:
             f.write("")
+        
+        out_json = Path(str(self.output_file).replace(".txt", ".json"))
+        if out_json.exists():
+            with open(self.output_file, 'w') as f:
+                f.write("")
+        
+        out_pretty = Path(str(self.output_file).replace(".txt", "_pretty.txt"))
+        if out_pretty.exists():
+            with open(self.output_file, 'w') as f:
+                f.write("")
 
         # creating task queue
         self.task_queue = PandasQueue(self.TASK_TYPES, self.WORKER_ROLES.keys(), self.TASK_ASSOCIATIONS)
@@ -153,6 +164,7 @@ class Swarm:
         self.stop()
 
         self.log("All agents have finished their work")
+        return True
 
     def create_report_qa_task(self):
         """Creates a task that will be used to evaluate the report quality.
@@ -204,6 +216,12 @@ class Swarm:
         self.timeout = config["swarm"]["timeout_min"]*60
         
         self.data_dir = Path(".", config["swarm"]["run_dir"]).resolve()
+        # first, try to delete the directory with all the data
+        try:
+            shutil.rmtree(self.data_dir)
+        except Exception:
+            pass
+
         self.data_dir.mkdir(parents=True, exist_ok=True)
 
         # getting the tasks