Update app.py

app.py

@@ -9,14 +9,59 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def my_custom_tool(activity_type: str, participants: int) -> str:
+    """
+    A tool that fetches a random activity from the Bored API, allowing you to specify
+    an activity type and the number of participants.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        activity_type: The type of activity to search for. Possible values include:
+            'education', 'recreational', 'social', 'diy', 'charity', 'cooking',
+            'relaxation', 'music', 'busywork'.
+        participants: The number of participants needed for the activity.
+
+    Returns:
+        A string describing the suggested activity, with additional information
+        such as its accessibility, price, and link if available.
     """
-    return "What magic will you build ?"
+    try:
+        response = requests.get(
+            "https://www.boredapi.com/api/activity",
+            params={
+                "type": activity_type.lower(),
+                "participants": participants
+            }
+        )
+        if response.status_code != 200:
+            return f"Failed to fetch an activity. HTTP status code: {response.status_code}"
+
+        data = response.json()
+
+        # 如果 Bored API 返回 error，表示没有符合条件的活动
+        if data.get("error"):
+            return (
+                f"Sorry, no activity found for type='{activity_type}' with "
+                f"participants={participants}. Try different parameters!"
+            )
+        
+        # 从返回的数据中提取信息
+        activity = data.get("activity", "N/A")
+        a_type = data.get("type", "N/A")
+        price = data.get("price", "N/A")
+        accessibility = data.get("accessibility", "N/A")
+        link = data.get("link", "No link provided")
+
+        return (
+            f"Here's a fun idea:\n\n"
+            f"Activity: {activity}\n"
+            f"Type: {a_type}\n"
+            f"Price: {price}\n"
+            f"Accessibility: {accessibility}\n"
+            f"Link: {link}"
+        )
+
+    except Exception as e:
+        return f"An error occurred while fetching activity suggestions: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +100,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[image_generation_tool,final_answer], ## add your tools here (don't remove final answer)
+    tools=[image_generation_tool,my_custom_tool,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,