Update app.py

app.py

@@ -16,16 +16,19 @@ 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
+    print("\n=== DEBUG: my_custom_tool started ===")
+    print(f"Received query: {search_query}")
     
     try:
         # Initialize the search tool
+        print("Initializing DuckDuckGoSearchTool...")
         search_tool = DuckDuckGoSearchTool()
-        print("DuckDuckGoSearchTool initialized")  # Debug print
         
         # Perform the search
+        print("Performing search...")
         results = search_tool(search_query)
-        print(f"Search returned {len(results) if results else 0} results")  # Debug print
+        
+        print(f"Search complete. Got {len(results) if results else 0} results")
         
         if not results:
             return "No results found for the query."
@@ -33,13 +36,24 @@ def my_custom_tool(search_query: str, generate_image: bool = False) -> str:
         # Get the first result
         first_result = results[0]
         
-        # Use final_answer to return the result
-        result_text = f"Latest AI News:\n{first_result['title']}\n{first_result['snippet']}"
+        # Create formatted result
+        result_text = (
+            f"Latest AI News:\n"
+            f"Title: {first_result['title']}\n"
+            f"Summary: {first_result['snippet']}"
+        )
+        
+        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)}"
-        print(error_message)  # Debug print
+        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