Update app.py

app.py

@@ -27,6 +27,7 @@ def run_script(text: str):
         try:
             driver = webdriver.Chrome(options=options)
             driver.get(text)
+            html=driver.page_source()
             driver.implicitly_wait(30)
             driver.set_window_size(1920, 1080)
             ######################
@@ -49,9 +50,9 @@ def run_script(text: str):
             if driver:
                 driver.quit()
 
-        return [Image.open(BytesIO(screenshot)), 'operation success.',cookie_jar]
+        return [Image.open(BytesIO(screenshot)), 'operation success.',cookie_jar,html]
     else:
-        return [None, 'Please enter a valid URL of a website/host.',None]
+        return [None, 'Please enter a valid URL of a website/host.',None,None]
 
 with gr.Blocks() as app:
     inp = gr.Textbox()
@@ -62,5 +63,6 @@ with gr.Blocks() as app:
         with gr.Column():
             outp = gr.Textbox()
             cook = gr.JSON()
-    btn.click(run_script,inp,[outim,outp,cook])
+            html=gr.Textbox()
+    btn.click(run_script,inp,[outim,outp,cook,html])
 app.launch()