Update extract.py

extract.py

@@ -17,6 +17,8 @@ def take_webdata(url):
         # Get the page title
         page_title = wd.title
         screenshot = wd.get_screenshot_as_png()
+        html = wd.execute_script("return document.documentElement.outerHTML;")
+
 
     except WebDriverException as e:
         return Image.new('RGB', (1, 1)), page_title