Update app.py

app.py

@@ -58,7 +58,7 @@ $body
 """
 
 
-def get_screenshot(chat):
+def get_screenshot(chat,css_in):
     print(chat)
     html_body=""
     for user,bot in chat:
@@ -70,8 +70,8 @@ def get_screenshot(chat):
     #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.screenshot(html_str=html, css_str=css, save_as='red_page.png')
+    #hti.size = (1500, 1000)
+    hti.screenshot(html_str=html, css_str=css_in, save_as='red_page.png')
     return 'red_page.png',html
 
 
@@ -141,9 +141,10 @@ with gr.Blocks(css=css) as app:
                 
                 im_btn=gr.Button("Screenshot")
                 img=gr.Image(type='filepath')
+    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],[img,html_view])
+    im_btn.click(get_screenshot,[chat_b,css_box],[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()