Update app.py

app.py

@@ -122,38 +122,40 @@ ti_tool = TextInspectorTool(model, text_limit)
 browser = SimpleTextBrowser(**BROWSER_CONFIG)
 
 
-class DuckDuckGoSearchTool(Tool):  # Inherit from base Tool class
+class DuckDuckGoSearchTool(Tool):
     """Search tool using DuckDuckGo"""
     name = "web_search"
     description = "Search the web using DuckDuckGo (current information)"
+    inputs = [{
+        "name": "query",
+        "type": "string",
+        "description": "Search query string",
+        "required": True
+    }]
+    outputs = [{
+        "name": "results",
+        "type": "string",
+        "description": "Formatted search results"
+    }]
     
-    def __init__(self, max_results: int = 5):
+    def __init__(self, max_results: int = 5): 
         super().__init__()
         self.max_results = max_results
         
     def run(self, query: str) -> str:
-        """Return formatted search results (snippets from webpages)"""
+        """Return formatted search results"""
         try:
-            web_results = []
             with DDGS() as ddgs:
-                for result in ddgs.text(query, max_results=self.max_results):
-                    web_results.append({
-                        'title': result['title'],
-                        'url': result['href'],
-                        'content': result['body']
-                    })
+                results = list(ddgs.text(query, max_results=self.max_results))
             
-            formatted_results = []
-            for idx, res in enumerate(web_results[:self.max_results], 1):
-                formatted_results.append(
-                    f"[{idx}] {res['title']}\n"
-                    f"URL: {res['url']}\n"
-                    f"Content: {res['content'][:500]}{'...' if len(res['content']) > 500 else ''}"
-                )
-            return "\n\n".join(formatted_results)
+            formatted = [
+                f"[{i+1}] {r['title']}\nURL: {r['href']}\n{r['body'][:500]}"
+                for i, r in enumerate(results)
+            ]
+            return "\n\n".join(formatted)
         except Exception as e:
             return f"Search error: {str(e)}"
-
+            
 WEB_TOOLS = [
     DuckDuckGoSearchTool(max_results=5),
     VisitTool(browser),
@@ -164,7 +166,6 @@ WEB_TOOLS = [
     ArchiveSearchTool(browser),
     TextInspectorTool(model, text_limit),
 ]
-
 # Agent creation in a factory function
 def create_agent():
     """Creates a fresh agent instance for each session"""
@@ -325,4 +326,4 @@ Advanced question answering using DuckDuckGo search and custom AI models""")
         demo.launch(debug=True, share=False, **kwargs)
 
 if __name__ == "__main__":
-    GradioUI().launch()
+    GradioUI().launch()