Update app.py

app.py

@@ -31,6 +31,7 @@ def run_script(text: str):
             #html=driver.page_source
             driver.implicitly_wait(30)
             driver.set_window_size(1920, 1080)
+            obj = driver.find_element_by_id("in_html")
             ######################
             #inp = driver.find_element("id","input")
             '''
@@ -44,7 +45,7 @@ def run_script(text: str):
             cookie_jar = []
             #cookie_jar.append(driver.get_cookies())
             #print(cookie_jar)
-            screenshot = driver.get_screenshot_as_png()
+            screenshot = obj.get_screenshot_as_png()
         except WebDriverException as e:
             return [Image.new('RGB', (1, 1)), e, None]
         finally: