Update tools.py

tools.py

@@ -5,6 +5,7 @@ from huggingface_hub import list_models
 import requests
 from bs4 import BeautifulSoup
 import openpyxl
+import wikipedia
 
 class WikipediaTool(Tool):
     name = "wikipedia_api"
@@ -17,10 +18,9 @@ class WikipediaTool(Tool):
     }
     output_type = "string"
 
-    def forward(self, question: str):
-        search_tool = DuckDuckGoSearchTool()
-        result = search_tool(question)
-        return result
+    def forward(self, title: str):
+        page = wikipedia.page(title)
+        return page.content
 
 class ExcelTool(Tool):
     name = "read_excel"
@@ -49,10 +49,14 @@ class WebscraperTool(Tool):
     }
     output_type = "string"
 
-    def forward(self, question: str):
-        search_tool = DuckDuckGoSearchTool()
-        result = search_tool(question)
-        return result
+    def forward(self, url: str):
+        response = requests.get(url, stream=True)
+        if response.status_code == 200:
+            soup = BeautifulSoup(response.content, 'html.parser')
+            html_text = soup.get_text()
+            return html_text
+        else:
+            raise Exception(f"Failed to retrieve the webpage. Status code: {response.status_code}")
 
 # Initialize the DuckDuckGo search tool
 class InternetSearchTool(Tool):