Update app.py

app.py

@@ -16,38 +16,32 @@ def my_custom_tool(search_query: str, generate_image: bool = False) -> str:
         search_query: The search query to look up
         generate_image: Whether to also generate an image based on the search results
     """
+    print("my_custom_tool called with query:", search_query)  # Debug print
+    
     try:
         # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
+        print("DuckDuckGoSearchTool initialized")  # Debug print
         
-        # Perform the search and add debug output
-        print(f"Searching for: {search_query}")
+        # Perform the search
         results = search_tool(search_query)
-        print(f"Got {len(results) if results else 0} results")
+        print(f"Search returned {len(results) if results else 0} results")  # Debug print
         
         if not results:
             return "No results found for the query."
         
-        # Get the first result and format it
+        # Get the first result
         first_result = results[0]
-        summary = f"Latest AI News:\nTitle: {first_result['title']}\nSummary: {first_result['snippet']}"
-        
-        # Add image generation if requested
-        if generate_image:
-            try:
-                print("Attempting to generate image...")
-                image_result = image_generation_tool(search_query)
-                summary += f"\n\nGenerated visualization based on the query."
-            except Exception as e:
-                print(f"Image generation failed: {str(e)}")
-                summary += f"\n\nNote: Image generation was requested but failed."
         
-        return summary
+        # Use final_answer to return the result
+        result_text = f"Latest AI News:\n{first_result['title']}\n{first_result['snippet']}"
+        return result_text
         
     except Exception as e:
-        error_msg = f"Error in my_custom_tool: {str(e)}"
-        print(error_msg)  # For debugging
-        return error_msg
+        error_message = f"Error in my_custom_tool: {str(e)}"
+        print(error_message)  # Debug print
+        return error_message
+        
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -86,7 +80,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool], ## add your tools here (don't remove final answer)
+    tools=[my_custom_tool, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,