some fixes with finishing

gradio_app/interacton_with_swarm.py

@@ -220,4 +220,4 @@ def read_swarm_logs():
 
 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
+    raise TimeoutError("Swarm timed out. Please try again.")

gradio_app/interface.py

@@ -11,6 +11,7 @@ sys.path.append(str(root_dir))
 from gradio_app.interacton_with_swarm import *
 
 SWARM_IS_RUNNING = False
+SWARM_THREAD = None
 
 def display_logs():
     return read_swarm_logs()
@@ -19,7 +20,7 @@ def display_output():
     return read_swarm_output()
 
 def swarm_interface(swarm_role, swarm_global_goal, swarm_goals, n_managers, n_analysts, n_googlers):
-    global SWARM_IS_RUNNING
+    global SWARM_IS_RUNNING, SWARM_THREAD
     # please, don't judge me for this hardcoding. it's 3am and it's the first time i use gradio =)))
     # Call the necessary set_ functions with the user inputs
     set_swarm_role(swarm_role)
@@ -32,12 +33,14 @@ 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, args=(SWARM_IS_RUNNING,))
-    t.start()
+    SWARM_THREAD = threading.Thread(target=execute_swarm, args=(SWARM_IS_RUNNING,))
+    SWARM_THREAD.start()
     SWARM_IS_RUNNING = True
+
     print(f"Swarm is running. SWARM_IS_RUNNING = {SWARM_IS_RUNNING}")
 
 def create_gradio_interface():
+    global SWARM_IS_RUNNING, SWARM_THREAD
     title = """
     <h1 align="center">🐝🐝 Swarm Intelligence 🐝🐝</h1>
     <div align="center">
@@ -100,6 +103,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):
+            SWARM_IS_RUNNING = SWARM_THREAD.is_alive()
             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)