Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -6,61 +6,80 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Custom tool for web search to find authentic Ethiopian food images
+# Define custom tools
+
 @tool
-def search_ethiopian_food_authentic(food_name: str) -> str:
-    """Search for authentic Ethiopian food images to understand exact appearance
+def search_images(query: str) -> str:
+    """A tool that searches for images using DuckDuckGo and returns URLs of the top 3 results.
     Args:
-        food_name: Name of the Ethiopian food (e.g., "injera", "doro wat", "tibs")
+        query: The search query for images (e.g., 'Ethiopian injera food').
     """
     try:
-        # Use web search to find authentic images and descriptions
-        search_query = f"authentic Ethiopian {food_name} traditional appearance ingredients presentation"
-        # This would connect to your web_search_exa tool
-        return f"Searching for authentic Ethiopian {food_name}..."
+        from duckduckgo_search import DDGS
+        with DDGS() as ddgs:
+            results = ddgs.images(query, max_results=3)
+            image_urls = [result['image'] for result in results]
+        return "\n".join(image_urls)
     except Exception as e:
-        return f"Error searching for {food_name}: {str(e)}"
+        return f"Error searching images: {str(e)}"
 
-# Custom tool for generating Ethiopian food images
 @tool
-def generate_ethiopian_food_image(food_description: str, style: str = "authentic") -> str:
-    """Generate an authentic Ethiopian food image based on research
+def visit_webpage(url: str) -> str:
+    """A tool that visits a webpage and returns its content in markdown format.
     Args:
-        food_description: Detailed description of the food to generate
-        style: Style of image - "authentic", "restaurant", "home-cooked"
+        url: The URL of the webpage to visit.
     """
     try:
-        # This will use the gr1_qwen_image_fast_generate_image tool
-        return f"Generating {style} Ethiopian food image: {food_description}"
+        response = requests.get(url)
+        response.raise_for_status()
+        from markdownify import markdownify as md
+        return md(response.text)
     except Exception as e:
-        return f"Error generating image: {str(e)}"
+        return f"Error visiting webpage: {str(e)}"
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 final_answer = FinalAnswerTool()
 
+# Load image generation tool from Hub
+image_generator = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+# Instantiate web search tool
+web_search = DuckDuckGoSearchTool()
+
 # Model configuration
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # Use another endpoint if overloaded
     custom_role_conversions=None,
 )
 
-# Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the Ethiopian food agent
+# Initialize the agent with all tools
 agent = CodeAgent(
     model=model,
-    tools=[search_ethiopian_food_authentic, generate_ethiopian_food_image, final_answer],
+    tools=[web_search, search_images, visit_webpage, image_generator, final_answer],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name="ethiopian_food_image_generator",
-    description="Generates authentic Ethiopian food images by researching exact appearance first",
+    name=None,
+    description=None,
     prompt_templates=prompt_templates
 )
 
-# Launch the Gradio UI
 GradioUI(agent).launch()