Spaces:

flatindo
/

analyze-data

Runtime error

flatindo commited on Sep 21, 2023

Commit

850d477

1 Parent(s): ecabf86

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from selenium.common.exceptions import WebDriverException
 from PIL import Image
 from io import BytesIO
-def take_screenshot(url):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
@@ -27,11 +27,12 @@ def take_screenshot(url):
     return Image.open(BytesIO(screenshot))
 iface = gr.Interface(
-    fn=take_screenshot,
-    inputs=gr.inputs.Textbox(label="Website URL", default="https://google.com"),
-    outputs=gr.Image(type="pil", height=360, width=540),  # Adjust the image size here
-    title="Website Scrape",
-    description="Take data a website.",
 )
 iface.launch()

 from PIL import Image
 from io import BytesIO
+def web_scrape(url):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
     return Image.open(BytesIO(screenshot))
 iface = gr.Interface(
+    fn=web_scrape,
+    inputs=gr.inputs.Textbox(label="Website URL", default="https://example.com"),
+    outputs=gr.outputs.Textbox(label="Web Content"),
+    title="Web Scraping with Selenium (Body Tag)",
+    description="Scrape the content of a website's <body> tag using Selenium.",
 )
 iface.launch()