Update app.py

app.py

@@ -1,9 +1,8 @@
 import os
 import requests
-import matplotlib.pyplot as plt
-import numpy as np
 import gradio as gr
 
+# Define Agent and Swarm classes based on fractal geometry
 class Agent:
     def __init__(self, id, api_key=None):
         self.id = id
@@ -49,81 +48,56 @@ class Swarm:
     def gather_results(self):
         return [agent.results for agent in self.agents if agent.results]
 
+# Generate tasks for the swarm
 def generate_tasks(api_url, num_tasks):
     return [api_url] * num_tasks
 
-def plot_pentagonal_and_mirrored(results):
-    fig, ax = plt.subplots()
-    ax.set_aspect('equal')
-
-    angle = 2 * np.pi / 5
-    radius = 1
-    pentagon_points = np.array([(radius * np.cos(i * angle), radius * np.sin(i * angle)) for i in range(5)])
-
-    for i in range(5):
-        ax.plot([pentagon_points[i][0], pentagon_points[(i + 1) % 5][0]],
-                [pentagon_points[i][1], pentagon_points[(i + 1) % 5][1]], 'k-')
-
-    center = np.array([0, 0])
-    points = [
-        center,
-        (center + pentagon_points[0]) / 2,
-        (center + pentagon_points[1]) / 2,
-        (center + pentagon_points[2]) / 2,
-        (center + pentagon_points[3]) / 2,
-        (center + pentagon_points[4]) / 2,
-        (pentagon_points[0] + pentagon_points[1]) / 2,
-        (pentagon_points[1] + pentagon_points[2]) / 2,
-        (pentagon_points[2] + pentagon_points[3]) / 2
-    ]
-
-    for i, point in enumerate(points[:len(results)]):
-        ax.plot(point[0], point[1], 'bo')
-        ax.text(point[0], point[1], results[i], fontsize=9, ha='right')
-
-        mirrored_x = [-point[0], point[0]]
-        mirrored_y = [-point[1], point[1]]
-
-        for mx in mirrored_x:
-            for my in mirrored_y:
-                if (mx, my) != (point[0], point[1]):
-                    ax.plot(mx, my, 'ro')
-                    ax.text(mx, my, results[i], fontsize=9, ha='right')
-
-    plt.show()
-
+# Function to run the swarm and gather results
 def run_swarm(api_url, api_key, num_agents, num_tasks):
+    # Create a swarm with a fractal pattern (Pentagonal spread)
     swarm = Swarm(num_agents=num_agents, fractal_pattern="Pentagonal", api_key=api_key)
     tasks = generate_tasks(api_url, num_tasks)
     swarm.assign_tasks(tasks)
     swarm.execute()
 
+    # Gather results
     results = swarm.gather_results()
 
+    # Print all results
     print("\nAll results retrieved by the swarm:")
     for i, result in enumerate(results):
         print(f"Result {i + 1}: {result}")
 
-    if results:
-        plot_pentagonal_and_mirrored(results)
-
     return results
 
+# Gradio interface
 def gradio_interface(api_url, api_key, num_agents, num_tasks):
     results = run_swarm(api_url, api_key, num_agents, num_tasks)
     return "\n".join(str(result) for result in results)
 
+# Default values for the inputs
+default_api_url = "https://meowfacts.herokuapp.com/"
+default_num_agents = 5
+default_num_tasks = 2
+
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
-        gr.Textbox(label="API URL", placeholder="Enter the API URL"),
+        gr.Textbox(label="API URL", placeholder="Enter the API URL", value=default_api_url),
         gr.Textbox(label="API Key (Optional)", placeholder="Enter the API Key"),
-        gr.Number(label="Number of Agents", value=9, precision=0),
-        gr.Number(label="Number of API Calls", value=9, precision=0)
+        gr.Number(label="Number of Agents", value=default_num_agents, precision=0),
+        gr.Number(label="Number of API Calls", value=default_num_tasks, precision=0)
     ],
     outputs=gr.Textbox(label="Results"),
-    title="Swarm API Call and Plotter",
-    description="Enter the API URL, API Key (Optional), number of agents, and number of API calls. The results will be plotted in a pentagonal pattern with mirrored points."
+    title="Swarm API Call and Result Gatherer",
+    description="""
+        This Gradio app demonstrates a swarm of agents making API calls and gathering results.
+        - The swarm is created based on a fractal geometry pattern.
+        - Each agent makes an API call to the specified URL and retrieves data.
+        - The results from all agents are gathered and displayed.
+        - Enter the API URL, API Key (optional), number of agents, and number of API calls to see the process in action.
+        - By default, the app uses the 'Meow Facts' API with 5 agents and 2 API calls.
+    """
 )
 
 iface.launch()