Update app.py

app.py

@@ -50,7 +50,7 @@ def run_script(text: str):
             cookie_jar = []
             #cookie_jar.append(driver.get_cookies())
             #print(cookie_jar)
-            screenshot = obj.screenshot_as_png()
+            screenshot = obj.screenshot('tmp.png')
             #screenshot = obj.get_screenshot_as_png()
         except WebDriverException as e:
             return [Image.new('RGB', (1, 1)), e, None,None]
@@ -58,7 +58,8 @@ def run_script(text: str):
             if driver:
                 driver.quit()
 
-        return [Image.open(BytesIO(screenshot)), 'operation success.',cookie_jar,html]
+        #return [Image.open(BytesIO(screenshot)), 'operation success.',cookie_jar,html]
+        return [Image.open('tmp.png'), 'operation success.',cookie_jar,html]
     else:
         return [None, 'Please enter a valid URL of a website/host.',None,None]