Update app.py

app.py

@@ -10,10 +10,20 @@ from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.common.by import By
 #from selenium_webdriver import WebElement
+from html2image import Html2Image
+hti = Html2Image()
 
 driver_type = 'chromedriver'
 driver=False
 
+
+def html2image():
+
+    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')
+    return 'red_page.png'
+    
 def get_concat_h_cut(in1, in2):
     im1=Image.open(in1)
     im2=Image.open(in2)
@@ -117,6 +127,6 @@ with gr.Blocks() as app:
             out=gr.Image()
             outgal=gr.Gallery()
         outim = gr.Image()
-
-    btn.click(run_script,[inp,height,width,check_b,check_h],[outim,message,outgal,out])
+    btn.click(html2image,None,out)
+    #btn.click(run_script,[inp,height,width,check_b,check_h],[outim,message,outgal,out])
 app.queue(default_concurrency_limit=5).launch()