Update app.py

app.py

@@ -16,26 +16,38 @@ 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
     """
-    # First, search using DuckDuckGo
-    search_tool = DuckDuckGoSearchTool()
-    results = search_tool(search_query)
-    
-    if not results:
-        return "No results found for the query."
-    
-    # Get the first result
-    first_result = results[0]
-    summary = f"Found: {first_result['title']}\nSummary: {first_result['snippet']}"
-    
-    # If image generation is requested, also generate an image
-    if generate_image:
-        try:
-            image_result = image_generation_tool(search_query)
-            summary += f"\n\nGenerated visualization based on the query."
-        except Exception as e:
-            summary += f"\n\nImage generation failed: {str(e)}"
-    
-    return summary
+    try:
+        # Initialize the search tool
+        search_tool = DuckDuckGoSearchTool()
+        
+        # Perform the search and add debug output
+        print(f"Searching for: {search_query}")
+        results = search_tool(search_query)
+        print(f"Got {len(results) if results else 0} results")
+        
+        if not results:
+            return "No results found for the query."
+        
+        # Get the first result and format it
+        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
+        
+    except Exception as e:
+        error_msg = f"Error in my_custom_tool: {str(e)}"
+        print(error_msg)  # For debugging
+        return error_msg
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: