Update app.py

app.py

@@ -16,45 +16,40 @@ 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("\n=== DEBUG: my_custom_tool started ===")
-    print(f"Received query: {search_query}")
-    
     try:
-        # Initialize the search tool
-        print("Initializing DuckDuckGoSearchTool...")
-        search_tool = DuckDuckGoSearchTool()
-        
-        # Perform the search
-        print("Performing search...")
-        results = search_tool(search_query)
-        
-        print(f"Search complete. Got {len(results) if results else 0} results")
-        
+        # First try to create the search tool
+        print("Attempting to create DuckDuckGoSearchTool...")
+        search_tool = None
+        try:
+            search_tool = DuckDuckGoSearchTool()
+            print("Successfully created DuckDuckGoSearchTool")
+        except Exception as search_tool_error:
+            return f"Failed to create search tool: {str(search_tool_error)}"
+
+        # Then try to perform the search
+        print(f"Attempting search for: {search_query}")
+        try:
+            results = search_tool(search_query)
+            print(f"Search completed with {len(results) if results else 0} results")
+        except Exception as search_error:
+            return f"Failed to perform search: {str(search_error)}"
+
+        # Process the results
         if not results:
-            return "No results found for the query."
-        
-        # Get the first result
+            return "Search completed but no results found."
+
         first_result = results[0]
-        
-        # Create formatted result
         result_text = (
-            f"Latest AI News:\n"
-            f"Title: {first_result['title']}\n"
-            f"Summary: {first_result['snippet']}"
+            f"Here are the latest findings about artificial intelligence:\n\n"
+            f"Title: {first_result.get('title', 'No title available')}\n"
+            f"Summary: {first_result.get('snippet', 'No summary available')}"
         )
         
         print("Successfully created result text")
-        print("=== DEBUG: my_custom_tool completed ===\n")
-        
         return result_text
-        
+
     except Exception as e:
-        error_message = f"ERROR in my_custom_tool: {str(e)}\n"
-        error_message += f"Error type: {type(e)}\n"
-        error_message += f"Error location: {e.__traceback__.tb_frame.f_code.co_filename}, line {e.__traceback__.tb_lineno}"
-        print(error_message)
-        print("=== DEBUG: my_custom_tool failed ===\n")
-        return error_message
+        return f"Unexpected error in my_custom_tool: {str(e)}"
         
 
 @tool