Update app.py

app.py

@@ -8,28 +8,19 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
-# @tool
-# def search_ddgs(query:str, max_results:int=5)-> list: #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 searches the internet to answer user's question and returns top research results.
-    
-#     Args:
-#         query: The search query
-#         max_results: Number of results to return. Maximum number is 5. 
-
-#     Returns:
-#     List of dictionaries with title, URL, and description.
-#     """
-#     results = []
-#     with DuckDuckGoSearchTool as ddgs:
-#         for result in ddgs.text(query, max_results = max_results):
-#             results.append({
-#                 "title": result["title"],
-#                 "url": result["href"],
-#                 "snippet":result["body"]
-#             })
-    
-#     return results
+@tool
+def my_search_tool(query:str)-> 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 searches DuckDuckGo for a given query.
+    Args:
+        query: The search query.
+    Returns:
+        response: The response with the search results.
+    """
+    search_tool = DuckDuckGoSearchTool()
+    search_results = search_tool(query)
+    response = f"Here's what i've found about {query}: {search_results}"
+    return response
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -49,13 +40,6 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# Search tool
-search_tool =  DuckDuckGoSearchTool()
-# search_tool =  DuckDuckGoSearchTool(max_results=3, output_format="json")
-
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
@@ -67,14 +51,15 @@ custom_role_conversions=None,
 )
 
 
-
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, search_tool, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[my_search_tool, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,