Update app.py

app.py

@@ -58,7 +58,7 @@ $body
 """
 
 
-def get_screenshot(chat,css_in):
+def get_screenshot(chat,css_in,w,h):
     print(chat)
     html_body=""
     for user,bot in chat:
@@ -70,7 +70,7 @@ def get_screenshot(chat,css_in):
     #css = css
     #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.size = (1500, 1000)
+    hti.size = (w, h)
     hti.screenshot(html_str=html, css_str=css_in, save_as='red_page.png')
     return 'red_page.png',html
 
@@ -133,7 +133,7 @@ with gr.Blocks(css=css) as app:
         with gr.Column(scale=1):
             with gr.Group():
                 with gr.Row():
-                    im_height=gr.Number(label="Height",value=5000)
+                    im_height=gr.Number(label="Height",value=500)
                     im_width=gr.Number(label="Width",value=500)
                 wait_time=gr.Number(label="Wait Time",value=3000)
                 theme=gr.Radio(label="Theme", choices=["light","dark"],value="light")
@@ -144,7 +144,7 @@ with gr.Blocks(css=css) as app:
     css_box=gr.Textbox(lines=50,value=css)
     html_view=gr.HTML()
     btn.click(chat_inf,[sys_inp,inp,chat_b],chat_b)
-    im_btn.click(get_screenshot,[chat_b,css_box],[img,html_view])
+    im_btn.click(get_screenshot,[chat_b,css_box,im_width,im_height],[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()