Update app.py

app.py

@@ -10,46 +10,34 @@ from Gradio_UI import GradioUI
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 # Replace this part in app.py
 @tool
-def my_custom_tool(search_query: str, generate_image: bool = False) -> str:
-    """A tool that combines web search and optional image generation
+def my_custom_tool(topic: str, include_links: bool = False) -> str:
+    """A tool that fetches comprehensive information about any topic using DuckDuckGo search
     Args:
-        search_query: The search query to look up
-        generate_image: Whether to also generate an image based on the search results
+        topic: The topic to search for information about
+        include_links: Whether to include source links in the response (default: False)
     """
     try:
-        # 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
+        print(f"Searching for information about: {topic}")
+        search_tool = DuckDuckGoSearchTool()
+        results = search_tool(topic)
+        
         if not results:
-            return "Search completed but no results found."
-
-        first_result = results[0]
-        result_text = (
-            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')}"
-        )
+            return f"No information found about {topic}"
+            
+        # Get the top 3 results for more comprehensive information
+        summary = f"Here's what I found about {topic}:\n\n"
+        
+        for i, result in enumerate(results[:3], 1):
+            summary += f"{i}. {result['title']}\n"
+            summary += f"{result['snippet']}\n"
+            if include_links:
+                summary += f"Source: {result['link']}\n"
+            summary += "\n"
+            
+        return summary.strip()
         
-        print("Successfully created result text")
-        return result_text
-
     except Exception as e:
-        return f"Unexpected error in my_custom_tool: {str(e)}"
+        return f"Error while searching: {str(e)}"
         
 
 @tool