Update app.py

app.py

@@ -1,20 +1,19 @@
 import os
-from html2image import Html2Image
+#from html2image import Html2Image
 import gradio as gr
 print(os.environ)
-hti = Html2Image()
-def txt2img(inp):
-    html = f"""
-    <div class='im_div'>
-      {inp}
-      <div>
-    """
-    css = ".im_div {background: red;}"
 
-    hti.screenshot(html_str=html, css_str=css, save_as='page.png')
-
-    #imgkit.from_string(body, 'out.png')
-    return (html,'page.png')
+import asyncio
+from playwright.async_api import async_playwright
+ 
+async def main(inp):
+    async with async_playwright() as p:
+        browser = await p.chromium.launch()
+        page = await browser.new_page()
+        await page.goto('https://www.bbc.co.uk/news/business-63709754')
+        await page.locator('.ssrcss-hmf8ql-BoldText').screenshot(path='element.png')
+        await browser.close()
+    return ("done",'element.png')
 
 
 with gr.Blocks() as app:
@@ -22,5 +21,5 @@ with gr.Blocks() as app:
     btn=gr.Button()
     outhtml=gr.HTML()
     outp=gr.Image()
-    btn.click(txt2img,inp,[outhtml,outp])
+    btn.click(main,inp,[outhtml,outp])
 app.launch()