Update app.py

app.py

@@ -17,11 +17,12 @@ driver_type = 'chromedriver'
 driver=False
 
 
-def html2im():
+def html2im(html,css=None):
 
-    html = """<h1> An interesting title </h1> This page will be red"""
-    css = "body {background: red;}"
-    hti.screenshot(html_str=html, css_str=css, save_as='red_page.png')
+    html = html
+    #css = css
+    hti.screenshot(html_str=html, save_as='red_page.png')
+    #hti.screenshot(html_str=html, css_str=css, save_as='red_page.png')
     return 'red_page.png'
     
 def get_concat_h_cut(in1, in2):
@@ -127,6 +128,6 @@ with gr.Blocks() as app:
             out=gr.Image()
             outgal=gr.Gallery()
         outim = gr.Image()
-    btn.click(html2im,None,out)
+    btn.click(html2im,inp,out)
     #btn.click(run_script,[inp,height,width,check_b,check_h],[outim,message,outgal,out])
 app.queue(default_concurrency_limit=5).launch()