Update app.py

app.py

@@ -53,33 +53,15 @@ class Swarm:
     def gather_results(self):
         return [agent.results for agent in self.agents if agent.results]
 
-def generate_tasks_from_model(api_url, num_tasks):
-    tasks = []
-    for _ in range(num_tasks):
-        prompt = f"Generate an API call task for the following API URL: {api_url}"
-        inputs = tokenizer(prompt, return_tensors="pt")
-        print(f"Generating task with prompt: {prompt}")
-        try:
-            outputs = model.generate(**inputs, max_length=50)  # Limit max length for speed
-            task = tokenizer.decode(outputs[0], skip_special_tokens=True)
-            print(f"Generated task: {task}")
-            # Check if the generated task is a valid URL
-            if task.startswith("http://") or task.startswith("https://"):
-                tasks.append(task)
-            else:
-                print(f"Invalid task generated: {task}")
-                tasks.append(f"Error: Invalid task generated: {task}")
-        except Exception as e:
-            print(f"Error generating task: {str(e)}")
-            tasks.append(f"Error generating task: {str(e)}")
-    return tasks
+def generate_tasks(api_url, num_tasks):
+    return [api_url] * num_tasks
 
 def run_swarm(api_url, api_key, num_agents, num_tasks):
     try:
-        tasks = generate_tasks_from_model(api_url, num_tasks)
+        tasks = generate_tasks(api_url, num_tasks)
         print(f"Generated tasks: {tasks}")
     except Exception as e:
-        return f"Error generating tasks from model: {str(e)}"
+        return f"Error generating tasks: {str(e)}"
 
     try:
         swarm = Swarm(num_agents=num_agents, fractal_pattern="Pentagonal", api_key=api_key)
@@ -99,11 +81,15 @@ def run_swarm(api_url, api_key, num_agents, num_tasks):
 
     return results
 
-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)
+def gradio_interface(prompt, api_url, api_key, num_agents, num_tasks):
+    if prompt == "Autobots, Assemble!":
+        results = run_swarm(api_url, api_key, num_agents, num_tasks)
+        return "\n".join(str(result) for result in results)
+    else:
+        return "Prompt not recognized. Please use 'Autobots, Assemble!' to execute the swarm."
 
 # Default values for the inputs
+default_prompt = "Autobots, Assemble!"
 default_api_url = "https://meowfacts.herokuapp.com/"
 default_api_key = ""
 default_num_agents = 5
@@ -112,6 +98,7 @@ default_num_tasks = 2
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
+        gr.Textbox(label="Prompt", placeholder="Enter the prompt", value=default_prompt),
         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", value=default_api_key),
         gr.Number(label="Number of Agents", value=default_num_agents, precision=0),
@@ -120,14 +107,13 @@ iface = gr.Interface(
     outputs=gr.Textbox(label="Results"),
     title="Swarm Model Processing and Result Gatherer",
     description="""
-        This Gradio app demonstrates a swarm of agents using a language model to generate API call tasks and gather results.
-        - The language model generates API calls based on the provided API URL.
-        - The swarm is created based on a fractal geometry pattern.
-        - Each agent makes an API call to the generated URLs and retrieves data.
+        This Gradio app demonstrates a swarm of agents making API calls and gathering results.
+        - When the prompt 'Autobots, Assemble!' is entered, a swarm is created and the API calls are executed.
+        - 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 tasks to see the process in action.
-        - By default, the app uses the API URL 'https://meowfacts.herokuapp.com/' with 5 agents and 2 tasks.
+        - Enter the prompt 'Autobots, Assemble!', API URL, API Key (optional), number of agents, and number of tasks to see the process in action.
+        - By default, the app uses the prompt 'Autobots, Assemble!' and the API URL 'https://meowfacts.herokuapp.com/' with 5 agents and 2 tasks.
     """
 )
 
-iface.launch()
+iface.launch()