Update app.py

app.py

@@ -40,23 +40,26 @@ def run_script(text: str):
             #inp.send_keys("test")
             ######################
             
-            print(driver.get_cookies())
+            cookies = driver.get_cookies()
             screenshot = driver.get_screenshot_as_png()
         except WebDriverException as e:
-            return [Image.new('RGB', (1, 1)), e]
+            return [Image.new('RGB', (1, 1)), e, None]
         finally:
             if driver:
                 driver.quit()
 
-        return [Image.open(BytesIO(screenshot)), 'operation success.']
+        return [Image.open(BytesIO(screenshot)), 'operation success.',cookies]
     else:
-        return [None, 'Please enter a valid URL of a website/host.']
+        return [None, 'Please enter a valid URL of a website/host.',None]
 
 with gr.Blocks() as app:
     inp = gr.Textbox()
     btn= gr.Button()
     with gr.Row():
-        outim = gr.Image()
-        outp = gr.Textbox()
-    btn.click(run_script,inp,[outim,outp])
+        with gr.Column():
+            outim = gr.Image()
+        with gr.Column():
+            outp = gr.Textbox()
+            cook = gr.JSON()
+    btn.click(run_script,inp,[outim,outp,cook])
 app.launch()