Update tools.py

tools.py

@@ -12,15 +12,16 @@ class MediaTool(Tool):
     name = "media_tool"
     description = "Used for deciphering video and audio files."
     inputs = {
-        "title": {
+        "filename": {
             "type": "string",
             "description": "The name of the media file to transcribe."
         }
     }
     output_type = "string"
 
-    def forward(self, title: str):
-        return "this tool hasn't been implemented yet. Please return 0 if the task cannot be solved without knowing the contents of this file."
+    def forward(self, filename: str):
+        print(f"called MediaTool with {filename}")
+        return "This tool hasn't been implemented yet. Please return 0 if the task cannot be solved without knowing the contents of this file."
 
 class WikipediaTool(Tool):
     name = "wikipedia_api"
@@ -34,8 +35,13 @@ class WikipediaTool(Tool):
     output_type = "string"
 
     def forward(self, title: str):
-        page = wikipedia.page(title)
-        return page.content
+        print(f"called WikipediaTool with {title}")
+        try:
+            page = wikipedia.page(title)
+            content = page.content[:3000]
+            return content
+        except Exception as e:
+            return f"ERROR: {e}"
 
 class ExcelTool(Tool):
     name = "read_excel"
@@ -48,9 +54,13 @@ class ExcelTool(Tool):
     }
     output_type = "string"
 
-    def forward(self, path: str):
-        df = pd.read_excel(path, engine = "openpyxl")
-        return df
+    def forward(self, filename: str):
+        print(f"called ExcelTool with {filename}")
+        try:
+            df = pd.read_excel(filename, engine = "openpyxl")
+            return df
+        except Exception as e:
+            return f"ERROR: {e}"
 
 class WebscraperTool(Tool):
     name = "webscraper"
@@ -64,13 +74,17 @@ class WebscraperTool(Tool):
     output_type = "string"
 
     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}")
+        print(f"called WebscraperTool with {url}")
+        try:
+            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:
+                return "Website cannot be reached."
+        except Exception as e:
+            return f"ERROR: {e}"
 
 # Initialize the DuckDuckGo search tool
 class InternetSearchTool(Tool):
@@ -85,32 +99,13 @@ class InternetSearchTool(Tool):
     output_type = "string"
 
     def forward(self, question: str):
-        search_tool = DuckDuckGoSearchTool()
-        result = search_tool(question)
-        return result
-
-
-class WeatherInfoTool(Tool):
-    name = "weather_info"
-    description = "Fetches dummy weather information for a given location."
-    inputs = {
-        "location": {
-            "type": "string",
-            "description": "The location to get weather information for."
-        }
-    }
-    output_type = "string"
-
-    def forward(self, location: str):
-        # Dummy weather data
-        weather_conditions = [
-            {"condition": "Rainy", "temp_c": 15},
-            {"condition": "Clear", "temp_c": 25},
-            {"condition": "Windy", "temp_c": 20}
-        ]
-        # Randomly select a weather condition
-        data = random.choice(weather_conditions)
-        return f"Weather in {location}: {data['condition']}, {data['temp_c']}°C"
+        print(f"called InternetSearchTool with {question}")
+        try:
+            search_tool = DuckDuckGoSearchTool()
+            result = search_tool(question)
+            return result
+        except Exception as e:
+            return f"ERROR: {e}"
 
 class HubStatsTool(Tool):
     name = "hub_stats"