Update app.py

app.py

@@ -19,13 +19,10 @@ def daily_fortune(name: str) -> str:
     import random
     
     try:
-        # Invoke DuckDuckGoSearchTool via agent
-        search_result = agent.invoke_tool(
-            "DuckDuckGoSearchTool",
-            {"query": "top news today", "max_results": 1}
-        )
-        
-        headline = search_result.get('results', [{}])[0].get('snippet', 'No news found today.')
+        # Use DuckDuckGoSearchTool directly
+        search_tool = DuckDuckGoSearchTool()
+        search_results = search_tool(query="top news today", max_results=1)
+        headline = search_results.get('results', [{}])[0].get('snippet', 'No news found today.')
 
         # Random fortunes
         fortunes = [
@@ -35,16 +32,10 @@ def daily_fortune(name: str) -> str:
             "Today is a perfect day for creativity and adventure. 🎨🚀",
             "Someone from your past may reach out. Stay open. 💌"
         ]
-
-        # Pick a random fortune
         fortune = random.choice(fortunes)
 
-        # Generate image using the image_generation_tool
-        image = agent.invoke_tool(
-            "agents-course/text-to-image",
-            {"prompt": f"A digital painting representing '{fortune}' with a whimsical style"}
-        )
-
+        # Use image generation tool directly
+        image = image_generation_tool(prompt=f"A digital painting representing '{fortune}' in a mystical style")
         image_url = image.get('url', 'No image generated.')
 
         return (
@@ -53,12 +44,10 @@ def daily_fortune(name: str) -> str:
             f"🔮 Your Fortune: {fortune}\n"
             f"🖼️ Fortune Vibes Image: {image_url}\n"
         )
-
     except Exception as e:
         return f"Error generating your fortune: {str(e)}"
-        
-
 
+   
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -97,7 +86,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,daily_fortune], ## add your tools here (don't remove final answer)
+    tools=[final_answer,image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,