Update app.py

app.py

@@ -61,7 +61,7 @@ def run_script(html: str, style: str):
                 driver.quit()
 
         #return [Image.open(BytesIO(screenshot)), 'operation success.',cookie_jar,html]
-        return [Image.open('tmp.png'), 'operation success.',cookie_jar,html]
+        return [Image.open('tmp.png'), 'operation success.',url,html]
     else:
         return [None, 'Please enter a valid URL of a website/host.',None,None]
 
@@ -85,7 +85,7 @@ with gr.Blocks() as app:
             outim = gr.Image()
         with gr.Column():
             outp = gr.Textbox()
-            cook = gr.JSON()
+            cook = gr.Textbox()
             html=gr.Textbox()
     btn.click(run_script,[inp,style],[outim,outp,cook,html])
 app.launch()