Update app.py

added a new tool called promote Chakwal

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, image_generation_tool
 import datetime
 import requests
 import pytz
@@ -52,6 +52,45 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
     
     return "\n".join(messages)
 
+@tool
+def promote_chakwal(arg1: str, arg2: int) -> str:
+    """
+    A tool that promotes tourism and investment in Chakwal by fetching live information 
+    and generating a visual representation of the city's attractions.
+    
+    Args:
+        arg1: A custom note or call-to-action message.
+        arg2: The number of search results to include from DuckDuckGo regarding tourism/investment.
+    """
+    output_lines = []
+    
+    # Use DuckDuckGoSearchTool to fetch current news or information on Chakwal tourism and investment
+    duck_tool = DuckDuckGoSearchTool()
+    search_query = "Chakwal tourism investment latest news attractions"
+    search_results = duck_tool.run(search_query)
+    
+    # Process and include search results (assuming a list of result dictionaries)
+    if isinstance(search_results, list):
+        output_lines.append("Latest Information on Chakwal:")
+        for idx, result in enumerate(search_results[:arg2]):
+            title = result.get("title", "No Title")
+            snippet = result.get("snippet", "No snippet available.")
+            output_lines.append(f"{idx+1}. {title} - {snippet}")
+    else:
+        output_lines.append(f"Search Results: {search_results}")
+    
+    # Use the text-to-image generation tool to create an attractive image prompt for Chakwal
+    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    image_prompt = "A stunning view of Chakwal with lush landscapes, historic landmarks, and vibrant local culture"
+    generated_image = image_generation_tool.run(image_prompt)
+    output_lines.append("\nVisual Promotion:")
+    output_lines.append(f"Generated Image URL/Result: {generated_image}")
+    
+    # Append any additional note from arg1
+    if arg1:
+        output_lines.append(f"\nAdditional Note: {arg1}")
+    
+    return "\n".join(output_lines)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -90,7 +129,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,my_custom_tool,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer,my_custom_tool,get_current_time_in_timezone,promote_chakwal], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,