Update tools.py

tools.py

@@ -180,7 +180,7 @@ class audio_or_mp3__interpreter(Tool):
 
 class Wikipedia_reader(Tool):
     name="wiki_tool"
-    description = "To be used whenever you need to read a Wikipedia page. Will return all the text of the Wikipedia page, to easily read it and find information"
+    description = "To be used whenever you need to read a Wikipedia page. Will return all the html code of the Wikipedia page, to easily read it and find information. IF you need numeric information and can not easily have it with this function, use html_tool"
     inputs = {
         "url": {
             "type": "string",
@@ -191,8 +191,27 @@ class Wikipedia_reader(Tool):
 
     def forward(self, url: str):
         try:
-            page = requests.get(url)
+            page = requests.get({url})
         except Exception as e:
             print('Error downloading page: ',e)
         soup = BeautifulSoup(page.text, 'html.parser')
-        return soup.text
+        return soup
+
+class HTML_find(Tool):
+    name="html_tool"
+    description = "To be used whenever you need to read a Wikipedia page for NUMERICAL INFORMATION. It will extract the wikitables, hence making it easy to read. It completes the wiki_tool"
+    output_type = "string"
+    inputs = {
+        "url": {
+            "type": "string",
+            "description": "The wikippedia url page"
+        },
+        #"subject" : {
+         #   "type" : "string",
+          #  "description": "If no subject is defined, the tool will return all the tables. If you know the numerical output (you must be sure about it) that you're searching, as a year, you can specify it with this input. 
+        #}
+    }
+    def forward(self,url)#,subject="Inconnu"):
+        #if({subject}=="Inconnu"):
+        tables=pd.read_html(url)
+        return tables