Update app.py

app.py

@@ -36,7 +36,7 @@ def get_screenshot(chat,css=None):
     #hti.screenshot(html_str=html, save_as='red_page.png')
     #hti.screenshot(html_str=html_card, css_str=css, save_as='red_page.png')
     hti.screenshot(html_str=html, save_as='red_page.png')
-    return 'red_page.png'
+    return 'red_page.png',html
 
 
 def format_prompt(message, history):
@@ -105,8 +105,9 @@ with gr.Blocks() as app:
                 
                 im_btn=gr.Button("Screenshot")
                 img=gr.Image(type='filepath')
+    html_view=gr.HTML()
     btn.click(chat_inf,[sys_inp,inp,chat_b],chat_b)
-    im_btn.click(get_screenshot,[chat_b],img)
+    im_btn.click(get_screenshot,[chat_b],[img,html_view])
     #im_btn.click(get_screenshot,[chat_b,im_height,im_width,chatblock,theme,wait_time],img)
     #app.load(get_screenshot,inp,img)
 app.launch()