Update app.py

app.py

@@ -37,6 +37,31 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def generate_themed_image(subject: str, style: str) -> str:
+    """Generates an image by combining a subject and an art style into a prompt.
+
+    Args:
+        subject: The main subject of the image (e.g., 'a cat on a beach').
+        style: The artistic style to apply (e.g., 'watercolor', 'pixel art', 'oil painting').
+    """
+    prompt = f"{subject} in {style} style"
+    return image_generation_tool(prompt)
+
+@tool
+def search_and_summarize(query: str, max_results: int) -> str:
+    """Searches the web using DuckDuckGo and returns the top results for a given query.
+
+    Args:
+        query: The search term or question to look up.
+        max_results: The number of results to return, between 1 and 5.
+    """
+    max_results = max(1, min(max_results, 5))
+    search = DuckDuckGoSearchTool()
+    results = search(query)
+    lines = results.strip().split("\n")
+    return "\n".join(lines[:max_results])
+
 # The Agent
 
 final_answer = FinalAnswerTool()
@@ -60,7 +85,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool], # added my_custom_tool here
+    tools=[final_answer, my_custom_tool, generate_themed_image, search_and_summarize], # added my additional tools here
     max_steps=6,
     verbosity_level=1,
     grammar=None,