Update app.py

app.py

@@ -24,7 +24,7 @@ class Agent:
                 if response.status_code == 200:
                     self.results = response.json()
                 else:
-                    self.results = "Error: Unable to fetch data"
+                    self.results = f"Error: Unable to fetch data, status code {response.status_code}"
                 print(f"Agent {self.id} received: {self.results}")
             except Exception as e:
                 self.results = f"Error: {str(e)}"
@@ -61,14 +61,27 @@ def generate_tasks_from_model(api_url, num_tasks):
     return tasks
 
 def run_swarm(api_url, api_key, num_agents, num_tasks):
-    tasks = generate_tasks_from_model(api_url, num_tasks)
-    swarm = Swarm(num_agents=num_agents, fractal_pattern="Pentagonal", api_key=api_key)
-    swarm.assign_tasks(tasks)
-    swarm.execute()
-    results = swarm.gather_results()
+    try:
+        tasks = generate_tasks_from_model(api_url, num_tasks)
+    except Exception as e:
+        return f"Error generating tasks from model: {str(e)}"
+
+    try:
+        swarm = Swarm(num_agents=num_agents, fractal_pattern="Pentagonal", api_key=api_key)
+        swarm.assign_tasks(tasks)
+        swarm.execute()
+    except Exception as e:
+        return f"Error executing swarm tasks: {str(e)}"
+
+    try:
+        results = swarm.gather_results()
+    except Exception as e:
+        return f"Error gathering results: {str(e)}"
+
     print("\nAll results retrieved by the swarm:")
     for i, result in enumerate(results):
         print(f"Result {i + 1}: {result}")
+
     return results
 
 def gradio_interface(api_url, api_key, num_agents, num_tasks):
@@ -102,4 +115,4 @@ iface = gr.Interface(
     """
 )
 
-iface.launch()
+iface.launch()