Update app.py

app.py

@@ -34,29 +34,50 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-@tool
-def news_to_image(topic: str) -> str:
+#@tool
+#def news_to_image(topic: str) -> str:
     """
     A tool that searches for recent news on a given topic and creates an image from the top headline.
 
     Args:
         topic: The subject to search for.
     """
-    try:
+ #   try:
         # Use DuckDuckGo to search for news
-        search_tool = DuckDuckGoSearchTool()
-        results = search_tool.run(f"{topic} news")  # Returns a list of links and snippets
+      #  search_tool = DuckDuckGoSearchTool()
+       # results = search_tool.run(f"{topic} news")  # Returns a list of links and snippets
 
         # Extract the first headline (title or snippet)
-        first_result = results[0]["snippet"] if results else "Breaking news"
+        #first_result = results[0]["snippet"] if results else "Breaking news"
 
         # Generate the image using your existing image_generation_tool
-        image_url = image_generation_tool.run(prompt=first_result)
+        #image_url = image_generation_tool.run(prompt=first_result)
+
+        #return f"Created image for headline: '{first_result}' → {image_url}"
 
-        return f"Created image for headline: '{first_result}' → {image_url}"
+  #  except Exception as e:
+       # return f"Failed to generate image from news: {str(e)}"
+
+
+@tool
+def news_to_image(topic: str) -> str:
+    """
+    Fetches recent news on a topic and generates an image from the top headline.
+    Args:
+        topic: The news topic to search for.
+    """
+    # Instead of calling DuckDuckGoSearchTool directly...
+    search_tool = agent.tool_dict["DuckDuckGoSearchTool"]
+    image_tool = agent.tool_dict["TextToImageTool"]  # Assuming that's the name of your imported tool
+
+    try:
+        search_results = search_tool(topic + " news")
+        first_result = search_results[0]["snippet"] if search_results else "Breaking news"
+        image = image_tool(first_result)
+        return f"Image created for: '{first_result}' → {image}"
 
     except Exception as e:
-        return f"Failed to generate image from news: {str(e)}"
+        return f"Failed to generate image: {str(e)}"
 
 final_answer = FinalAnswerTool()