Update app.py

app.py

@@ -6,9 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-@tool
 def my_custom_tool(query: str, max_results: int = 3) -> list[str]:
     """
     Search DuckDuckGo and return up to max_results hits.
@@ -21,6 +19,8 @@ def my_custom_tool(query: str, max_results: int = 3) -> list[str]:
     search = DuckDuckGoSearchTool()
     return search.run(query, max_results=max_results)
 
+
+
 @tool
 def generate_images(prompt: str, n: int = 3) -> list[str]:
     """
@@ -34,8 +34,21 @@ def generate_images(prompt: str, n: int = 3) -> list[str]:
     #image_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
     #return image_tool.run(prompt, max_results=n)
     return image_generation_tool.run(prompt, max_results=n)
+@tool
+def ss(prompt: str) -> str:
+    """
+    Generate up to n images matching the prompt.
+    Args:
+        prompt: the text description for the image
+        n: number of images to generate
+    Returns:
+        A list of image asset IDs.
+    """
+    #image_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    #return image_tool.run(prompt, max_results=n)
 
-
+    search=DuckDuckGoSearchTool()
+    return search.run(prompt)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -74,7 +87,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,my_custom_tool,generate_images,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer,my_custom_tool,generate_images,get_current_time_in_timezone,ss], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,