Update app.py

app.py

@@ -16,10 +16,16 @@ def daily_fortune(name: str) -> str:
     Args:
         name: The name of the person receiving the fortune.
     """
+    import random
+    
     try:
-        # Fetch top news
-        search_results = DuckDuckGoSearchTool().search("top news today", max_results=1)
-        headline = search_results[0]['title'] if search_results else "No news found today."
+        # 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.')
 
         # Random fortunes
         fortunes = [
@@ -33,18 +39,25 @@ def daily_fortune(name: str) -> str:
         # Pick a random fortune
         fortune = random.choice(fortunes)
 
-        # Generate image for the vibe of the fortune
-        image_prompt = f"A digital painting representing '{fortune}' with a whimsical style"
-        image = image_generation_tool.generate(prompt=image_prompt)
+        # 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"}
+        )
+
+        image_url = image.get('url', 'No image generated.')
 
         return (
             f"🌞 Hello {name}!\n"
             f"📰 Top News Today: {headline}\n"
             f"🔮 Your Fortune: {fortune}\n"
-            f"🖼️ Fortune Vibes Image: {image['url']}\n"
+            f"🖼️ Fortune Vibes Image: {image_url}\n"
         )
+
     except Exception as e:
         return f"Error generating your fortune: {str(e)}"
+        
+
 
 
 @tool