Update app.py

Added website to image feature

app.py

@@ -13,6 +13,18 @@ def html_to_image(html_code, width, height):
     image = Image.open(BytesIO(imgkit.from_string(html_code, False, options=options)))
     return image
 
+def website_to_image(website_link, width, height):
+    if not website_link:
+        website_link = "https://www.google.com"
+    options = {
+        'format': 'png',
+        'width': str(width),
+        'height': str(height),
+        'encoding': "UTF-8"
+    }
+    image = Image.open(BytesIO(imgkit.from_url(website_link, False, options=options)))
+    return image
+
 with gr.Blocks() as interface:
     gr.Markdown("# HTML to Image Converter")
     gr.Markdown("Enter HTML code and set dimensions to generate an image")
@@ -34,7 +46,15 @@ with gr.Blocks() as interface:
                 value=720,
                 step=10,
                 )
+            with gr.Row(variant="panel", ):
+                with gr.Column():
+                    website_link = gr.Textbox(
+                        label="Website Link",
+                        placeholder="https://www.google.com"
+                    )
+                snapshotwebsite = gr.Button("Show Website", size="lg")
         output_image = gr.Image(type="pil", format="PNG", show_label=False)
+    generate_button = gr.Button("Refresh")
 
     html_code_input.change(
         fn=html_to_image,
@@ -52,12 +72,17 @@ with gr.Blocks() as interface:
         outputs=output_image
     )
 
-    generate_button = gr.Button("Refresh")
     generate_button.click(
         fn=html_to_image,
         inputs=[html_code_input, width_input, height_input],
         outputs=output_image
     )
+    
+    snapshotwebsite.click(
+        fn=website_to_image,
+        inputs=[website_link, width_input, height_input],
+        outputs=output_image
+    )
 
     gr.Examples(
         examples=[